Python的cmp_to_key()函数详解【部分翻译＋leetcode题目分析】

WageningenUR

已于 2022-02-24 19:27:51 修改

阅读量2.6k

点赞数 3

分类专栏：办公自动化文章标签： python

于 2022-02-24 19:26:11 首次发布

本文链接：https://blog.youkuaiyun.com/Father_of_Python/article/details/123118854

版权

办公自动化专栏收录该内容

9 篇文章

订阅专栏

本文介绍了Python内置functools模块的cmp_to_key函数，如何通过自定义函数实现列表排序和查找最大最小值。通过mycmp函数实例演示了如何对比元素并作为键进行操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

定义

cmp_to_key() 使用键来比较元素
它内置在 functools 模块中，因此必须先导入 functools 才能使用该功能
与接受诸如 min()、max()、sorted() 等关键函数的工具一起使用。
只接受一个严格来说应该是可调用的参数
此函数返回可用于比较元素的特殊键
用于比较两个值并返回 1、-1 或 0

示例 1：使用 cmp_to_key() 函数提供的键对列表进行排序的程序

import functools


def mycmp(a, b):
	print("comparing ", a, " and ", b)
	if a > b:
		return 1
	elif a < b:
		return -1
	else:
		return 0


print(sorted([1, 2, 4, 2], key=functools.cmp_to_key(mycmp)))

输出：

comparing  2  and  1
comparing  4  and  2
comparing  2  and  4
comparing  2  and  2
comparing  2  and  4
[1, 2, 2, 4]

示例 2：使用 cmp_to_key() 函数提供的键从列表中打印最大和最小数字的程序

import functools


def mycmp(a, b):
	print("comparing ", a, " and ", b)
	if a > b:
		return 1
	elif a < b:
		return -1
	else:
		return 0


print(min([45, 78, 813], key=functools.cmp_to_key(mycmp)))
print(max([45, 78, 813], key=functools.cmp_to_key(mycmp)))

输出：

comparing  78  and  45
comparing  813  and  45
45
comparing  78  and  45
comparing  813  and  78
813

-------------------------------------------------------以下是leetcode题目-----------------------------------------------

解决方法：

class Solution:
    def largestNumber(self, nums):
        def cmp_func(x, y):
            if x + y > y + x:
                return 1
            elif x == y:
                return 0
            else:
                return -1
        nums = [str(num) for num in nums]
        nums.sort(key = cmp_to_key(cmp_func), reverse = True)
        return ''.join(nums).lstrip('0') or '0'