[프로그래머스 고득점 Kit/Java] 폰켓몬

 

다음과 같이 풀었다.

nums를 돌며 map에 종류 번호와 해당 번호가 나온 횟수를 저장한다.

1) 종류 번호가 nums 배열 크기 / 2 보다 크다면 -> nums 배열 크기 / 2만큼만 선택할 수 있다.

2) 작거나 같다면 -> 종류 번호 만큼 선택한다.

 

import java.util.*;
import java.io.*;
class Solution {
    public int solution(int[] nums) {
        int answer = 0;
        HashMap<Integer, Integer> pokemon = new HashMap<>();
        for (int num : nums) {
            pokemon.put(num, pokemon.getOrDefault(num, 0) + 1);
        }
        if (pokemon.size() > nums.length / 2) {
            answer = nums.length / 2;
        } else {
            answer = pokemon.size();
        }
        return answer;
    }
}