[Java] 프로그래머스 폰켓몬

문제

https://programmers.co.kr/learn/courses/30/lessons/1845

코딩테스트 연습 - 폰켓몬

 

풀이

서로다른 가장 많은 종류의 폰켓몬을 반환하면 된다. 따라서 Set 자료구조를 이용해 중복은 무시하고 size() 메소드를 사용해서 서로다른 종류의 폰켓몬 갯수를 얻을 수 있다. 다만 처음 N/2마리라는 제한이 있으므로 만약 size 메소드에서 구한 값이 N/2보다 크면 N/2를 반환하고 그렇지 않으면 size 메소드에서 구한 값을 반환하면 된다.

 

코드

import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.Test;

import java.util.HashSet;

public class Solution {
    public int solution(int[] nums) {
        int answer = 0;
        int allowedNum = nums.length/2; // 허용된 최대 값

        HashSet<Integer> set = new HashSet<>();

        for (int x : nums) {    // Set에 데이터 추가
            set.add(x); 
        }
        int max = set.size();   // 서로 다른 종류의 포켓몬 수

        if (max > allowedNum)   // 허용된 수보다 더 많다면 결국 허용된 수가 최대값
            max = allowedNum;

        answer = max;
        return answer;
    }

    @Test
    void test() {
        Assertions.assertEquals(2,solution(new int[]{3,1,2,3}));
        Assertions.assertEquals(3,solution(new int[]{3,3,3,2,2,4}));
        Assertions.assertEquals(2,solution(new int[]{3,3,3,2,2,2}));
    }
}