글 읽기 - 여러가지 테스트 케이스 다해봤는데 어디서 틀리는지 모르겠습니다.

zaza1994 6년 전 1

HashMap을 이용했습니다.

입력받으면서 HashMap을 이용해서 count하고

바로 리턴합니다. 뭐가 문제일까요

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.HashMap;
import java.util.StringTokenizer;

public class Main {

	public static void main(String[] args) throws Exception {
		
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		int N = Integer.parseInt(br.readLine());
		StringTokenizer st = new StringTokenizer(br.readLine());
		HashMap<Integer,Integer> hm =new HashMap<Integer,Integer>();
		
		//입력받은값을 hashmap에 count
		int temp;
		for(int i = 0 ; i< N ;i++) {
			temp = Integer.parseInt(st.nextToken());
			if(hm.containsKey(temp)) {
				hm.put(temp, hm.get(temp)+1);
			}else {
				hm.put(temp, 1);
			}
		}
		
		//입력받은 값을 바로 hashmap에서 찾아서 data리턴
		int zero = 0;
		int M = Integer.parseInt(br.readLine());
		st = new StringTokenizer(br.readLine());
		for(int i = 0; i<M;i++) {
			temp = Integer.parseInt(st.nextToken());
			if(hm.containsKey(temp)) {
				System.out.println(hm.get(temp));
			}else {
				System.out.println(zero);
			}
		
		}
	}

}

keith 6년 전 1

hm.put(temp, hm.get(temp)+1); 으로, 해당 갯수를 늘려가는데요..

출력할때 보면, System.out.println(hm.get(temp)); 으로, 갯수를 출력하게 프로그래밍 하셨습니다.

문제에 보면 있으면 그냥 1을 출력하면 되는거 같습니다. 따라서, System.out.println(1); 로 변경하시면 될거 같습니다.

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.HashMap;
import java.util.StringTokenizer;

public class Main {

	public static void main(String[] args) throws Exception {
		
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		int N = Integer.parseInt(br.readLine());
		StringTokenizer st = new StringTokenizer(br.readLine());
		HashMap<Integer,Integer> hm =new HashMap<Integer,Integer>();
		
		//입력받은값을 hashmap에 count
		int temp;
		for(int i = 0 ; i< N ;i++) {
			temp = Integer.parseInt(st.nextToken());
			if(hm.containsKey(temp)) {
				hm.put(temp, hm.get(temp)+1);
			}else {
				hm.put(temp, 1);
			}
		}
		
		//입력받은 값을 바로 hashmap에서 찾아서 data리턴
		int zero = 0;
		int M = Integer.parseInt(br.readLine());
		st = new StringTokenizer(br.readLine());
		for(int i = 0; i<M;i++) {
			temp = Integer.parseInt(st.nextToken());
			if(hm.containsKey(temp)) {
                // 이부분이 문제인걸로 보입니다.
				// System.out.println(hm.get(temp));
                System.out.println(1);    // 이걸로 변경
			}else {
				System.out.println(zero);
			}
		
		}
	}
}

zaza1994 6년 전 0

맞네요.

감사합니다. 중요한걸 놓치고 있었네요. 감사합니다.!

keith 6년 전 1

@zaza1994 이런류의 프로그램은 솔직히 Map 보다는 Set을 쓰는게 나을거 같네요(뭐 속도차이는 거의 없지만, 메모리 상?^^; 물론 저도, 그냥 첫 생각난게 Map이라 Map으로 코딩하긴 했지만... 아닌가??? 기억이 가물가물..)

zaza1994 6년 전 0

네 맞는거같습니다.

저는 count하기 위해서 map을 썻는데 있는지 확인하는 용도면 그게 훨씬 좋을꺼 같네요 감사합니다.

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

여러가지 테스트 케이스 다해봤는데 어디서 틀리는지 모르겠습니다.

zaza1994 6년 전 1

keith 6년 전 1

zaza1994 6년 전 0

keith 6년 전 1

zaza1994 6년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말