Leetcode 题解 --二分查找--大于给定元素的最小元素

大于给定元素的最小元素

744. Find Smallest Letter Greater Than Target (Easy)

Input:

letters = ["c", "f", "j"]

target = "d"

Output: "f"

 

Input:

letters = ["c", "f", "j"]

target = "k"

Output: "c"

题目描述:给定一个有序的字符数组 letters 和一个字符 target,要求找出 letters 中大于 target 的最小字符,如果找不到就返回第 1 个字符。

解法:利用二分查找,找得到返回letters 中大于 target 的最小字符,找不到就返回第 1 个字符。不如先设定返回值为第一个字符,由于是二分查找,我们发现一个比target大的就给c赋值,因为h是不断变小的,c的值在循环的过程中肯定会变成大于 target 的最小字符

class Solution {
    public char nextGreatestLetter(char[] letters, char target) {
        int n = letters.length;
        int l = 0, h = n-1;
        char c = letters[0];
        while(l <= h){
            int m = l + (h - l) / 2;
            if(letters[m] > target){
                c = letters[m];
                h = m - 1;
            }
            else
                l = m + 1;
        }
        return c;
    }
}

另一种写法

Leetcode 题解 --二分查找--大于给定元素的最小元素_第1张图片

class Solution {
    public char nextGreatestLetter(char[] letters, char target) {
        int n = letters.length;
        int l = 0, h = n - 1;
        while (l <= h) {
            int m = l + (h - l) / 2;
            if (letters[m] <= target) {
                l = m + 1;                最后l停下的位置 
            } else {
                h = m - 1;                    
            }
        }
        return l < n ? letters[l] : letters[0];
    }
}

对while (l <= h) 还是 l < h做个人的总结:

做了这两道题目,有一点感触 l <= h 的时候,假设l = h 的时候 均等于m = 2 最后的返回 l = m + 1 h = m或者 h = m - 1, l = m为1

也就是返回的情况需要考虑到左右两边的情况 

当 l < h的时候,就是 l = h = m = 2就可以返回了,此时 l 和 h 是相等的,也就是说我要得就是相等时候的index

比如上面那道题 显然是要返回 l = m + 1的那个index。

你可能感兴趣的:(leetcode)