[프로그래머스] 폰켓몬

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

프로그래머스

 

LV 1. 답게 메소드에 익숙하면 빠르게 풀 수 있는 문제이다.

만약 종류의 수가 N / 2보다 작으면 조합의 최대 크기는 종류의 수 그 자체이고, 그렇지 않다면 최대로 고를 수 있는 수인 N / 2가 된다는 아이디어만 살리면 된다.

 

#include <vector>
#include <algorithm>
using namespace std;

int solution(vector<int> nums)
{
    int answer = 0;
    int sz = nums.size();
    
    sort(nums.begin(), nums.end());
    nums.erase(unique(nums.begin(), nums.end()), nums.end());

    if(nums.size() < sz / 2) {
        answer = nums.size();
    }
    else {
        answer = sz / 2;
    }
    return answer;
}