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;
}
}
반응형