Acwing #789(数的范围)

1.解法一

思路：因为数组为升序排列，所以只需要知道k的起始位置和个数，即可知道k的终止位置（起始为值 + 个数 - 1）。

#include <iostream>
using namespace std;
const int MAXN = 100005;
typedef pair<int, int> PII;
int value[MAXN];
//cnt[i]表示值为i的起始位置和值为i的个数,其中first表示开始位置,second表示个数
PII cnt[MAXN];  
int main() {
    int n, q;
    cin >> n >> q;
    for(int i = 0; i < n; i++) {
        cin >> value[i];
        if(!cnt[value[i]].second) { //如果second为0,则表示value[i]第一次出现,记录起始位置
            cnt[value[i]].first = i;
        }
        cnt[value[i]].second += 1;  //value[i]的个数自增1
    }
    while(q--) {
        int k;
        cin >> k;
        if(!cnt[k].second) {    //如果k的个数为0,则表示没有此数字
            cout << "-1 -1" << endl;
        }else {
            //起始位置为first,终点位置为first + 个数 - 1
            int l = cnt[k].first, r = cnt[k].second + cnt[k].first - 1;
            cout << l << ' ' << r << endl;  
        }
    }
    return 0;
}

2.解法二

思路：题目为有序数组，所以用二分查找比较合适，分别查找起点和终点即可。

#include <iostream>
using namespace std;
const int MAXN = 100005;
int value[MAXN];
int main() {
    int n, q;
    cin >> n >> q;
    for(int i = 0; i < n; i++) {
        cin >> value[i];
    }
    while(q--) {
        int k;
        cin >> k;
        //二分查找起点(第一个不小于k的值)
        int l = 0, r = n - 1;
        while(l < r) {
            int mid = (l + r) >> 1;
            if(value[mid] >= k) {
                r = mid;
            }else {
                l = mid + 1;
            }
        }
        if(value[r] != k) { //即未找到k
            cout << "-1 -1" << endl;
            continue;
        }
        //输出左值
        cout << r << ' ';
        //二分查找终点(最后一个不大于k的值) 
        l = 0, r = n - 1;
        while(l < r) {
            int mid = (l + r + 1) >> 1;
            if(value[mid] <= k) {
                l = mid;
            }else {
                r = mid - 1;
            }
        }
        cout << l << endl;
    }
    return 0;
}