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

以你之姓@ 2022-04-22 04:22 115阅读 0赞

给定一个只包含小写字母的有序数组`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]