思路
其实很简单,我们可以将所有数值相同的值的下标存入一个 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;
}
标签:Count,int,题解,哨兵,Range,查找,ABC248D
From: https://www.cnblogs.com/WaterSun/p/18261971