[Swift] 가장 큰 수 | 프로그래머스

https://programmers.co.kr/learn/courses/30/lessons/42746

코딩테스트 연습 - 가장 큰 수

문제 설명

0 또는 양의 정수가 주어졌을 때, 정수를 이어 붙여 만들 수 있는 가장 큰 수를 알아내 주세요.

예를 들어, 주어진 정수가 [6, 10, 2]라면 [6102, 6210, 1062, 1026, 2610, 2106]를 만들 수 있고, 이중 가장 큰 수는 6210입니다.

0 또는 양의 정수가 담긴 배열 numbers가 매개변수로 주어질 때, 순서를 재배치하여 만들 수 있는 가장 큰 수를 문자열로 바꾸어 return 하도록 solution 함수를 작성해주세요.

 

제한 사항

• numbers의 길이는 1 이상 100,000 이하입니다.
• numbers의 원소는 0 이상 1,000 이하입니다.
• 정답이 너무 클 수 있으니 문자열로 바꾸어 return 합니다.

 

입출력 예

numbers	return
[6, 10, 2]	"6210"
[3, 30, 34, 5, 9]	"9534330"

 

문제풀이

주어진 정수를 문자열 비교를 해야 하므로 map함수를 활용해 문자열로 치환한다.

그 후 $0+$1과 $1+$0을 비교해 정렬한 후 문자열을 return 해준다.

입력값으로 0만 두 개 이상 주어졌을 경우 return값은 "000..." 이 된다. 이 경우에는 "0"이 되어야 하므로 예외처리를 해준다.

 

소스코드

import Foundation

func solution(_ numbers:[Int]) -> String {
    let array: [String] = numbers.map({String($0)}).sorted { $0+$1 > $1+$0 }

    return array.first == "0" ? "0" : array.joined()
}