一直以来都不知道python应该怎么做到自定义排序,今天刷leetcode的时候终于学习了一波。
题目链接:https://leetcode-cn.com/problems/longest-word-in-dictionary/description/
题目的大概意思是在一个字符串列表中尝试找出一个字符串,使得它是由列表中其他字符串逐步添加一个字符生成的。我的思路很简单,先按长度大到小排序,一旦找到符合的就输出,问题在于,如果有相同长度的字符串满足要求,输出字典序小的,那么直接按长度大到小排序肯定还要做很多没用的判断。这时就需要自定义排序了。
def MyCmp(x1, x2):#定义一个比较函数
if 条件1:
return 1
elif 条件2:
return -1
l.sort(cmp=MyCmp):#把比较函数传进sort里即可
自定义的比较函数也很简单,只需要记住,根据需要条件,如果要把x1放到x2后面就return 1,如果要把x1放到x2前面就return -1,相同的话return 0
上一下这道题的代码:
#encoding=utf-8
import numpy as np
class Solution(object):
def string_cmp(self, s1, s2):
if len(s1) < len(s2):
return 1
elif len(s1) > len(s2):
return -1
else:
if s1 > s2:
return 1
elif s1 < s2:
return -1
return 0
def longestWord(self, words):
"""
:type words: List[str]
:rtype: str
"""
words.sort(cmp=self.string_cmp)
for item in words:
top = len(item)
flag = True
while top > 0:
if item[0:top] in words:
top -= 1
continue
else:
flag = False
break
if flag == True:
return item
return ''