kingsubin

class Solution {
    public int solution(int n, int[] lost, int[] reserve) {
        int answer = n;
        int[] init = new int[n]; // 2 <= n <= 30, [0 ~ n-1]

        // 체육복 잃어버린 사람
        for (int i = 0; i < lost.length; i++) {
            init[lost[i]-1]--;
        }

        // 여벌 가지고 있는 사람
        for (int i = 0; i < reserve.length; i++) {
            init[reserve[i]-1]++;
        }

        for (int i = 0; i < init.length; i++) {
            if (init[i] == -1) { // 체육복 없는 사람
                if (i-1 >= 0 && init[i-1] == 1) { // 배열범위안에서 앞사람 여벌
                    init[i]++;
                    init[i-1]--;
                } else if (i+1 < init.length && init[i+1] == 1) { // 뒷사람 여벌
                    init[i]++;
                    init[i+1]--;
                } else {
                    answer--;
                }
            }
        }

        return answer;
    }
}

- 테스트케이스 1개에서 계속 실패해서 다른 코드봄

- 내가 처음에 적었던거랑 거의 비슷한데 배열돌리때 약간 달라서 수정

- 앞사람 뒷사람 계산할때 for문 따로 2개 구별해서 적었는데 i-1 >= 0 이런식으로 범위 지정하니까 더 간단해짐

 

 

 

---

programmers.co.kr/learn/challenges