폰켓몬(찾아라 프로그래밍 마에스터)

문제 링크

코딩테스트 연습 - 폰켓몬

 

 

풀이

return 하는 값이 폰켓몬 종류의 최대값이므로 내가 몇 마리의 폰켓몬을 선택했는지는 알려주지 않아도 된다.

따라서, 폰켓몬 nums 배열을 sort한 후 같은 폰켓몬을 고르지 않으면서 num.size()/2 만큼 고르게 되거나 num.size()안에서 최대 폰켓몬 종류 개수를 return 하면된다.

 

 

 

#include <vector>
#include <algorithm>

using namespace std;

int solution(vector<int> nums)
{
    int answer = 0;
    sort(nums.begin(), nums.end());
    
    int repeat = nums[0];
    answer++;
    for(int i=1; i<nums.size(); i++)
    {
        if(answer >= nums.size()/2) { break; }
        if(repeat != nums[i])
        {
            answer++;
            repeat = nums[i];
        }
    }
    
    return answer;      // 폰켓몬 종류 개수 최대값
}