[정렬] 프로그래머스 가장 큰 수 - 파이썬

김아현·2023년 5월 5일
0

코딩테스트

목록 보기
8/26
post-thumbnail
post-custom-banner

가장 큰 수 파이썬 풀이

문제설명

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"

접근 방법

처음엔 주어진 수를 조합해서 가장 큰 수를 만드는 방법이 뭐가 있을까? 하다가, 파이썬의 permutation을 써서 조합 중에 가장 큰 수를 max()로 찾아내려 했다.

1차 작성 코드

from itertools import permutations

def solution(numbers):
    # permutations 결과를 비교해서 최대값 연산 
    ans = max(permutations(numbers, len(numbers)), key=lambda x: int(''.join(map(str, x))))

    # tuple into string
    return ''.join(map(str,ans))

1차 작성 코드의 문제점

  • 테스트 케이스 1~11번까지 모두 시간초과!
  • 단순 조합 및 최대값 비교가 아니라 로직을 생각해야 한다.

조합을 시간에 관계없이 풀려했더니 야무지게 시간 초과났다. 노가다로 모든 조합을 비교하는게 아니라, 로직을 최적화하는 방법이 있어야할 텐데 도무지 혼자 생각하기 힘들어서 웹 검색으로 참고해 수정해보기로 했다.

1차 코드 수정하기

TIL

  • 시간 제한을 고려해 풀어보자. 어케 하지
profile
멘티를 넘어 멘토가 되는 그날까지 파이팅
post-custom-banner

0개의 댓글