全球今热点:二分查找法upper版(找大于某个值的最小下标)递归+非递归版

来源:博客园   2023-06-22 14:12:38


(相关资料图)

需求:比如说查询一个班级大于60分的最低分等等。

思路与二分法基本相同,只不过是对比的逻辑发生了一些小变化,这里所说的上界就是指大于某个值的最小下标。

当mid < target :说明 target 的上界还在mid的右边,所以要去找比mid大的

当mid > target:说明 mid 有可能是target的上界,所以我们加个判断,如果mid前一个元素就刚好是target,说明mid就是我们要找的上界,否则继续找。

另一个注意的点,就是我们取变量 r 的时候,不再是取数组最大的下标了,而是要超过一个,因为如果你找的是数组最后一个元素的上界,其实是不在数组里的。

递归版:
package com.Search;/** * @Author: 翰林猿 * @Description: 二分查找upper版(找大于某个值的最小下标) **/public class BinarySearchUpper {  public BinarySearchUpper() {   }​  public static > int SearchUpper(E[] data, E target) {    return SearchUpper(data, 0, data.length, target);   //是length而不是length-1,因为有可能上界值不在该数组里,所以此时r就应该取大于数组最大值下标+1的位置   }​  /**   * @Description 递归版本   */  public static > int SearchUpper(E[] data, int l, int r, E target) {    if (l >= r) return -1;    int mid = l + (r - l) / 2;    if (data[mid].compareTo(target) <= 0) {     //如果mid小于目标值,说明target的上界还在右边,要去找比mid大的      return SearchUpper(data, mid + 1, r, target);     }    if (data[mid].compareTo(target) > 0 && data[mid - 1].compareTo(target) == 0) {      return mid;     }    //说明mid又大于target,但是mid-1又不等于target,说明当前的mid不是上界。比如说测试用例1, 1, 1, 2, 2, 3, 6, 8, 18, 20    //我们要找大于3的最小下标也就是6的下标,但是经过两次递归,mid=18,但是18所在的下标为8,8-1=7的元素是8,但是8并不等于3,所以mid=18时不是我们要找的upper    //此时继续递归    return SearchUpper(data, l, mid, target);  //mid已经大于目标值了,而且当前的mid不是上界,所以我们往左边找​   }​    public static void main (String[]args){      Integer[] arr = {1, 1,1, 2, 2, 3, 6, 8, 18, 20};      int index = SearchUpper(arr, 3);      System.out.println(index);     }   }
非递归版:
package com.Search;/** * @Author: 翰林猿 * @Description: 二分查找upper版(找大于某个值的最小下标) **/public class BinarySearchUpper {  public BinarySearchUpper() {   }​  public static > int SearchUpper2(E[] data, E target) {    return SearchUpper2(data, 0, data.length, target);   //是length而不是length-1,因为有可能上界值不在该数组里,所以此时r就应该取大于数组最大值下标+1的位置   }    /**   * @Description 非递归版本   */  public static > int SearchUpper2(E[] data, int l, int r, E target) {    while (l < r) {      int mid = l + (r - l) / 2;      if (data[mid].compareTo(target) == 0) {        return mid + 1;       } else if (data[mid].compareTo(target) > 0) { // 这个r = mid是因为mid的位置可能是目标值        r = mid;       } else {        l = mid + 1;       }     }    // l和r最后都都指向同一个位置。没找到则返回arr.length    return l;   }​    public static void main (String[]args){      Integer[] arr = {1, 1,1, 2, 2, 3, 6, 8, 18, 20};      int index2 = SearchUpper2(arr, 3);      System.out.println(index2);     }   }
资讯