题目描述:
给你一个排序后的字符列表 letters ,列表中只包含小写英文字母。另给出一个目标字母target,请你寻找在这一有序列表里比目标字母大的最小字母。
在比较时,字母是依序循环出现的。举个例子:
如果目标字母 target = ‘z’ 并且字符列表为letters = [‘a’, ‘b’],则答案返回’a’
题意解读:
首先给定一个只含英文字母的列表和一个目标字母target, 我们要找到这个有序列表里面比目标字母大的最小字母,但是如果列表中没有比目标字母大的字母,就返回列表第一个字母。所以只要列表的最后一个字母比目标字母大,那么整个列表就没有比目标字母大的字母。
我自己的解法:
class Solution:
def nextGreatestLetter(self, letters: list, target: str) -> str:
start, end = 0, len(letters) - 1
while start <= end:
if target >= letters[-1]:
return letters[0]
mid = (start + end) // 2
if letters[mid] > target:
if mid == 0:
return letters[mid]
elif letters[mid - 1] > target:
end = mid - 1
else:
return letters[mid]
else:
start = mid + 1
官方解法:
线性查找:
class Solution:
def nextGreatestLetter(self, letters: List[str], target: str) -> str:
return next((letter for letter in letters if letter > target), letters[0])
二分查找:
利用列表有序的特点,可以使用二分查找降低时间复杂度。
首先比较目标字母和列表中的最后一个字母,当目标字母大于或等于列表中的最后一个字母时,答案是列表的首个字母。当目标字母小于列表中的最后一个字母时,列表中一定存在比目标字母大的字母,可以使用二分查找得到比目标字母大的最小字母。
初始时,二分查找的范围是整个列表的下标范围。每次比较当前下标处的字母和目标字母,如果当前下标处的字母大于目标字母,则在当前下标以及当前下标的左侧继续查找,否则在当前下标的右侧继续查找。
class Solution:
def nextGreatestLetter(self, letters: List[str], target: str) -> str:
return letters[bisect_right(letters, target)] if target < letters[-1] else letters[0]