체육복

programmers.co.kr/learn/courses/30/lessons/42862?language=java

코딩테스트 연습 - 체육복

 

class Solution {
    public int solution(int n, int[] lost, int[] reserve) {
        int[] clothes = new int[n];
        
        for (int l : lost) {
            clothes[l - 1] -= 1;
        }
        
        for (int r : reserve) {
            clothes[r -1] += 1;
        }
            
        /**
        * c[i] 값이 음수면 체육복이 없는 상태, 
        * c[i] 값이 0이면 1개만 갖고있는 상태
        * c[i] 값이 1이상이면 여분이 있는 상태
        */
        for (int i = 0; i < n; i++) {
            // 앞번호에게 빌리기
            if (i > 0 && clothes[i] < 0 && clothes[i - 1] > 0) {
                clothes[i] += 1;
                clothes[i - 1] -= 1;
            }
            
            // 뒷번호에게 빌리기
            if (i + 1 < n && clothes[i] < 0 && clothes[i + 1] > 0) {
                clothes[i] += 1;
                clothes[i + 1] -= 1;
            }
        }
        
        int answer = 0;
        for (int cloth : clothes) {
            if (cloth >= 0) {
                answer += 1;
            }
        }
        
        return answer;
    }
}