[题解]AT_abc248_d [ABC248D] Range Count Query

思路

其实很简单，我们可以将所有数值相同的值的下标存入一个 vector 里面。因为，我们既然要查找 \(X\)，不妨把所有值为 \(X\) 的下标存在一起，方便查找。（可以在输入的时候完成）

我们不妨在每一个数值后面添加一个哨兵，然后二分查找第一个大于等于 \(l\) 的数和第一个大于等于 \(r + 1\) 的数，再把两数相减即为答案。

这时候，哨兵的作用就体现出来了，如果我们的第二个数取到了哨兵，那么说明 \(l \sim r\) 是在最后的，我们将它设为 \(\operatorname{size}(v_x) + 1\) 即可。

Code

#include <bits/stdc++.h>  
#define re register  
  
using namespace std;  
  
const int N = 2e5 + 10,inf = 1e6 + 10;  
int n,q,Max;  
vector<int> v[N];  
  
inline int read(){  
    int r = 0,w = 1;  
    char c = getchar();  
    while (c < '0' || c > '9'){  
        if (c == '-') w = -1;  
        c = getchar();  
    }  
    while (c >= '0' && c <= '9'){  
        r = (r << 3) + (r << 1) + (c ^ 48);  
        c = getchar();  
    }  
    return r * w;  
}  
  
int main(){  
    n = read();  
    for (re int i = 1;i <= n;i++){  
        int x;  
        x = read();  
        Max = max(Max,x);  
        v[x].push_back(i);  
    }  
    for (re int i = 1;i <= Max;i++) v[i].push_back(inf);//哨兵   
    q = read();  
    while (q--){  
        int l,r,x;  
        l = read();  
        r = read();  
        x = read();  
        int a = lower_bound(v[x].begin(),v[x].end(),l) - v[x].begin();//二分查找   
        int b = lower_bound(v[x].begin(),v[x].end(),r + 1) - v[x].begin();//这里貌似也可以用 upper_bound   
        if (b == inf) b = v[x].size() + 1;//取到哨兵   
        printf("%d\n",b - a);//输出结果   
    }  
    return 0;  
}