[Java] 프로그래머스. 체육복

문제

https://school.programmers.co.kr/learn/courses/30/lessons/42862?language=java# 

프로그래머스

 

필요한 개념

• set 자료구조

 

풀이

로직은 다음과 같다.

1. 체육복을 가진 학생 배열을 오름차순 정렬한다.
2. 체육복 있는 학생 배열을 탐색하면서 , 도난당한 경우를 제외한다.
3. 체육복 있는 학생 배열을 탐색하면서 , 내 번호의 왼쪽 학생, 오른쪽 학생 순으로 체육복 잃어버린 학생을 체크
4. 전체 학생 수 - 체육복을 받지 못한 학생 수 리턴.

 

코드

import java.util.*;

class Solution {
    public int solution(int n, int[] lost, int[] reserve) {
        HashSet<Integer> set = new HashSet<Integer>();
        Arrays.sort(reserve);
        
        for(int x: lost)
            set.add(x);
        
        for(int i =0 ;i < reserve.length; i++){
            if(set.contains(reserve[i])){
              set.remove(reserve[i]);
            reserve[i] = -1;
            } 
        }
        
        int ans =reserve.length ;
        for(int i =0 ;i < reserve.length; i++){
            int left = reserve[i] -1;
            int right = reserve[i]+1;
            if(set.contains(left)){
                set.remove(left);
            }else if(set.contains(right)){
                set.remove(right);
            }
        }
        return n - set.size();
    }
}