int answer = n - lost.length;
Arrays.sort(lost);
Arrays.sort(reserve);
체육 수업을 들을 수 있는 학생의 최댓값 = 전체 학생수(n) - 체육복을 도난당한 학생 수(lost.length)
체육복을 도난당한 학생들의 번호가 담긴 배열과 여벌의 체육복을 가져온 학생들의 번호가 담긴 배열을 Arrays.sort()
를 사용하여 정렬
for (int i = 0; i < lost.length; i++) {
for (int j = 0; j < reserve.length; j++) {
if (lost[i] == reserve[j]) {
answer++; // 자기 옷을 입으면 되니까 증가
lost[i] = -1; // 도난당한 학생 리스트에서 제외
reserve[j] = -1; // 체육복을 빌려줄 수 없음
break;
}
}
}
여벌 체육복을 가져온 학생이 도난 당한 경우 체육복을 다른 사람한테 빌려줄 수 없음
lost, reserve 배열을 반복하면서 두 값이 일치할 경우(자기 자신이 포함되어 있다면)
자신의 옷을 입으면 되므로 참여자 수 증가
lost, reserve 배열의 각 인덱스를 무의미한 값인 -1로 대입하여 카운트 되지 않도록 처리
break
for (int i = 0; i < lost.length; i++) {
for (int j = 0; j < reserve.length; j++) {
if (lost[i] == reserve[j] - 1 || lost[i] == reserve[j] + 1) {
answer++;
reserve[j] = -1; // 빌려주는 학생은 더 이상 빌려줄 수 없음
lost[i] = -1;
break;
}
}
}
도난 당한 학생에게 체육복을 빌려주는 경우(여벌을 챙겨온 사람이 앞, 뒷번호에 체육복을 빌려줄 경우)
lost, reserve 배열을 반복
reserve 배열의 인덱스 값 -1 또는 +1이 lost 배열의 인덱스 값과 일치하면 빌려줌
lost[i]
는 더 이상 빌릴 수 없고 reserve[i]
는 더 이상 빌려줄 수 없으므로 무의미한 값 -1 대입
break
import java.util.Arrays; class Solution { public int solution(int n, int[] lost, int[] reserve) { // 전체학생 수 : n // 체육복을 도난 당한 학생들 번호 int[] lost // 여벌의 체육복을 가져온 학생의 번호 int[] reserve // 체육 수업을 들을 수 있는 학생의 최대값 // 전체 n - lost 인원 int answer = n - lost.length; Arrays.sort(lost); Arrays.sort(reserve); // 여벌 체육복을 가져온 학생이 도난 당한 경우 for (int i = 0; i < lost.length; i++) { for (int j = 0; j < reserve.length; j++) { if (lost[i] == reserve[j]) { answer++; // 자기 옷을 입으면 되니까 증가 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] == reserve[j] - 1 || lost[i] == reserve[j] + 1) { answer++; lost[i] = -1; reserve[j] = -1; // 빌려주는 학생은 더 이상 빌려줄 수 없음 break; } } } return answer; } }