日常——查询数的范围

题目描述

给定一个按照升序排列的长度为n的整数数组，以及 q 个查询。

对于每个查询，返回一个元素k的起始位置和终止位置（位置从0开始计数）。

如果数组中不存在该元素，则返回“-1 -1”。
输入格式

第一行包含整数n和q，表示数组长度和询问个数。

第二行包含n个整数（均在1~10000范围内），表示完整数组。

接下来q行，每行包含一个整数k，表示一个询问元素。
输出格式

共q行，每行包含两个整数，表示所求元素的起始位置和终止位置。

如果数组中不存在该元素，则返回“-1 -1”。

数据范围

$1\le n\le 100000$
$1\le q\le 10000$
$1\le k\le 10000$

输入样例：

6 3
1 2 2 3 3 4
3
4
5

输出样例：

3 4
5 5
-1 -1

Solution

package cn.zxy.test;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

/**
 * 数的范围
 * 6 3
 * 1 2 2 3 3 4
 * 3
 * 4
 * 5
 */
public class NumSelect{

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String[] s = br.readLine().split(" ");
        int n = Integer.parseInt(s[0]);
        int q = Integer.parseInt(s[1]);
        s = br.readLine().split(" ");
        int[] a = new int[n];
        for (int i = 0; i < n; i++) {
            a[i] = Integer.parseInt(s[i]);
        }
        for (int i = 0; i < q; i++) {
            int k = Integer.parseInt(br.readLine());
            //low 起始位置， high终止位置
            int low = 0, high = 0;
            //查找第一个值等于给定值元素
            int l = 0, r = n - 1;
            while (l <= r){
                int mid = l + r >> 1;
                if (a[mid] > k) r = mid - 1;
                else if (a[mid] < k) l = mid + 1;
                else {
                    if (mid == 0 || a[mid - 1] != k){
                        low = mid;
                        break;
                    }
                    else r = mid - 1;
                }
            }
            if (a[low] != k){
                System.out.println("-1 -1");
            }else {
                //查找最后一个值等于给定值的元素
                l = 0;
                r = n - 1;
                while(l <= r){
                    int mid = l + r >> 1;
                    if(a[mid] > k) r = mid - 1;
                    else if(a[mid] < k) l = mid + 1;
                    else{
                        if(mid == n - 1 || a[mid + 1] != k){
                            high = mid;
                            break;
                        }else{
                            l = mid + 1;
                        }
                    }
                }
                System.out.println(low + " " + high);
            }
        }
    }
}

简单模拟，这里是二分查找