[백준 1081] 숫자 카드

🔗문제 풀러가기
단계별로 풀어보기 단계 14의 1번째 문제이다.

문제 분석

Set 컨네이너를 이용해 문제를 해결하였다.

코드

#include <iostream>
#include <set>
using namespace std;

int main()
{
	ios_base::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);

	int n, m;
	set<int> _set;

	cin >> n;
	for (int i = 0; i < n; i++)
	{
		int input;
		cin >> input;
		
		_set.insert(input);
	}

	cin >> m;
	for (int i = 0; i < m; i++)
	{
		int input;
		cin >> input;

		if (_set.find(input) != _set.end())
		{
			cout << "1 ";
		}
		else
		{
			cout << "0 ";
		}
	}
}

해석

1. 수를 입력받아 set에 저장한다.
2. 수를 다시 입력받으며 전에 입력받은 수를 모은 set에 현재 입력받은 수가 있는지 확인하여 있다면 1을, 없다면 0을 출력하면 끝!

   _set.find(input)을 하였을 때 _set에 input이 없다면
   _set.end()가 반환된다.