Codeforces Round #136 B - Little Elephant and Array(莫队算法)

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/kuronekonano/article/details/83625401

在这里插入图片描述
题意:给n个数,m次询问,每次询问给出一个区间,问区间内其值和出现次数相等的数有多少个。
简单莫队,注意在因为数字值大小给出的是1e9的,而我们要记录每个数的出现次数即要用数组标记,但是其实因为n的大小只有1e5,也就是说超过1e5的值都是不可能被记入答案的,忽略掉就好。并不用离散化。直接用莫队跑一边就好了。

#include<bits/stdc++.h>
#define LL long long
#define M(a,b) memset(a,b,sizeof a)
#define pb(x) push_back(x)
using namespace std;
const int maxn=1e5+7;
int n,m,a[maxn],sum,pos[maxn];
struct node
{
    int l,r,id;
    bool operator <(const node &a)const
    {
        return pos[l]==pos[a.l]?r<a.r:pos[l]<pos[a.l];
    }
} q[maxn];
int cnt[maxn],ans[maxn];
void add(int x)
{
    if(a[x]>n)return;
    if(cnt[a[x]]==a[x])sum--;
    cnt[a[x]]++;
    if(cnt[a[x]]==a[x])sum++;
}
void del(int x)
{
    if(a[x]>n)return;
    if(cnt[a[x]]==a[x])sum--;
    cnt[a[x]]--;
    if(cnt[a[x]]==a[x])sum++;
}
int main()
{
    sum=0;
    scanf("%d%d",&n,&m);
    int sz=sqrt(n);
    for(int i=1; i<=n; i++) scanf("%d",&a[i]),pos[i]=i/sz;
    for(int i=1; i<=m; i++) scanf("%d%d",&q[i].l,&q[i].r),q[i].id=i;
    sort(q+1,q+m+1);
    M(cnt,0);
    int L=1,R=0;
    for(int i=1; i<=m; i++)
    {
        while(L<q[i].l) del(L++);
        while(L>q[i].l) add(--L);
        while(R<q[i].r) add(++R);
        while(R>q[i].r) del(R--);
        ans[q[i].id]=sum;
    }
    for(int i=1;i<=m;i++)printf("%d\n",ans[i]);
}

猜你喜欢

转载自blog.csdn.net/kuronekonano/article/details/83625401