가장 큰 수

노누리·2022년 4월 28일
0

문제

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

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

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

제한 사항

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

입출력 예

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

문제 풀이

기존 풀이

def solution(numbers):
    answer=''
    if sum(numbers)==0:
        return '0'
    num=sorted(list(map(str,numbers)),key=lambda x: x*3)
    num.reverse()
    return "".join(num)
  • 문자형 숫자를 정렬하면 숫자의 크기가 아닌 숫자의 순서로 정렬한다
    아스키코드로 비교
  • 숫자를 한자리씩 비교해서 큰 숫자가 앞으로 오게 만들면 가장 큰 수가 조합된다.

시간 초과

from itertools import permutations
def solution(numbers):
    per=list(permutations(list(map(str,numbers)),len(numbers)))
    return max([''.join(per[i]) for i in range(len(per))])

성공한 풀이

def solution(numbers):
    if sum(numbers)==0:
        return '0'
    return "".join(sorted(list(map(str,numbers)),key=lambda x: x*3,reverse=True))
  • 이전에 작성한 코드가 이해가 안돼서 해석하다가 결국 코드를 보고했다
  • x*3을 한 이유는 330보다 큰 숫자로 인식되어야 하는데 작은 숫자로 인식되어서 333 303030을 하여 두번째 자리를 비교하면 3이 더 크기 때문에 비교를 위해서 넣었다
  • 굳이 x*2가 아닌 x*3을 한 이유는 테스트 케이스에서 실패가 떴기 때문
    큰 숫자가 있을 때 x*3으로 비교해야 확실해짐
  • 만약 모든 원소가 0이면 0000은 없으니 0으로 반환해야한다
profile
백엔드 개발자입니다.

0개의 댓글