leetcode 744. 寻找比目标字母大的最小字母(python)

给定一个只包含小写字母的有序数组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后边的元素返回。

1. letters长度范围在[2, 10000]区间内。
2. letters 仅由小写字母组成，最少包含两个不同的字母。
3. 目标字母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]