[programmers] 체육복

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

코딩테스트 연습 - 체육복

import java.util.*;

class Solution {
    public int solution(int n, int[] lost, int[] reserve) {
        Set<Integer> reserveSet = new HashSet<>();
		Set<Integer> lostSet = new HashSet<>();
		for(int r : reserve) {
			reserveSet.add(r);
		}
		for(int l : lost) {
			if(reserveSet.contains(l)) {
				reserveSet.remove(l);
			} else {
				lostSet.add(l);
			}
		}
		List<Integer> temp = new ArrayList<>(lostSet);
		for(int l : temp) {
			if(reserveSet.contains(l - 1)) {	// 앞사람에게 빌린다.
				lostSet.remove(l);
				reserveSet.remove(l - 1);
			} else if(reserveSet.contains(l + 1)) {	// 뒷사람에게 빌린다.
				lostSet.remove(l);
				reserveSet.remove(l + 1);
			}
		}
        return n - lostSet.size();
    }
}