[c++] 프로그래머스 - 체육복( Level 1)

문제

코딩테스트 연습 - 체육복

 

코드

#include <string>
#include <vector>
#include <map>

using namespace std;

int solution(int n, vector<int> lost, vector<int> reserve) {
    int answer = 0;
    map<int,int> student;
    for(int i=1; i<=n; i++) student[i] = 1; // 체육복 셋팅
    for(int peo:lost) student[peo]--;   // 체육복 잃어버림
    for(int reser: reserve) student[reser]++;   // 여유 체육복 가지고 있음
    
    for(auto it = student.begin(); it!=student.end(); it++){
        if(it->second==0){  // 체육복이 없을 경우
            
            int left = (it->first!=0)?(it->first)-1:-1;   // 왼쪽 사람
            int right = (it->first!=n)?(it->first)+1:-1;  // 오른쪽 사람
            
            if(student[left]>1){    // 왼쪽 사람이 여유복이 있을 경우
                student[left]--;    // 빌려줌
                it->second++;   
            }
            else if(student[right]>1){ // 오른쪽 사람이 여유복이 있을 경우
                student[right]--;   //빌려줌
                it->second++;
            }
        }
    }
    
     for(auto it = student.begin(); it!=student.end(); it++){
        if(it->second!=0) answer++;   
     }
    
    return answer;
}

 

풀이(10분)

간단한 자료구조 문제였습니다. 

1. 처음에 모든 인원의 체육복을 1개로 세팅한다.
2. 잃어버린 인원은 체육복을 감소시킨다.
3. 여유복을 가진 인원은 체육복을 증가시킨다.

본인의 체육복 개수가 0일 때 좌우 양옆을 살펴서 그 사람이 여유복이 있으면 그사람 걸 가져온다. 다만 여기서 본인이 첫 번째일 때는 왼쪽이 없게끔, 마지막일 때는 오른쪽이 없게끔 설정해주어야 한다.