수 찾기

https://www.acmicpc.net/problem/1920

시간 제한	메모리 제한	제출	정답	맞힌 사람	정답 비율
1 초	128 MB	178424	53533	35489	29.831%

문제

N개의 정수 A[1], A[2], …, A[N]이 주어져 있을 때,

이 안에 X라는 정수가 존재하는지 알아내는 프로그램을 작성하시오.

입력

첫째 줄에 자연수 N(1 ≤ N ≤ 100,000)이 주어진다.

다음 줄에는 N개의 정수 A[1], A[2], …, A[N]이 주어진다. 다음 줄에는 M(1 ≤ M ≤ 100,000)이 주어진다.

다음 줄에는 M개의 수들이 주어지는데, 이 수들이 A안에 존재하는지 알아내면 된다.

모든 정수의 범위는 -2³¹ 보다 크거나 같고 2³¹보다 작다.

출력

M개의 줄에 답을 출력한다.

존재하면 1을, 존재하지 않으면 0을 출력한다.

예제 입력 1

5
4 1 5 2 3
5
1 3 7 9 5

예제 출력 1

1
1
0
0
1

출처

• 데이터를 추가한 사람: BaaaaaaaaaaarkingDog, djm03178
• 문제의 오타를 찾은 사람: bupjae
• 잘못된 데이터를 찾은 사람: tncks0121

알고리즘 분류

• 자료 구조
• 정렬
• 이분 탐색

---

문제를 정리하면 처음에 N을 입력받고 N의 수 만큼 숫자를 입력받는다.

다음으로 M을 입력받고 M의 수 만큼 숫자가 주어지는 데

그 숫자들이 전에 N의 수 만큼을 입력받을 때 존재했다면 1을 출력 아니면 0을 출력하는 문제

나는 set에 N의 수들을 넣어주고 (Set은 중복을 허용하지 않는다.)

M의 수 만큼의 숫자를 입력 받을 때 중복을 체크하여 답을 구하였다.

시간 초과된 코드

#include <iostream>
#include <set>

using namespace std;

set<int> mySet;

int N, temp;

int main()
{
	cin >> N;
	while (N-- > 0) {
		cin >> temp;
		mySet.insert(temp);
	}

	cin >> N;
	while (N-- > 0) {
		cin >> temp; // m을 입력받고 
                // 입력 받은 수가 N에 있는지 확인
		if (mySet.find(temp) != mySet.end()) { cout << "1\n"; }
		else { cout << "0\n"; }
	}

	return 0;
}

시간 초과의 원인은 내 성질이 급해서 입력을 받는 즉시 판별을 하려고 해서 그렇다.

(입력 단에서 시간초과라고 인지하는 것)

그래서 입력과 판별을 분리하여 다시 작성하였다.

통과된 코드

#include <iostream>
#include <set>
#include <list>

using namespace std;

set<int> mySet;
list<int> myList;
int N, temp;

int main()
{
	cin >> N;
	while (N-- > 0) {
		cin >> temp;
		mySet.insert(temp);
	}

	cin >> N;

	while (N-- > 0) {
		cin >> temp;
		myList.push_back(temp);
	}

	for (auto it = myList.begin(); it != myList.end(); it++) {
		if (mySet.find(*it) != mySet.end()) { cout << "1\n"; }
		else { cout << "0\n"; }
	}

	return 0;
}

다른 분들의 풀이를 보니 이진 탐색을 사용하는 문제였던 것 같다.

(set도 ‘이진탐색트리’ 라서 상관 없을 듯)