【牛客網刷題】Python3之functools.cmp_to_key()學習
阿新 • • 發佈:2018-11-07
牛客網原題:
連結: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 @華科平凡 解析