8、找出第一个大于或小于目标的索引

厨子大约 4 分钟数据结构算法算法基地面试刷题

我们在上面的变种中,描述了如何找出目标元素在数组中的上下边界,然后我们下面来看一个新的变种。

如何从数组或区间中找出第一个大于或最后一个小于目标元素的数的索引。

例 nums = {1,3,5,5,6,6,8,9,11} 我们希望找出第一个大于 5 的元素的索引,

那我们需要返回 4 ,因为 5 的后面为 6,第一个 6 的索引为 4,如果希望找出最后一个小于 6 的元素,那我们则会返回 3 ,

因为 6 的前面为 5 最后一个 5 的索引为 3。好啦题目我们已经了解,下面我们先来看一下如何在数组或区间中找出第一个大于目标元素的数吧。

大于目标元素

题目描述

找出第一个大于目标元素的数,大概有以下几种情况

模糊边界情况
模糊边界情况

1.数组包含目标元素,找出在他后面的第一个元素

2.目标元素不在数组中,数组内的部分元素大于它,此时我们需要返回第一个大于他的元素

3.目标元素不在数组中,且数组中的所有元素都大于它,那么我们此时返回数组的第一个元素即可

4.目标元素不在数组中,且数组中的所有元素都小于它,那么我们此时没有查询到,返回 -1 即可。

既然我们已经分析完所有情况,那么这个题目对咱们就没有难度了,下面我们描述一下案例的执行过程

nums = {1,3,5,5,6,6,8,9,11} target = 7

上面的例子中,我们需要找出第一个大于 7 的数,那么我们的程序是如何执行的呢?

二分查找模糊边界目标值
二分查找模糊边界目标值

上面的例子我们已经弄懂了,那么我们看一下,当 target = 0 时,程序应该怎么执行呢?

模糊边界目标0
模糊边界目标0

OK!我们到这一步就能把这个变种给整的明明白白的了,下面我们看一哈程序代码吧,也是非常简单的。

代码

Java Code:

public static int lowBoundnum(int[] nums,int target,int left, int right) {

    while (left <= right) {
        //求中间值
        int mid = left + ((right - left) >> 1);
        //大于目标值的情况
        if (nums[mid] > target) {
                //返回 mid
            if (mid == 0 || nums[mid-1] <= target) {
                return mid;
            }
            else{
                right = mid -1;
            }

        } else if (nums[mid] <= target){
            left = mid + 1;
        }
    }
    //所有元素都小于目标元素
    return -1;
}

Go Code:

func lowBoundnum(nums []int, target, left, right int) int {

    for (left <= right) {
        //求中间值
        mid := left + ((right - left) >> 1);
        //大于目标值的情况
        if (nums[mid] > target) {
                //返回 mid
            if (mid == 0 || nums[mid-1] <= target) {
                return mid
            }else{
                right = mid -1
            }

        } else if (nums[mid] <= target){
            left = mid + 1
        }
    }
    //所有元素都小于目标元素
    return -1
}

小于目标元素

题目描述

通过上面的例子我们应该可以完全理解了那个变种,下面我们继续来看以下这种情况,那就是如何找到最后一个小于目标数的元素。还是上面那个例子

nums = {1,3,5,5,6,6,8,9,11} target = 7

查找最后一个小于目标数的元素,比如我们的目标数为 7 ,此时他前面的数为 6,最后一个 6 的索引为 5,此时我们返回 5 即可,如果目标数元素为 12,那么我们最后一个元素为 11,仍小于目标数,那么我们此时返回 8,即可。这个变种其实算是上面变种的相反情况,上面的会了,这个也完全可以搞定了,下面我们看一下代码吧。

代码

Java Code:

public static int upperBoundnum(int[] nums,int target,int left, int right) {

    while (left <= right) {

        int mid = left + ((right - left) >> 1);
            //小于目标值
        if (nums[mid] < target) {
            //看看是不是当前区间的最后一位,如果当前小于,后面一位大于,返回当前值即可
            if (mid == right || nums[mid+1] >= target) {
                return mid;
            }
            else{
                left = mid + 1;
            }

        } else if (nums[mid] >= target){
            right = mid - 1;
        }
    }
    //没有查询到的情况
    return -1;
}

Go Code:

func upperBoundnum(nums []int, target, left, right int) int {

	for left <= right {

		mid := left + ((right - left) >> 1)
		//小于目标值
		if nums[mid] < target {
			//看看是不是当前区间的最后一位,如果当前小于,后面一位大于,返回当前值即可
			if mid == right || nums[mid+1] >= target {
				return mid
			} else {
				left = mid + 1
			}

		} else if nums[mid] >= target {
			right = mid - 1
		}
	}
	//没有查询到的情况
	return -1
}