1. 程式人生 > >【牛客網刷題】Python3之functools.cmp_to_key()學習

【牛客網刷題】Python3之functools.cmp_to_key()學習

牛客網原題:

連結:https://www.nowcoder.com/questionTerminal/a6a656249f404eb498d16b2f8eaa2c60
來源:牛客網

設有n個正整數,將他們連線成一排,組成一個最大的多位整數。
如:n=3時,3個整數13,312,343,連成的最大整數為34331213。
如:n=4時,4個整數7,13,4,246連線成的最大整數為7424613。

輸入描述:
有多組測試樣例,每組測試樣例包含兩行,第一行為一個整數N(N<=100),第二行包含N個數(每個數不超過1000,空格分開)。

輸出描述:

每組資料輸出一個表示最大的整數。

from functools import cmp_to_key
import sys

def concat_the_largest_number(a_list):
    if len(a_list) == 0:
        return 0

    str_list = [str(x) for x in a_list]
    sorted_str_list = sorted(str_list, key=cmp_to_key(mycmp), reverse=True)

    concat_str = ''
    for x in sorted_str_list:
        concat_str = concat_str + x

    return int(concat_str)


def mycmp(str_x, str_y):
    if str_x + str_y > str_y + str_x:
        return 1
    elif str_x + str_y == str_y + str_x:
        return 0
    else:
        return -1


if __name__ == '__main__':
    # list1 = [936, 93, 93]
    # print(concat_the_largest_number(list1))
    # list2 = [536, 53, 53, 9]
    # print(concat_the_largest_number(list2))

    for couter, value in enumerate(sys.stdin):
        if couter % 2 == 1:
            str_list = value.split(' ')
            integer_list = [int(x) for x in str_list]
            print(concat_the_largest_number(integer_list))

  

參考:

1、https://www.nowcoder.com/profile/737855725/codeBookDetail?submissionId=37009983  @華科平凡 解析