[백준] 단계별로 풀어보기(단계:12, 집합과 맵,JAVA)10815번, 숫자 카드

문제 링크

10815번: 숫자 카드

---

주의사항

• JAVA를 사용하여 프로그램을 사용하였습니다.
• 백준에서 코드를 작성하였을 때 아래 형태에서 Main에서 결과가 출력되어야 합니다.

public class Main{ 	
	public static void main(String[] args){
    }
}

---

문제 설명

---

접근 방법

이 문제에 핵심은

1. N개의 카드들은 중복되지 않습니다.

2. N개의 카드에서 M개의 카드에서 존재하면 1, 존재하지 않으면 0을 결과로 출력한다.

 

 

이 문제에서는 N개의 숫자카드들을 배열에 저장하여 비교하면

2중 for문을 사용하면서 시간초과가 발생합니다.

 

저는 N개의 값들을 HashSet<>을 이용하여 저장하였으며

M개의 값들을 HashSet<>에서 있는지 확인하여 있으면 1, 없으면 0을 결과로 출력되도록 하였습니다.

 

문제를 해결한 알고리즘의 과정입니다.

1. N개의 숫자카드들을 HashSet<>에 저장합니다.

2. M번 숫자카드들을 받아서 HashSet<>에 존재하는지 확인합니다.

3. 존재하면 1, 존재하지 않으면 0을 결과로 출력하였습니다.

 

• BufferedReader를 사용하여 입력 값을 받았습니다.
• StringTokenizer를 이용하여 N,M개 숫자번호를 나누었습니다.
• M개의 숫자에 HashSet.contains()를 이용하여 N개의 숫자카드에 존재하는 지 확인하였습니다.
• 존재하면 1, 존재하지 않으면 0을 BufferedWriter 저장하였습니다.
• BufferedWriter에 저장된 결과값을 출력하였습니다.

import java.util.*;
import java.io.*;
public class Main{
    static int n,m;
    static int[] arr;
    public static void main(String[] args) throws IOException{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        //입력값 처리하는 BufferedReader
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        //결과값 출력하는 BufferedWriter
        n = Integer.parseInt(br.readLine());
        arr = new int[n];
        HashSet<Integer> map = new HashSet<>();
        StringTokenizer st = new StringTokenizer(br.readLine());
        //HashSet() 초기화
        for(int i=0;i<n;i++){
            int num = Integer.parseInt(st.nextToken());
            map.add(num);
        }
        m = Integer.parseInt(br.readLine());
        st = new StringTokenizer(br.readLine()," ");
        for(int i=0;i<m;i++){
            int temp = Integer.parseInt(st.nextToken());
            if(map.contains(temp))		//해당 카드 N개의 숫자에 존재할 경우
                bw.write("1 ");
            else		//존재하지 않을 경우
            	bw.write("0 ");
        }

        bw.flush();		//결과 출력
        bw.close();
        br.close();
    }
}