체육복

사용 알고리즘: 그리디..?

사용 언어: java

 

import java.util.*;
import java.io.*;

class Solution {
    static int[] PEkit;
    public int solution(int n, int[] lost, int[] reserve) {
        PEkit = new int[n];
        Arrays.fill(PEkit, 1);
        
        for(int i: reserve){
            PEkit[i-1]++;
        }
        for(int i: lost){
            PEkit[i-1]--;
        }
        
        //        
        int answer = 0;
        
        for(int i=0; i<n; i++){
            if(PEkit[i]!=0) answer++;
            else{
                if(i!=0 && PEkit[i-1]==2){
                    PEkit[i-1]--;
                    PEkit[i]++;
                    answer++;
                }
                else if(i!=n-1 && PEkit[i+1]==2){
                    PEkit[i+1]--;
                    PEkit[i]++;
                    answer++;
                }
            }
        }
        
        return answer;
    }
}