Hodustory/프로그래밍&DB

프로그래머스 : 폰켓몬(JAVA)

호두밥 2021. 7. 22. 18:05

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

 

코딩테스트 연습 - 폰켓몬

당신은 폰켓몬을 잡기 위한 오랜 여행 끝에, 홍 박사님의 연구실에 도착했습니다. 홍 박사님은 당신에게 자신의 연구실에 있는 총 N 마리의 폰켓몬 중에서 N/2마리를 가져가도 좋다고 했습니다.

programmers.co.kr

import java.util.HashSet;

class Solution {
    public int solution(int[] nums) {
        int answer = 0;
        int pick = nums.length / 2; // 뽑아야 하는 포켓몬수
	    
        //중복값을 허용하지 않는 set을 이용
	    HashSet<Integer> hashSet = new HashSet<>();
	    
	    for(int n : nums) {
	    	hashSet.add(n);
	    }
	    
        //set의 크기(포켓몬 종류)가 pick보다 작으면 set의 크기를 반환
	    if(hashSet.size() < pick) {
	    	answer = hashSet.size();
	    }else {
        	//set의 크기(포켓몬 종류)가 pick보다 크거나 같으면 pick을 반환
	    	answer = pick;
	    }
        
        
        return answer;
    }
}
반응형