[정렬] 가장 큰 수

yejichoi·2023년 5월 11일
0

알고리즘 스터디

목록 보기
51/153
post-thumbnail

가장 큰 수

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):    
#     s = list(map(str,numbers))
#     a = sorted(s,key=lambda x: x*3,reverse=1)
#     return str(int("".join(a)))
import functools

def comparator(a,b):
    #comparator 함수는 두 개의 문자열 a와 b를 비교
    #이 두 문자열을 합한 두 가지 경우의 수를 비교하여 더 큰 수를 리턴
    t1 = a+b
    t2 = b+a
    return (int(t1) > int(t2)) - (int(t1) < int(t2)) 
    #t1이 크다면 1  // t2가 크다면 -1  //  같으면 0
    #비교 결과가 양수(1)이면, 두 개의 원소의 순서를 바꿉니다.
    #비교 결과가 0이면, 두 개의 원소를 서로 바꾸지 않습니다.
    #비교 결과가 음수(-1)이면, 두 개의 원소의 순서를 바꾸지 않습니다.

def solution(numbers):
    n = [str(x) for x in numbers]
    n = sorted(n, key=functools.cmp_to_key(comparator),reverse=True)
    # 가장 큰 수를 만들기 위해 내림차순 정렬 reverse = True
    answer = str(int(''.join(n)))
    return answer

functools

high-order-function(고차원 함수)를 위해 고안된 것으로 다른 함수에 적용하거나 다른 함수를 반환하는 모듈

  • lru_cached
  • cmp_to_key
  • total_ordering
  • partial
  • reduce

functools.cmp_to_key

sorted와 같은 정렬 함수의 key에 함수로 정렬을 하기 위해서 사용
함수는 두 개의 인수를 받아야 하고, 그들을 비교해서 작다면 음수,같으면 0, 크다면 양수를 반환

>>> import functools

# 비교 함수: 첫 번째 요소가 두 번째 요소보다 작으면 음수, 같으면 0, 크면 양수를 반환합니다.
>>> def compare_func(a, b):
...     if a < b:
...         return -1
...     elif a == b:
...         return 0
...     else:
...         return 1

# 비교 함수를 키 함수로 변환합니다.
>>> key_func = functools.cmp_to_key(compare_func)

# 키 함수를 사용하여 정렬합니다.
>>> numbers = [5, 2, 4, 1, 3]
>>> sorted_numbers = sorted(numbers, key=key_func)
>>> print(sorted_numbers)  # 결과: [1, 2, 3, 4, 5]
def solution(numbers):
    numbers = sorted(map(str,numbers), 
    key = lambda x: x*10, reverse = True
    #map(str, nums) : 리스트 nums의 모든 요소들을 문자열로 변환
    #key=lambda x: x*10는 각 문자열을 10번 반복해서 길이를 늘린 후 
    #비교하도록 설정
    return str(int(''.join(numbers)))

TIL

🍁 ''.join(list)

문자열을 구성하는 요소들을 하나의 문자열로 연결해주는 함수
특히 리스트에 있는 문자열들을 하나의 문자열로 합치는데 자주 사용

0개의 댓글