BZOJ2743 HEOI2012采花（离线+树状数组）

最新推荐文章于 2018-12-23 21:21:00 发布

weixin_30740581

最新推荐文章于 2018-12-23 21:21:00 发布

阅读量93

点赞数

原文链接：http://www.cnblogs.com/Gloid/p/9608875.html

版权

本文介绍了一种使用树状数组解决区间内颜色第二次出现位置查询问题的方法。通过离线处理，按左端点排序并更新第二次出现的颜色位置，实现了高效查询。代码示例展示了如何初始化树状数组，进行添加和求和操作，以及如何处理输入和输出。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

　　如果能够把所有区间内第二次出现某颜色的位置标记出来，树状数组查询一下就可以了。

　　考虑离线。按左端点从小到大排序，不断移动左端点并更新第二次出现的位置。

#include<iostream> 
#include<cstdio>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;
int read()
{
    int x=0,f=1;char c=getchar();
    while (c<'0'||c>'9') {if (c=='-') f=-1;c=getchar();}
    while (c>='0'&&c<='9') x=(x<<1)+(x<<3)+(c^48),c=getchar();
    return x*f;
}
#define N 2000010
int n,m,c,a[N],nxt[N],p[N],tree[N],ans[N],flag[N];
struct data
{
    int l,r,i;
    bool operator <(const data&a) const
    {
        return l<a.l;
    }
}q[N];
void add(int k,int x){while (k<=n) tree[k]+=x,k+=k&-k;}
int sum(int k){int s=0;while (k) s+=tree[k],k-=k&-k;return s;}
int main()
{
#ifndef ONLINE_JUDGE
    freopen("bzoj2743.in","r",stdin);
    freopen("bzoj2743.out","w",stdout);
    const char LL[]="%I64d\n";
#else
    const char LL[]="%lld\n";
#endif
    n=read(),c=read(),m=read();
    for (int i=1;i<=n;i++)
    {
        a[i]=read(),nxt[p[a[i]]]=i,p[a[i]]=i;
        if (flag[a[i]]==1) add(i,1),flag[a[i]]=2;
        else if (!flag[a[i]]) flag[a[i]]=1;
    }
    for (int i=1;i<=n+1;i++) if (!nxt[i]) nxt[i]=n+1;
    for (int i=1;i<=m;i++) q[i].l=read(),q[i].r=read(),q[i].i=i;
    sort(q+1,q+m+1);
    q[0].l=1;
    for (int i=1;i<=m;i++)
    {
        while (q[i].l>q[i-1].l) 
        add(nxt[q[i-1].l],-1),add(nxt[nxt[q[i-1].l]],1),q[i-1].l++;
        ans[q[i].i]=sum(q[i].r)-sum(q[i].l-1);
    }
    for (int i=1;i<=m;i++) printf("%d\n",ans[i]);
    return 0;
}