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

传送门


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

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


#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
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.r<b.r;}
bool cmp2(ques a,ques b){return a.id<b.id;}
int 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;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值