[ Programmers / CodingTest / Python ] 가장 큰 수

문제 설명

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"

접근 방법

처음에는 간단하게 생각하여 sort()함수에 lambda를 사용하여 정렬해 해결하려고 하였다.(문자열의 정렬은 '10'보다 '6'이 큼). 그러나 테스트 케이스 1~6이 모두 오답을 출력하였고 고민하던 중에 질문하기에서 힌트를 얻었다.

numbers의 원소 범위는 0부터 1000까지이다. 그러므로 최대 4자리의 수까지 입력될 수 있다. 원소들을 비교하기 위해서는 원소들을 4자리로 만들어야 한다.
예를 들어 1과 10이 있을 때, 문자열 정렬을 하게 되면 1이 10보다 작다고 인식되어진다. 이를 1111, 1010으로 비교한다면 1이 더 크게 인식되어진다. 정답 문자열에 초기의 원소를 합치기 위해서 초기 원소의 길이도 같이 저장한다. 결과 문자열에 더할 때에는 슬라이싱을 이용하여 초기 원소의 길이만큼만 더하도록 한다.

  • 정답을 저장하는 문자열 answer를 선언한다.
  • 4자리로 만든 원소와 초기 원소의 길이를 저장하기 위한 배열 tmp를 선언한다.
  • numbers를 순회하는 i에 대한 for문을 돌린다.
    -> 임시 변수 num을 선언하고 i를 문자열 처리하여 길이를 4로 만들어 저장한다.
    -> 임시 변수 length를 선언하고 i를 문자열 처리하여 i의 초기 길이를 저장한다.
    -> tmp에 [num, i]를 넣는다.
  • tmp를 오름차순 정렬한다.
  • tmp가 있을 동안 반복하는 while문을 돌린다.
    -> 임시 변수 p1을 선언하고 tmp.pop()을 대입한다.
    -> 임시 변수 p2를 선언하고 p1[1]을 대입한다. (원소의 초기 길이)
    -> answer에 p1[0][:p2]를 더한다. (초기 원소만 더하는 과정)
  • 만약 answer가 0으로만 구성되어 있다면 answer를 '0'으로 갱신한다.
  • answer를 반환한다.

solution.py

def solution(numbers):
    answer = ''
    tmp=[]
    for i in numbers:
        num=(str(i)*4)[:4]
        length=len(str(i))
        tmp.append([num, length])
    tmp.sort()
    while tmp:
        p1=tmp.pop()
        p2=p1[1]
        answer+=p1[0][:p2]
    if answer=='0'*len(answer):
        answer='0'
    return answer

좋은 웹페이지 즐겨찾기