744、寻找比目标字母大的最小字母
给你一个排序后的字符列表 letters ,列表中只包含小写英文字母。另给出一个目标字母 target,请你寻找在这一有序列表里比目标字母大的最小字母。
在比较时,字母是依序循环出现的。举个例子:
- 如果目标字母 target = ‘z’ 并且字符列表为 letters = [‘a’, ‘b’],则答案返回 ‘a’
示例1:
输入: letters = ["c", "f", "j"],target = "a"
输出: "c"
示例2:
输入: letters = ["c","f","j"], target = "c"
输出: "f"
示例3:
输入: letters = ["c","f","j"], target = "d"
输出: "f"
思路:
解法1:线性查找
查找比较所有的字符,如果没有找到,说明最小的字符是第一个
class Solution:
def nextGreatestLetter(self, letters: List[str], target: str) -> str:
ans = None
for letter in letters:
if letter > target:
if not ans: ans = letter
elif letter < ans: ans = letter
if not ans: letters[0]
return ans
解法2:二分查找
class Solution:
def nextGreatestLetter(self, letters: List[str], target: str) -> str:
# 如果目标值大于最后一个字母,那就返回第一个字母
if target >= letters[-1]: return letters[0]
l, r = 0, len(letters) - 1
# 在[start, end]中寻找比目标值大的最小值,
# 如果中间值小于等于目标值,说明最小的不再左侧,直接将左侧数据丢弃,也就是l + 1
# 如果中间值大于目标值,那么说明[mid, r]中间可能有最优解,所以将r-1尝试找出最优解
while start <= end:
mid = (l + r) >> 1
if letters[mid] <= target:
l = mid + 1
else:
r = mid - 1
return letters[l]