<백준> 10815번 : 숫자 카드

<링크>

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

 

<문제>

숫자 카드는 정수 하나가 적혀져 있는 카드이다. 상근이는 숫자 카드 N개를 가지고 있다. 정수 M개가 주어졌을 때, 이 수가 적혀있는 숫자 카드를 상근이가 가지고 있는지 아닌지를 구하는 프로그램을 작성하시오.

<입력 >

첫째 줄에 상근이가 가지고 있는 숫자 카드의 개수 N(1 ≤ N ≤ 500,000)이 주어진다. 둘째 줄에는 숫자 카드에 적혀있는 정수가 주어진다. 숫자 카드에 적혀있는 수는 -10,000,000보다 크거나 같고, 10,000,000보다 작거나 같다. 두 숫자 카드에 같은 수가 적혀있는 경우는 없다.

셋째 줄에는 M(1 ≤ M ≤ 500,000)이 주어진다. 넷째 줄에는 상근이가 가지고 있는 숫자 카드인지 아닌지를 구해야 할 M개의 정수가 주어지며, 이 수는 공백으로 구분되어져 있다. 이 수도 -10,000,000보다 크거나 같고, 10,000,000보다 작거나 같다

<출력>

 

첫째 줄에 입력으로 주어진 M개의 수에 대해서, 각 수가 적힌 숫자 카드를 상근이가 가지고 있으면 1을, 아니면 0을 공백으로 구분해 출력한다.

 

 

<고려사항>

1. 단위가 크다.. 단위가 크다...

<풀이 - Java>

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.HashSet;
import java.util.StringTokenizer;

public class Main {

	public static void main(String[] args) throws Exception {
		// 데이터 입력
		BufferedReader in = new BufferedReader(new InputStreamReader(System.in));

		// 상근이가 가지고 있는 숫자 카드의 개수 N
		int N = Integer.parseInt(in.readLine());
		// 담아주세요 헷
		HashSet<Integer> cardSet = new HashSet<>();
		StringTokenizer st = new StringTokenizer(in.readLine(), " ");
		
		for (int i = 0; i < N; i++) {
			cardSet.add(Integer.parseInt(st.nextToken()));
		}

		// M
		int M = Integer.parseInt(in.readLine());
		StringBuilder sb = new StringBuilder();
		st = new StringTokenizer(in.readLine(), " ");

		for (int i = 0; i < M; i++) {
			int card =Integer.parseInt(st.nextToken());
			
			if (cardSet.contains(card)) {
				sb.append("1 ");
			}else {
				sb.append("0 ");
			}
		}
		System.out.println(sb.toString().trim());

	}

}

 

- 본 문제의 경우,  이중 for문으로 풀경우 시간초과가 난다..

- 그래서 구글링 한결과, hashSet이라는 좋은 자료구조가 있어서 활용해보았다.

-  마지막에 출력할때, trim()을 써주었는데, 이 방식은 마지막 출력값의 공백 ( " " )을 삭제해주는 기능이 있다.

- trim을 쓰지 않을경우, "1 "로 출력이 되기 때문에 오답으로 나온다 ㅠㅠ....

- 이중 포문 되게 해주세요 ㅠㅠㅠㅠㅠㅠ