给定一个只包含小写字母的有序数组letters
和一个目标字母 target
,寻找有序数组里面比目标字母大的最小字母。
数组里字母的顺序是循环的。举个例子,如果目标字母target = 'z'
并且有序数组为 letters = ['a', 'b']
,则答案返回 'a'
。
示例:
输入:
letters = ["c", "f", "j"]
target = "a"
输出: "c"
输入:
letters = ["c", "f", "j"]
target = "c"
输出: "f"
输入:
letters = ["c", "f", "j"]
target = "d"
输出: "f"
输入:
letters = ["c", "f", "j"]
target = "g"
输出: "j"
输入:
letters = ["c", "f", "j"]
target = "j"
输出: "c"
输入:
letters = ["c", "f", "j"]
target = "k"
输出: "c"
注:
思路:
1、利用集合特性排除多余元素,利用内置的sorted()函数排序给经过结合去重的列表元素进行排序。
2、对排好序的元素进行处理,如果target在列表中,返回的是target后边的元素;如果不在,append()函数插入target元素后再找target后边的元素返回。
letters
长度范围在[2, 10000]
区间内。letters
仅由小写字母组成,最少包含两个不同的字母。- 目标字母
target
是一个小写字母。
class Solution(object):
def nextGreatestLetter(self, letters, target):
"""
:type letters: List[str]
:type target: str
:rtype: str
"""
letters_sort = sorted(set(letters))
if target in letters_sort and letters_sort.index(target) < len(letters_sort) - 1:
return letters_sort[letters_sort.index(target) + 1]
elif target in letters_sort and letters_sort.index(target) == len(letters_sort) - 1:
return letters_sort[0]
elif target not in letters_sort:
letters_sort.append(target)
letters_sortNew = sorted(letters_sort)
if letters_sortNew.index(target) == len(letters_sortNew) - 1:
return letters_sortNew[0]
else:
return letters_sortNew[letters_sortNew.index(target) + 1]