ABC 248 | D - Range Count Query

题目描述

给定一个数组$A = (a_1, a_2, ..., a_3)$，给定$Q$组询问，每组询问给出$l、r、x$，输出下标$[l, r]$范围内数值等于$x$的元素个数。

数据范围

• $1 \le N \le 2 \times 10^5$
• $1 \le A_i \le N$
• $1 \le Q \le 2 \times 10^5$
• $1 \le L \le R \le N, 1 \le X \le N$

题解

由于本题中每个数字的范围是$2 \times 10^5$，所以可以预处理出每个数字的下标，对于每组询问$L, R, X$，只需二分查找$X$的位置数组，计算在$[L, R]$范围内的下标的个数即可。
若存入vector中，可直接使用lower_bound()和upper_bound()查询，准确且方便

代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>

using namespace std;

const int N = 2e5 + 10;

vector<int> vec[N];
int n, q;

int main()
{
    scanf("%d", &n);
    for(int i = 1; i <= n; i ++){
        int x;
        scanf("%d", &x);
        vec[x].push_back(i);
    }
    scanf("%d", &q);
    while(q --){
        int l, r, x;
        scanf("%d%d%d", &l, &r, &x);
        auto itl = lower_bound(vec[x].begin(), vec[x].end(), l);
        auto itr = upper_bound(vec[x].begin(), vec[x].end(), r);
        printf("%d\n", itr - itl);
    }
    return 0;
}