[ 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
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
Author And Source
이 문제에 관하여([ Programmers / CodingTest / Python ] 가장 큰 수), 우리는 이곳에서 더 많은 자료를 발견하고 링크를 클릭하여 보았다 https://velog.io/@xx0hn/Programmers-CodingTest-Python-가장-큰-수저자 귀속: 원작자 정보가 원작자 URL에 포함되어 있으며 저작권은 원작자 소유입니다.
우수한 개발자 콘텐츠 발견에 전념 (Collection and Share based on the CC Protocol.)