프로그래머스[Java] - (Level2)폰켓몬

https://programmers.co.kr/learn/courses/30/lessons/1845?language=java

프로그래머스

코드

import java.util.*;

class Solution {
    	public int solution(int[] nums) {
		int answer = 0;
		List<Integer> list = new ArrayList<Integer>();
		list.add(nums[0]);
        
		// nums[] 배열의 중복제거
		for(int i=1; i<nums.length; i++) 
			if(!list.contains(nums[i]))
				list.add(nums[i]);
        
		/* list.size(): 포켓몬의 종류, nums.length: 가질 수 있는 포켓몬의 수 */
		answer = (list.size() < nums.length/2) ? list.size() : nums.length/2;
		return answer;
	}
}

풀이

문제는 길지만 읽어보니 단순했다.

주어진 포켓몬의 배열 nums 에서 중복을 제거하면 될 것 같았고, 중복을 제거한 후 남은 폰켓몬의 종류 수를 비교해주면 될 것 같았다.

 

nums가 다음과 같이 주어진다고 생각해보자. 가질 수 있는 폰켓몬의 수는 3마리이다.

1) nums = [1, 1, 1, 1, 2, 2]

2) nums = [1, 1, 1, 1, 1, 1]

3) nums = [1, 2, 3, 4, 5, 6]

4) nums = [1, 1, 2, 2, 3, 3]

 

1)의 경우, 중복을 제거하고나면 [1,2] 가 남는다. 최대한의 종류를 가져야 하므로 중복을 제거한 후 모두 가져도 2마리이다. 따라서 가질 수 있는 종류는 두가지

 

2)의 경우, 중복을 제거하고 나면 [1] 이 남는다. -> 가질 수 있는 종류는 한가지

 

3)의 경우, 중복을 제거하고 나면 [1, 2, 3, 4, 5, 6] 이 남는다. -> 가질 수 있는 폰켓몬의 수는 3마리이므로, 종류는 세가지

 

4)의 경우, 중복을 제거하고 나면 [1, 2, 3] 이 남는다. -> 가질 수 있는 폰켓몬의 수와 종류가 동일하게 세가지 이다.

 

따라서 중복을 제거하고 남은 배열의 길이를 기준으로 비교하기 위해 ArrayList로 선언했고, 

 

만약 남은 폰켓몬의 사이즈가 가질 수 있는 폰켓몬의 수보다 작으면 -> (1, 2의 경우) 배열의 길이가 정답이고,

그 반대이면 -> (3, 4의 경우) 가질 수 있는 폰켓몬의 수가 정답이다.