AcWing789数的范围

AcWing789数的范围

题目：

题目大意：

​ 给定一个长度为n的升序数组nums，和q个查询k，输出k在nums中的起始位置和终止位置，如果k不存在则输出-1 -1。

数据范围：

如图所示

思路：

​ 二分模板。注意两种不同的情况，均找的是某一性质的边界点。注意初始时l = 0, r = n - 1。

• 满足左区间性质时，找左区间的最右点(及k的最右点)，mid = l + r + 1 >> 1, if(nums[mid] <= k) l = mid else r = mid - 1 。
• 满足右区间性质时，找右区间的最左点(及k的最左点)，mid = l + r >> 1, if(num[mid] >= k ) r = mid else l = mid + 1。

结束时，需要判断nums[l] == k，因为最终的nums[l]可能不是k，此时l，r值的含义有以下两种情况：

• 如果k超过了该数组的最小值和最大值的范围则l,r最终是0或n-1。
• 如果k在该数组的最大值和最小值范围里，则l为将k插入nums数组后，k在nums中的位置。（此时r = l - 1）。

如果nums[l] == k，则找到了性质的边界点l。

代码：

import java.util.Scanner;

public class Main {

    public static int N = 100010;
    public static int[] nums = new int[N];

    public static int getLRange(int[] nums, int l, int r, int k){
        while (l < r){
            int mid = l + r >> 1;
            if (nums[mid] >= k)
                r = mid;
            else
                l = mid + 1;
        }
        System.out.println("l = " + l);
        if (nums[l] == k)
            return l;
        return -1;
    }

    public static int getRRange(int[] nums, int l, int r, int k){
        while (l < r){
            int mid = l + r + 1 >> 1;
            if (nums[mid] <= k)
                l = mid;
            else
                r = mid - 1;
        }
        System.out.println("r = " + l);
        if (nums[l] == k)
            return l;
        return -1;
    }

    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int n = scanner.nextInt(), q = scanner.nextInt();
        for (int i = 0; i < n; i ++ )
            nums[i] = scanner.nextInt();
        while (q -- > 0){
            int k = scanner.nextInt();
            int resl = getLRange(nums, 0, n - 1, k);
            int resr = getRRange(nums, 0, n - 1, k);
            System.out.println(resl + " " + resr);
        }
    }
}

时空复杂度分析等：

• 时间复杂度 : O(logn)

• 空间复杂度 : O(1)

题目链接：

789. 数的范围 - AcWing题库