[프로그래머스 C++] 체육복 / Greedy

코딩테스트 연습 - 체육복 | 프로그래머스 (programmers.co.kr)

코딩테스트 연습 - 체육복

 

 

#include <string>
#include <vector>

using namespace std;

int solution(int n, vector<int> lost, vector<int> reserve) {
    int answer = 0;

    vector<int> array(n+2,1);
    
     for(int i=0;i<reserve.size();i++)
    {
        array[reserve[i]]++;
    }
    
    
    for(int i=0;i<lost.size();i++)
    {
        array[lost[i]]--;
    }
    
  
     for(int i=1;i<=n;i++)
    {
        if(array[i-1]==0&&array[i]==2)
        {
            array[i-1]=array[i]=1;      
        }
        else if(array[i]==2 && array[i+1]==0) 
        {
            array[i]=array[i+1]=1;  
        }
         
        
       
    }
    
     for(int i=1;i<=n;i++)
         {
             if(array[i]>0)
             {
                 answer++;
             }
         }
    
    
    return answer;
}