[C#] 체육복 - 프로그래머스

 

 

문제 : 전체 학생수 n, 체육복 을 도난당한 학생들의 번호가 담긴 배열 Lost, 여벌의 체육복을 가져온 학생들의 번호가 담긴 배열 reserve 가 매개변수로 주어질때, 체육수업을 들을 수 있는 학생의 최댓값을 return 하도록 solution 을 작성

 

단 학생들의 번호는 체격 순으로 매겨져 있어, 바로 앞번호의 학생이나 바로 뒷번호의 학생에게만 체육복을 빌려줄 수 있다. 

 

제한 사항

• 전체 학생의 수는 2명 이상 30명 이하입니다.
• 체육복을 도난당한 학생의 수는 1명 이상 n명 이하이고 중복되는 번호는 없습니다.
• 여벌의 체육복을 가져온 학생의 수는 1명 이상 n명 이하이고 중복되는 번호는 없습니다.
• 여벌 체육복이 있는 학생만 다른 학생에게 체육복을 빌려줄 수 있습니다.
• 여벌 체육복을 가져온 학생이 체육복을 도난당했을 수 있습니다. 이때 이 학생은 체육복을 하나만 도난당했다고 가정하며, 남은 체육복이 하나이기에 다른 학생에게는 체육복을 빌려줄 수 없습니다.

 

풀이 - 탐욕알고리즘 이라고 하는데 해당 알고리즘을 다시 공부할 필요가 있어보임..

C# 강의 들을 때 배열을 활용하여 이동 하는 방법이 기억나서 그방법으로 구현 진행

 

1. 전체 학생수의 배열 생성 후 1로 채우기 (체육복을 전부 가지고 있다고 전제함)

2. 잃어버린 한학생의 배열 [lost] 별로 -1 진행

3. 여벌의 수를 가지고있는 학생의 배열 [reserve] 별로 +1 을 진행

4. 조건에 맞춰 여벌을 건내면 -1 받으면 +1 을 진행

5. 전체 학생 수 배열이 1이상 값의 count 를 가져오기

 

시간이...오래 걸려...좋은 것은 아닌거같다...

 

 public static int solution(int n, int[] lost, int[] reserve)
        {
            int answer = 0;

            if (n <= 1) return 0;
             
            int[] total = new int[n];
            total = total.Select(x => x = 1).ToArray();

            for (int i = 0; i < lost.Length; i++)
                total[lost[i] - 1] -= 1;

            for (int i = 0; i < reserve.Length; i++)
                total[reserve[i] - 1] += 1;


            for (int i = 0; i < total.Length; i++)
            {
                if (total[i] == 0 && i == 0)
                {
                    if (total[i + 1] == 2)
                    {
                        total[i] += 1;
                        total[i + 1] -= 1;
                    }
                }
                if (total[i] == 0 && i == total.Length -1)
                {
                    if (total[i - 1] == 2)
                    {
                        total[i] += 1;
                        total[i - 1] -= 1;
                    }
                }

                if (total[i] == 0 && i > 1 && i < total.Length - 1)
                {
                    if (total[i - 1] == 2)
                    {
                        total[i] += 1;
                        total[i - 1] -= 1;
                    }

                    if (total[i + 1] == 2)
                    {
                        total[i] += 1;
                        total[i + 1] -= 1;
                    }
                }
            }

            answer = total.Where(x => x >= 1).Count();
            return answer;
         }