【每日一题】ABC248D - Range Count Query | 二分 | 简单

题目内容

原题链接

给定一个长度为 $n$ 的整数数组 $a$ 。

$Q$ 次询问，每次给定一个区间 $[L,R]$ ，问这个区间内有多少 $i\in [L,R]$ 且 $a_i=X$ 。

数据范围

• $1\le n\le 2\times 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$

题解

由于值域很小，所以可以对每个值构造一个数组，存下这些数对应的下标。

然后对于每个询问，二分对应的值域数组找到左右区间即可。

时间复杂度：$O(n\log n)$

代码

#include <bits/stdc++.h>
using namespace std;

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int n;
    cin >> n;

    vector<vector<int>> g(n);
    for (int i = 0; i < n; ++i) {
        int x; cin >> x;
        x--;
        g[x].push_back(i);
    }


    int Q;
    cin >> Q;
    while (Q--) {
        int L, R, X;
        cin >> L >> R >> X;
        L--, R--, X--;
        int ans = 0;
        auto itL = lower_bound(g[X].begin(), g[X].end(), L);
        if (itL != g[X].end()) {
            auto itR = upper_bound(g[X].begin(), g[X].end(), R);
            ans = int(itR - itL);
        }
        cout << ans << "\n";
    }

    return 0;
}