[Programmers] Lv.1 숫자 짝꿍 (Go)

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의 짝꿍은 상당히 큰 정수일 수 있으므로, 문자열로 반환합니다.

---

Solution

import (
	"strconv"
	"strings"
)

func solution(X string, Y string) string {

	x := getPairSlice(X)
	y := getPairSlice(Y)

	var sl []string

	for i := 9; i >= 0; i-- {
		idx := strconv.Itoa(i)

		for x[i] > 0 && y[i] > 0 {
			sl = append(sl, idx)
			x[i]--
			y[i]--
		}
	}

	if len(sl) == 0 {
		return "-1"
	}

	if sl[0] == "0" {
		return "0"
	}

	return strings.Join(sl, "")
}

func getPairSlice(num string) []int {

	var result = make([]int, 10)

	for _, s := range num {
		n, _ := strconv.Atoi(string(s))
		result[n]++
	}

	return result
}