코딩 테스트 - 체육복

프로그래머스 레벨 1 테스트

#include <string>
#include <vector>
 
using namespace std;
 
int solution(int n, vector<int> lost, vector<int> reserve) 
{
    int answer = n - lost.size();
    int iSize = reserve.size();
    
    //여벌을 가진사람중에 잃어버린 사람이 있는지
    for(auto& a : reserve)
    {
        for(auto& b : lost)
        {
            if(a==b)
            {
                answer++;
                a = -10;
                b = -1;
            }
        }
    }
    
    
    for(int i = 0 ; i < iSize ; i++)
    {
        
        for(int k = 0 ; k < lost.size() ; k++)
        {
            //여벌을 가진사람이 누구에게 빌려줄수
            if(reserve[i] -1 == lost[k] || reserve[i] +1 == lost[k])
            {
                answer++;
                reserve[i] = -10;
                lost[k] = -1;
            }
        }
    }
    
    
    return answer;
}

1. 전체 인원수에서 잃어버린 사람들의 수를 먼저 계산한다.

2. 여벌을 가진 목록과 잃어버린 사람 목록에 동일한 사람이 있는지 확인

3. 남은 사람들중 빌려줄 수 있는 체육복을 가진 사람을 확인

 

for문을 분리한 이유는 n = 10, lost = [3,9,10] reserve = [3,8,9]

같은 경우를 방지하기 위해서이다.

값을 수정한 이유는 삭제로인한 벡터의 재할당보다 값 변경이 싸게 먹힐거라 생각했기 때문이다.

 -> 벡터 컨테이너의 Capcity를 할당하면 삭제로 인한 재할당은 없을 것이다.