Algorithm/Programmers
[프로그래머스] 폰켓몬 : C++
감성적인 개발자
2021. 8. 11. 22:39
https://programmers.co.kr/learn/courses/30/lessons/1845
코딩테스트 연습 - 폰켓몬
당신은 폰켓몬을 잡기 위한 오랜 여행 끝에, 홍 박사님의 연구실에 도착했습니다. 홍 박사님은 당신에게 자신의 연구실에 있는 총 N 마리의 폰켓몬 중에서 N/2마리를 가져가도 좋다고 했습니다.
programmers.co.kr
#include <vector>
#include <algorithm>
using namespace std;
int findSameNum(vector<int> v){
int result = v.size();
for(int i = 0; i < v.size()-1; i++){
if(v[i] == v[i+1]) result--;
}
return result;
}
int solution(vector<int> nums){
int answer = nums.size()/2;
sort(nums.begin(),nums.end());
nums.resize( unique(nums.begin(), nums.end()) - nums.begin() );
if(nums.size() > answer) return answer;
else return nums.size();
}
최대로 고를 수 있는 종류를 길이/2로 미리 구해놓는다.
요소 정렬 후 unique를 통해 중복된 요소를 제거 + resize한다.
만약 resize했을 때 고를 수 있는 포켓몬의 종류가 더 작아졌다면, 최대로 고를 수 있는 포켓몬의 수는 resize.size()이다.