[코딩테스트] 체육복

문제

---

풀이

#include <string>
#include <vector>
#include <algorithm>

using namespace std;

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

	sort(lost.begin(), lost.end());
	sort(reserve.begin(), reserve.end());

	for (int i = 0; i < lost.size(); i++)
	{
		for (int j = 0; j < reserve.size(); j++)
		{
			if (lost[i] == reserve[j])
			{
				answer++;
				lost[i] = -1;
				reserve[j] = -1;
				break;
			}
		}
	}

	for (auto lostNumber = lost.begin(); lostNumber != lost.end(); lostNumber++)
	{
		bool check = false;

		for (auto reserveNumber = reserve.begin(); reserveNumber != reserve.end(); reserveNumber++)
		{
			if (lostNumber[0] == reserveNumber[0] + 1 || lostNumber[0] == reserveNumber[0] - 1)
			{
				check = true;
				reserve.erase(reserveNumber);
				break;
			}
		}

		if (!check)
		{
			answer--;
		}
	}

	return answer;
}