프로그래머스_고득점kit_그리디_체육복

풀이

import java.util.HashSet;

class Solution {
    public int solution(int n, int[] lost, int[] reserve) {
        // 1. Set을 사용해 중복 제거
        HashSet<Integer> lostSet = new HashSet<>();
        HashSet<Integer> reserveSet = new HashSet<>();

        for (int l : lost) {
            lostSet.add(l);
        }
        for (int r : reserve) {
            // 여벌이 있지만 도난당한 경우 먼저 처리
            if (lostSet.contains(r)) {
                lostSet.remove(r);
            } else {
                reserveSet.add(r);
            }
        }

        // 2. 체육복 빌려주기
        for (int r : reserveSet) {
            if (lostSet.contains(r - 1)) { // 앞번호 학생에게 빌려주기
                lostSet.remove(r - 1);
            } else if (lostSet.contains(r + 1)) { // 뒷번호 학생에게 빌려주기
                lostSet.remove(r + 1);
            }
        }

        // 3. 최종적으로 체육복을 가진 학생 수 반환
        return n - lostSet.size();
    }
}

---