[c++]최빈값 구하기

출처:프로그래머스

문제 설명

최빈값은 주어진 값 중에서 가장 자주 나오는 값을 의미합니다. 정수 배열 array가 매개변수로 주어질 때, 최빈값을 return 하도록 solution 함수를 완성해보세요. 최빈값이 여러 개면 -1을 return 합니다.

 

제한사항

• 0 < array의 길이 < 100
• 0 ≤ array의 원소 < 1000

 

입출력 예

array	result
[1, 2, 3, 3, 3, 4]	3
[1, 1, 2, 2]	-1
[1]	1

입출력 예 설명

입출력 예 #1

• [1, 2, 3, 3, 3, 4]에서 1은 1개 2는 1개 3은 3개 4는 1개로 최빈값은 3입니다.

입출력 예 #2

• [1, 1, 2, 2]에서 1은 2개 2는 2개로 최빈값이 1, 2입니다. 최빈값이 여러 개이므로 -1을 return 합니다.

입출력 예 #3

• [1]에는 1만 있으므로 최빈값은 1입니다.

문제풀이

#include <vector>
#include <algorithm>

using namespace std;

int solution(vector<int> array) {
	//배열이 1개면 그 값을 반환
    if (array.size() == 1) {
        return array[0];
    }
    sort(array.begin(), array.end());
	
    int answer = array[0], mostCount = 1, count = 1;
    //최빈 값 여러개인지 체크
    bool duple = false; 

    for (int i = 1; i < array.size(); i++) {
        if (array[i] == array[i - 1]) {
            count++;
        } else {
        	//값이 같지 않다면 넘어가면서 최빈값을 업데이트
            if (count > mostCount) {
                mostCount = count;
                answer = array[i - 1];
                duple = false;
            } else if (count == mostCount) {
                duple = true;
            }
            count = 1;
        }
    }

    // 마지막으로 최빈 값이 여러개인지 체크
    if (count > mostCount) {
        answer = array.back();
        duple = false;
    } else if (count == mostCount) {
        duple = true;
    }

    return duple ? -1 : answer;
}

 

처음에 마지막 최빈 값 여러개인지 체크하는 처리가 없어서 제출하고 통과를 하지 못하였다. 반복문이 돌면서 최빈값이 존재할 때와 존재하지 않고 현재 갯수 값과 같을 때를 boolean 형식으로 체크하여 리턴 값을 바꿀 수 있게 했다. 레벨 1 문제라 간단하게 봤는데 엄청 간단하지만은 않은 문제였다. 또한 이 문제를 통해 새로 배운 것들도 있어서 가끔 레벨 1를 풀어보는 것도 나쁘지 않은 것 같다.