[프로그래머스 lv1: 체육복] - C++

programmers.co.kr/learn/courses/30/lessons/42862#

코딩테스트 연습 - 체육복

#include <string>
#include <vector>

using namespace std;

int solution(int n, vector<int> lost, vector<int> reserve) {

	//n 전체수 
	// lost 도난당한 번호 
	//  reserve 여별 보휴
	int answer = 0;

	answer = n - lost.size(); //안잃어버린 수


	//도난 당했으나 여유복도 있는 애들
	for (int i = 0; i < lost.size(); i++)
	{
		for (int j = 0; j < reserve.size(); j++)
		{
			if (reserve[j] == lost[i])
			{
				reserve[j] = -10; //쓰레기값 넣어주기
				lost[i] = -10; //쓰레기값 넣어주기
				answer++;
				break;
			}
		}
	}

	//앞뒤 사이즈 확인
	for (int i = 0; i < lost.size(); i++)
	{
		for (int j = 0; j < reserve.size(); j++)
		{
			if (reserve[j] == lost[i] - 1 || reserve[j] == lost[i] + 1)
			{
				answer++;
				reserve[j] = -10; //쓰레기값 넣어주기
				break;
			}	
		}
	}
	return answer;
}