Codeforces Round #136 (Div. 1) B. Little Elephant and Array（简单莫队）

最新推荐文章于 2024-08-31 12:42:46 发布

原创最新推荐文章于 2024-08-31 12:42:46 发布 · 399 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#codeforces #莫队

莫队算法专栏收录该内容

5 篇文章

订阅专栏

本文通过一道具体题目，讲解了如何使用莫队算法解决区间查询问题，并采用离散化手段来处理大规模数据的情况。

题意：给你n个数，m次询问，每次询问一个区间[l, r]，问这个区间内元素出现个数等于元素本身的有几个。（n, m <= 1e5）

思路：裸的莫队，因为每个数比较大，离散化下就好了。

代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1e5+5;
int n, m, unit, tmp;
int a[maxn], aa[maxn], Hash[maxn];
int num[maxn], ans[maxn];
struct node
{
    int id, l, r, blk;
    bool operator <(const node &a) const
    {
        if(blk == a.blk) return r < a.r;
        else return blk < a.blk;
    }
}op[maxn];

void add(int id)
{
    num[aa[id]]++;
    if(num[aa[id]] == a[id]) tmp++;
    if(num[aa[id]] == a[id]+1) tmp--;
}

void del(int id)
{
    num[aa[id]]--;
    if(num[aa[id]] == a[id]) tmp++;
    if(num[aa[id]] == a[id]-1) tmp--;
}

void solve()
{
    tmp = 0;
    memset(num, 0, sizeof(num));
    int l = 1, r = 0;
    for(int i = 1; i <= m; i++)
    {
        while(r < op[i].r) add(++r);
        while(r > op[i].r) del(r--);
        while(l < op[i].l) del(l++);
        while(l > op[i].l) add(--l);
        ans[op[i].id] = tmp;
    }
    for(int i = 1; i <= m; i++)
        printf("%d\n", ans[i]);
}

int main(void)
{
    while(cin >> n >> m)
    {
        unit = (int)sqrt(n);
        for(int i = 1; i <= n; i++)
            scanf("%d", &a[i]), Hash[i] = a[i];
        sort(Hash+1, Hash+1+n);
        int d = unique(Hash+1, Hash+1+n)-Hash-1;
        for(int i = 1; i <= n; i++)
            aa[i] = lower_bound(Hash+1, Hash+1+d, a[i])-Hash;
        for(int i = 1; i <= m; i++)
            scanf("%d%d", &op[i].l, &op[i].r), op[i].id = i, op[i].blk = op[i].l/unit;
        sort(op+1, op+1+m);
        solve();
    }
    return 0;
}