bzoj 1878 HH的项链 (树状数组+离线)

5 篇文章 0 订阅
1 篇文章 0 订阅

题目大意:给你一个序列,求某区间出现不同的数的个数。

貌似离线树状数组是最好的解法

先把所有询问挂在它们询问的右端点上

然后从头到尾遍历这个序列,记录这个位置的值上一次出现的位置

那么,当遍历到第i位时,如果a[i]在之前出现过,就在它上一次出现的位置-1

这个操作的意义是,第i位已经有a[i]了,那么上一次出现a[i]的位置已经失去意义

接着在这个位置+1,更新last[a[i]]。差分操作用树状数组维护

然后我们遍历以这个位置为结尾的所有询问,用树状数组查前缀和,因为在a[i]相同的位置不会重复打差分,所以query(r)-query(l-1)即为这个询问的答案。

#include <cstdio>
#include <algorithm>
#include <cstring>
#include <queue>
#define inf 0x3f3f3f3f
#define ll long long 
#define N 500100
#define maxn 1000000
using namespace std;
//re
int n,m,cte;
int a[N],head[N];
int lst[N*2],sum[N*2];
struct Ques{
    int l,r,ans,id;
}ques[N];
struct EDGE{
    int to,nxt;
}edge[N];
void edge_add(int u,int v)
{
    cte++;
    edge[cte].to=v;
    edge[cte].nxt=head[u];
    head[u]=cte;
}
int cmp1(Ques s1,Ques s2) {return s1.r<s2.r;}
int cmp2(Ques s1,Ques s2) {return s1.id<s2.id;}
void update(int x,int w) {for(int i=x;i<=maxn;i+=(i&(-i)))sum[i]+=w;}
int query(int x) {int ans=0;for(int i=x;i>0;i-=(i&(-i)))ans+=sum[i];return ans;}

int main()
{
    //freopen("data.in","r",stdin);
    scanf("%d",&n);
    memset(head,-1,sizeof(head));
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    scanf("%d",&m);
    for(int i=1;i<=m;i++)
        scanf("%d%d",&ques[i].l,&ques[i].r),ques[i].id=i;
    sort(ques+1,ques+m+1,cmp1);
    for(int i=1;i<=m;i++)
        edge_add(ques[i].r,i);
    for(int i=1;i<=n;i++)
    {
        if(lst[a[i]]) update(lst[a[i]],-1);
        update(i,1),lst[a[i]]=i;
        for(int j=head[i];j!=-1;j=edge[j].nxt){
            int v=edge[j].to;
            ques[v].ans=query(ques[v].r)-query(ques[v].l-1);
        }
    }
    sort(ques+1,ques+m+1,cmp2);
    for(int i=1;i<=m;i++) printf("%d\n",ques[i].ans);
    return 0;
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值