【牛客网刷题】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　　@华科平凡 解析