[프로그래머스]탐욕법>체육복 Java

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

코딩테스트 연습 - 체육복

 

1. 도난 당한 학생보다 1 작거나 1 큰 수 학생이 여벌이 있으면 체육 수업을 들을 수 있다.

2. 여벌이 있는 학생이 도난 당했다면 -1로 해주면서 다른 학생에게 빌려주지 않도록 한다.

 

public static int solution(int n, int[] lost, int[] reserve) {
        int answer = n - lost.length; // 체육 수업을 들을 수 있는 학생 수
        
        // 여벌이 있는 학생이지만 체육복을 도난 당한 경우
        for(int i=0; i<lost.length; i++) {
        	for(int j=0; j<reserve.length; j++) {
        		if(lost[i] == reserve[j]) {
        			answer++; // 체육 수업을 들을 수 있는 학생이 추가됨
        			lost[i] = reserve[j] = -1; // -1로 바꿔주면서 더이상 빌려주지 않도록 한다.
        			break;
        		}
        	}
        }
        
        // 도난 당했고 여벌이 없는 경우
        for(int lostN : lost) {
        	for(int i = 0; i<reserve.length; i++) {
        		if(reserve[i] == lostN-1 || reserve[i] == lostN+1) {
        			answer++;
        			reserve[i] = -1; // -1로 바꿔주면서 더이상 빌려주지 않도록 한다.
        			break;
        		}
        	}
        }
        return answer;
    }