[백준/JAVA]10807번 개수 세기

문제

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

정수 배열에서 특정 숫자가 몇 번 나오는지 세는 문제였다.

문제 풀이

1. 내가 생각한 풀이

import java.util.Scanner;
public class Main{
	public static void main(String args[]){
        Scanner scanner = new Scanner(System.in);
        int N = scanner.nextInt();
        int[] array=new int[N];
        for(int i=0;i<N;i++){
            array[i]=scanner.nextInt();    
        }
        
        int v = scanner.nextInt();
        int count=0;
        for(int i=0;i<N;i++){
            if(array[i]==v){
                count++;
            }
        }
        System.out.println(count);
	}
}

내가 처음 생각한 방법은 배열을 두 번 순회하는 풀이인데 이 방법은 직관적이지만, 만약 배열의 크기가 매우 크고 여러 번의 쿼리를 처리해야 한다면, 느리기 때문에 해시맵을 사용하는 방법이 더 빠르다.

2. HashMap(해시맵)을 사용한 풀이

import java.util.HashMap;
import java.util.Map;
import java.util.Scanner;

public class Main {
    public static void main(String args[]) {
        Scanner scanner = new Scanner(System.in);
        int N = scanner.nextInt();
        int[] array = new int[N];
   
        HashMap<Integer, Integer> count = new HashMap<>();
        
        for (int i = 0; i < N; i++) {
            array[i] = scanner.nextInt();
            //count.getOrDefault(array[i], 0): array[i]가 HashMap에 있으면 그 값을 반환, 없으면 0을 반환
            count.put(array[i], count.getOrDefault(array[i], 0) + 1);
        }
        
        //출현 횟수를 확인하고자 하는 특정 숫자 v 입력받기
        int v = scanner.nextInt();
        
        //HashMap에서 v의 출현 횟수를 찾아 출력, 만약 v가 HashMap에 없다면, 기본값으로 0을 반환
        System.out.println(count.getOrDefault(v, 0));
    }
}

해시맵 사용: HashMap을 사용하여 각 숫자의 개수를 저장한다. put 메서드를 사용하여 숫자를 키로 하고, 그 숫자의 개수를 값으로 저장한다.
getOrDefault: 특정 숫자의 개수를 조회할 때 getOrDefault 메서드를 사용하여, 해당 숫자가 없을 경우 0을 반환하도록 한다.

https://yje44428.tistory.com/19

[JAVA]List, Set, Map 컬렉션(Collection)