【Leetcode】【python】Find Smallest Letter Greater Than Target 寻找比目标字母大的最小字母

题目大意

给定一个有序的字符数组 letters 和一个字符 target,要求找出 letters 中大于 target 的最小字符。letters 字符数组是循环数组。

解题思路

二分查找变种:

应该注意最后返回的是 l 位置的字符。

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
class Solution(object):
def nextGreatestLetter(self, letters, target):
"""
:type letters: List[str]
:type target: str
:rtype: str
"""
l = 0
r = len(letters) - 1
while l <= r:
m = l + (r - l) / 2
if letters[m] <= target:
l = m + 1
else:
r = m - 1
if l < len(letters):
return letters[l]
else:
return letters[0]

总结