[BZOJ2743][HEOI2012]采花(离线树状数组)

传送门


这题和1878做法是一样的。我也有写blog。
但是1878要求的是区间内出现次数>1的颜色个数,而这题求的是区间中出现次数>2的颜色个数,那我们树状数组修改的就是last[last[i]]就好咯,具体做法见上一题。

注意:这里的last会访问到0,注意在树状数组里面特判!


#include
#include
#include
#include
using namespace std;
const int maxn=1000010;
int n,m;
int lowbit(int x){return x&-x;}
int last[1000010],a[maxn],s[maxn];
int pos[maxn];
int getsum(int x)
{
    int ans=0;
    while(x>0)
    {
        ans+=s[x];
        x-=lowbit(x);
    }
    return ans;
}
void add(int x,int c)
{
    if(x==0) return;
    while(x<=n)
    {
        s[x]+=c;
        x+=lowbit(x);
    }
}
struct ques
{
    int l,r,id,ans;
}q[maxn*4];
bool cmp1(ques a,ques b){return a.rreturn a.idint main()
{
    int c;
    scanf("%d%d%d",&n,&c,&m);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    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,cmp1);
    int t=1;
    for(int i=1;i<=n;i++)
    {
        last[i]=pos[a[i]]; pos[a[i]]=i;
        add(last[i],1); 
        add(last[last[i]],-1);
        while(q[t].r==i)
        {
            q[t].ans=getsum(q[t].r)-getsum(q[t].l-1);
            t++;
        }
    }
    sort(q+1,q+m+1,cmp2);
    for(int i=1;i<=m;i++) printf("%d\n",q[i].ans);
    return 0;
}

你可能感兴趣的:(树状数组)