[JAVA] 프로그래머스 Lv.1 : 체육복

🔥 코딩 테스트 연습 > 탐욕법 (Greedy)

 

예제:

 

n	lost	reserve	return
5	[2, 4]	[1, 3, 5]	5
5	[2, 4]	[3]	4
3	[3]	[1]	2

 

 

제한 조건:

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

---

Solution #1

class Solution {
    public int solution(int n, int[] lost, int[] reserve) {
        int count = 0;
        for (int i = 0; i < lost.length; i++) {
            for (int j = 0; j < reserve.length; j++) {
                if (lost[i] == reserve[j]) {
                    lost[i] = reserve[j] = -1;
                    count++;
                    break;
                }
            }
        }

        for (int k : lost) {
            for (int j = 0; j < reserve.length; j++) {
                if (k == reserve[j] - 1 || k == reserve[j] + 1) {
                    reserve[j] = -1;
                    count++;
                    break;
                }
            }
        }

        return n - lost.length + count;
    }
}

 

Result #1

 

💡 여벌의 체육복을 가져온 학생이 체육복을 도난당했을 경우의 수(==)를 먼저 계산하고, 그 학생의 번호를 -1로 초기화한 다. 남은 학생들 중에서 뒷번호(-1)이거나 앞번호(+1)일 경우를 조건문으로  검사하고, 체육복을 빌린 학생의 번호를 -1로 초기화 해서 다음 검사 때 조건문에 포함이 안되게 한다.

결과 값은 전체 학생수(n) - 잃어버린 학생 수(lost 길이) + 빌린 학생 수 이다.

 

---

 

More Algorithm!

 

👇👇

 

github.com/ggujangi/ggu.programmers

ggujangi/ggu.programmers

 

 

출처 : 프로그래머스