[Level1/c++] 체육복

문제

코딩테스트 연습 - 체육복 | 프로그래머스

코드

	#include <string>
	#include <vector>
	#include <algorithm>
	#include <iostream>
	using namespace std;
	int solution(int n, vector<int> lost, vector<int> reserve) {
	int answer = n;
	int minus = lost.size();
	for (int i = 0; i < lost.size(); i++)
	{
	for (int j = 0; j < reserve.size(); j++)
	{
	if (lost[i] == reserve[j])
	{
	lost.erase(lost.begin() + i);
	reserve.erase(reserve.begin() + j);
	i--, j--;
	minus--;
	}
	}
	}
	for (int i = 0; i < lost.size(); i++)
	{
	for (int j = 0; j < reserve.size(); j++)
	{
	if (lost[i] - 1 == reserve[j] || lost[i] + 1 == reserve[j])
	{
	minus--;
	reserve.erase(reserve.begin() + j);
	break;
	}
	}
	}
	return answer - minus;;
	}

view raw 프로그래머스_체육복.cpp hosted with ❤ by GitHub