[프로그래머스/C++] 숫자 짝꿍

문제 정보

제목 : 숫자 짝꿍

난이도 : Lv.1

사용 언어 : C++

문제 링크

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

 

문제 설명

두 정수 X, Y의 임의의 자리에서 공통으로 나타나는 정수 k(0 ≤ k ≤ 9)들을 이용하여 만들 수 있는 가장 큰 정수를 두 수의 짝꿍이라 합니다(단, 공통으로 나타나는 정수 중 서로 짝지을 수 있는 숫자만 사용합니다). X, Y의 짝꿍이 존재하지 않으면, 짝꿍은 -1입니다. X, Y의 짝꿍이 0으로만 구성되어 있다면, 짝꿍은 0입니다.

예를 들어, X = 3403이고 Y = 13203이라면, X와 Y의 짝꿍은 X와 Y에서 공통으로 나타나는 3, 0, 3으로 만들 수 있는 가장 큰 정수인 330입니다. 다른 예시로 X = 5525이고 Y = 1255이면 X와 Y의 짝꿍은 X와 Y에서 공통으로 나타나는 2, 5, 5로 만들 수 있는 가장 큰 정수인 552입니다(X에는 5가 3개, Y에는 5가 2개 나타나므로 남는 5 한 개는 짝 지을 수 없습니다.)
두 정수 X, Y가 주어졌을 때, X, Y의 짝꿍을 return하는 solution 함수를 완성해주세요.

 

제한 사항

• 3 ≤ X, Y의 길이(자릿수) ≤ 3,000,000입니다.
• X, Y는 0으로 시작하지 않습니다.
• X, Y의 짝꿍은 상당히 큰 정수일 수 있으므로, 문자열로 반환합니다.

입출력 예

X	Y	result
"100"	"2345"	"-1"
"100"	"203045"	"0"
"100"	"123450"	"10"
"12321"	"42531"	"321"
"5525"	"1255"	"552"

입출력 예 설명

입출력 예 #1

• X, Y의 짝꿍은 존재하지 않습니다. 따라서 "-1"을 return합니다.

입출력 예 #2

• X, Y의 공통된 숫자는 0으로만 구성되어 있기 때문에, 두 수의 짝꿍은 정수 0입니다. 따라서 "0"을 return합니다.

입출력 예 #3

• X, Y의 짝꿍은 10이므로, "10"을 return합니다.

입출력 예 #4

• X, Y의 짝꿍은 321입니다. 따라서 "321"을 return합니다.

입출력 예 #5

• 지문에 설명된 예시와 같습니다.

---

나의 풀이

처음엔 중첩 반복문으로 문제를 해결하려했으나 시간 초과라는 큰 문제에 마주하게 되었습니다. 해결하기 위해 다른 방식을 알아보다 배열을 통해 0부터 9까지 카운트를 세고 두 배열을 비교하면 비교적 효율적으로 짤 수 있다는 것을 알게 되었습니다. X, Y 각각 길이만큼 반복하며 count 배열에 해당하는 숫자의 값을 +1씩 증가 시킵니다. 구하게 되면 큰 수부터 i--;하면서 x_count[i]와 y_count[i]의 최소값을 구합니다. 최소값이 1이상이면 반복하여 answer에 추가합니다. answer문자열을 리턴하는데, 문자열이 비어있다면 "-1"을 리턴하고, answer에 "0000"이런식으로 0이 여러개라면 "0"으로 리턴할 수 있도록 했습니다.

코드

#include <string>
#include <vector>
#include <algorithm>

using namespace std;

string solution(string X, string Y) {
    string answer = "";
    int x_count[10]={0};
    int y_count[10]={0};
    int tmp;
    
    for (int i = 0; i < X.size(); i++)
    {
        tmp = X[i]-'0';
        x_count[ X[i]-'0' ]++;
    }
    
    for (int i = 0; i < Y.size(); i++)
    {
        tmp = Y[i]-'0';
        y_count[tmp]++;
    }
    
    for(int i = 9; i >=0;i--)
    {
        int n = min( x_count[i], y_count[i]);
        for(int j=0; j<n; j++)
            answer+=to_string(i);
    }
    
    if ( answer == "")
        answer = "-1";
    else if (answer[0]=='0')
        answer = "0";

    return answer;
}