天天看點

CodeForces - 220B Little Elephant and Array 莫隊 離散化

題目:CodeForces - 220B Little Elephant and Array

題意:

給定n個數和m個區間,求每一個區間内 出現次數等于該數字的數的個數。

#include<iostream>
#include<algorithm>
#include<cstdlib>
#include<sstream>
#include<cstring>
#include<bitset>
#include<cstdio>
#include<string>
#include<deque>
#include<stack>
#include<cmath>
#include<queue>
#include<set>
#include<map>
using namespace std;
typedef long long ll;
const int maxn = 1e5+10;
struct node
{
    int l,r,id;
    int sum;
}t[maxn];
int ans;
int a[maxn],cnt[maxn],pos[maxn],block,b[maxn];
//          出現次數 , 位置
int c[maxn]; //存儲a[i]的相對大小     注意此數組的作用
bool cmp1(node a,node b)
{
    if(pos[a.l]==pos[b.l])
        return a.r<b.r;
    return a.l<b.l;
}
bool cmp2(node a,node b)
{
    return a.id<b.id;
}
void add(int x)
{
    if(a[x]==cnt[c[x]])
    {
        ans--;
        cnt[c[x]]++;
    }
    else
    {
        cnt[c[x]]++;
        if(a[x]==cnt[c[x]])
            ans++;
    }
}
void del(int x)
{
    if(a[x]==cnt[c[x]])
    {
        ans--;
        cnt[c[x]]--;
    }
    else
    {
        cnt[c[x]]--;
        if(a[x]==cnt[c[x]])
            ans++;
    }
}
int main()
{
    int n,m;
    int l,r;
    scanf("%d%d",&n,&m);
    block = sqrt(1.0*n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        b[i] = a[i];
    }
    
    sort(b+1,b+n+1);
    int k = unique(b+1,b+n+1)-b-1;  //去重  離散化
    for(int i=1;i<=n;i++)
        c[i] = lower_bound(b+1,b+k+1,a[i])-b;
        
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d",&t[i].l,&t[i].r);
        t[i].id = i;
        pos[i] = (i-1)/block+1;
    }
    sort(t+1,t+m+1,cmp1);
    
    l = t[1].l;
    r = t[1].l-1;
    ans = 0;
    for(int i=1;i<=m;i++)
    {
        while(l>t[i].l)  add(--l);
        while(r<t[i].r)  add(++r);
        while(l<t[i].l)  del(l++);
        while(r>t[i].r)  del(r--);
        t[i].sum = ans;
    }
    sort(t+1,t+m+1,cmp2);
    for(int i=1;i<=m;i++)
        printf("%d\n",t[i].sum);
    return 0;
}