744. 寻找比目标字母大的最小字母

744. 寻找比目标字母大的最小字母

题目链接：744. 寻找比目标字母大的最小字母

给你一个排序后的字符列表 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"

提示：

• letters 长度范围在 \([2, 10000]\) 区间内。
• letters 仅由小写字母组成，最少包含两个不同的字母。
• 目标字母 target 是一个小写字母。

思路

题目要求是在一个有序的字符数组中，找到大于 target 的第一个字符的位置。显然需要使用二分法，目标是找到 target 的右邻居。所以使用 二分查找 模板二比较好。

代码

首先考虑了边界情况，当 target 比所有的字符都大（小）时，返回 letter[0]。

class Solution {
public:
    char nextGreatestLetter(vector<char>& letters, char target) {
        int n = letters.size();
        if (target >= letters[n - 1] || target < letters[0]) {
            return letters[0];
        }
        int l = 0, r = n;
        while (l < r) {
            int mid = l + (r - l) / 2;
            if (target >= letters[mid]) {
                l = mid + 1;
            } else {
                r = mid;
            }
        }
        //上面这段代码返回是第一个大于target的位置，正常来说可能会产生越界
        //即l == letter.size()，但由于判断了target大于所有字符的情况
        //所以这里不用考虑
        return letters[l];
    }
};

优化

凡是存在循环的题目，可以考虑使用 \(\%\) 来应对边界情况，如下：

class Solution {
public:
    char nextGreatestLetter(vector<char>& letters, char target) {
        int n = letters.size();
        int l = 0, r = n;
        while (l < r) {
            int mid = l + (r - l) / 2;
            if (target >= letters[mid]) {
                l = mid + 1;
            } else {
                r = mid;
            }
        }
        return letters[l % n];
    }
};

总结

使用模板时，一定得考虑各个模板的适用情况，考虑跳出循环后的元素代表着什么