백준 S4 10816 : 숫자 카드 2 🅾️

[문제]

숫자 카드는 정수 하나가 적혀져 있는 카드이다. 상근이는 숫자 카드 N개를 가지고 있다.

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

10816번: 숫자 카드 2

 

[풀이]

import java.io.*;
import java.util.HashMap;
import java.util.Map;

public class Main {

	public static void main(String[] args) throws IOException {
		BufferedReader br=new BufferedReader(new InputStreamReader(System.in));
		BufferedWriter bw=new BufferedWriter(new OutputStreamWriter(System.out));
		
		br.readLine();
		String input1=br.readLine();
		String[] arr1=input1.split(" ");
		
		Map<Integer, Integer> cardMap=new HashMap<>();
		for(String s:arr1) {
			int i=Integer.parseInt(s);
			if(cardMap.containsKey(i)) cardMap.put(i, cardMap.get(i)+1);
			else cardMap.put(i, 1);
		}
		
		br.readLine();
		String input2=br.readLine();
		String[] arr2=input2.split(" ");
		br.close();
		
		StringBuilder sb=new StringBuilder();
		for(String s:arr2) {
			int i=Integer.parseInt(s);
			if(cardMap.get(i)!=null) sb.append(cardMap.get(i)+" ");
			else sb.append("0 ");
		}
		
		bw.write(sb.toString());
		bw.close();
	}

}

 

이번 문제는 중복의 개수가 point이다. 중복하면 hashmap.

map 변수를 만들어주고 중복인 경우 value+1, 중복이 아닌 경우 1을 value로 넣었다. (key는 해당 카드 번호)

그다음 타겟의 카드 번호가 존재하는 경우 그 번호의 value를, 존재하지 않는 경우 0을 stringBuilder에 append 해주었다.

bw.write에 stringBuilder를 toString한 값을 넣어주고 close해서 출력해주면 끝

 

결과는 이렇다. 다른 코드보다 살짝 빠른 편이지만 거의 880ms 정도 되는 코드도 볼 수 있었다.

시간을 줄일 수 있도록 더 효율적으로 짜봐야지..