[Greedy] 프로그래머스 체육복 - java

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

프로그래머스

문제

- 바로 앞번호의 학생이나 바로 뒷번호의 학생에게만 체육복을 빌려줄 수 있다.

- 체육복을 적절히 빌려 최대한 많은 학생이 체육수업을 듣게 하는 문제

 

조건

- 2 <= n <= 30 

- 체육복을 도난당한 학생수는 1명 이상 n명 이하이고, 중복되는 번호는 없다.

- 여벌의 체육복을 가져온 학생의 수는 1명 이상 n명 이하이고, 중복되는 번호는 없다.

- 여벌 체육복이 있는 학생만 다른 학생에게 체육복을 빌려줄 수 있다.

- 여벌 체육복을 가져온 학생이 체육복을 도난당했을 수도 있다. 이때 이 학생은 체육복을 하나만 도난당했다고 가정하며, 남은 체육복이 하나이기에 다른 학생에게는 체육복을 빌려줄 수 없다.

ex) 2번이 여벌 체육복이 있어서 총 체육복이 2벌인데 하나 도난당했으므로 남은 체육복은 하나라서 빌려줄 수 없다.

 

입력

- n: 전체 학생 수

- lost[]: 체육복을 도난당한 학생들의 번호가 담긴 배열

- reserve[]: 여벌의 체육복을 가져온 학생들의 번호가 담긴 배열

 

출력

- 체육수업을 들을 수 있는 학생의 최댓값

 

코드

package Greedy;

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

        // 여벌 체육복을 가져온 학생이 도난당한 경우
        for(int i = 0; i < lost.length; i++) {
            for(int j = 0; j < reserve.length; j++) {
                if(lost[i] == reserve[j]) {
                    answer++; // 여벌 체육복을 가져온 학생이 도난당했으므로 1개의 체육복을 가짐 -> 무조건 체육수업 들을 수 있음
                    // lost, reserve 배열에서 제외시킨다
                    lost[i] = -1;
                    reserve[i] = -1;
                    break;
                }
            }
        }

        // 여벌 체육복이 없고 도난당한 경우
        for(int i = 0; i < lost.length; i++) {
            for(int j = 0; j < reserve.length; j++) {
                if((lost[i]-1 == reserve[j]) || (lost[i]+1 == reserve[j])) { // 앞뒤로 빌려줄 수 있는 경우
                    answer++;
                    reserve[j] = -1; // 빌려줬으므로 reserve 배열에서 제외시킨다.
                    break;
                }
            }
        }

        return answer;
    }

    public static void main(String[] args) {
        int[] lost = {2, 4};
        int[] reserve = {1, 3, 5};

        System.out.println(solution(5, lost, reserve));
    }
}