[JAVA] 체육복

문제

https://programmers.co.kr/learn/courses/30/lessons/42862

코딩테스트 연습 - 체육복

 

주의할 점

체육복이 없다면 앞 뒤에 사람한테 체육복을 빌려야 한다. 하지만 예외적으로 1번인 사람은 앞의 사람이 없고 마지막 사람은 뒤에 사람이 없다. 그러므로 배열에 접근할 때 주의해야한다. 그렇지 않으면 에러가 발생한다.

원래는 boolean[] 형으로 체육복이 있는 사람과 없는 사람으로 분류했지만 int[]로 변경해서 -1이면 없는 상태, 0이면 있는 상태, 1이면 1개가 더 있는 상태로 분류했다. 

크게 흐름을 살펴보면

1.lost 배열에 있는 번호들은 값을 -1해준다.

2.reserve 배열에 있는 번호는 값을 +1해준다. 이렇게 하면 자연스럽게 없는 사람, 있는 사람, 여유분이 있는 사람으로 나뉘게 된다. 그리고 여유분이 있지만 도난당한 사람도 해결된다.

3. 모든 번호를 반복한다.

 만약 체육복이 없는 사람이면 

3-1. 만약 첫번째 사람이 아니고 앞사람이 여유분이 있으면 

3-2  그밖에 만약  마지막 사람이 아니고 뒷사람이 여유분이 있으면

3-3 그외에는 옷을 빌릴 수 없다.

 

 

코드

 

package 프로그래머스.레벨1.체육복;

public class Solution {
    public int solution(int n, int[] lost, int[] reserve) {
        int[] student = new int[n];
        int answer = n;

        for (int num : lost)
            student[num-1]--;
        for (int num : reserve)
            student[num-1]++;

        for (int i=0; i<n; i++) {
            if (student[i] == -1) {
                if (i-1>=0 && student[i-1] == 1) {
                    student[i]++;
                    student[i-1]--;
                } else if (i+1 < n && student[i+1] == 1) {
                    student[i]++;
                    student[i+1]--;
                } else
                    answer--;
            }
        }

        return answer;
    }