力扣 744. 寻找比目标字母大的最小字母 [二分]

题目

给你一个排序后的字符列表 letters ，列表中只包含小写英文字母。另给出一个目标字母 target，请你寻找在这一有序列表里比目标字母大的最小字母。

在比较时，字母是依序循环出现的。举个例子：

如果目标字母 target = ‘z’ 并且字符列表为 letters = [‘a’, ‘b’]，则答案返回 ‘a’

示例 1：

输入: letters = [“c”, “f”, “j”]，target = “a”
输出: “c”
示例 2:

输入: letters = [“c”,“f”,“j”], target = “c”
输出: “f”
示例 3:

输入: letters = [“c”,“f”,“j”], target = “d”
输出: “f”

提示：

2 <= letters.length <= 104
letters[i] 是一个小写字母
letters 按非递减顺序排序
letters 最少包含两个不同的字母
target 是一个小写字母

解释与代码

简单二分题

最后特判一下即可

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