bzoj2743 [HEOI2012]采花

题目

同样的，经典的离线处理，线段树。
把前一个标记为1，前一个的前一个标记为0。

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<string>
#include<cstring>
#define MAXN 1000000
using namespace std;
int m,n,c;
struct tree{
    int l;
    int r;
    int sum;
};tree T[MAXN*4+1];
struct qus{
    int id;
    int l;
    int r;
};qus B[MAXN+1];
int A[MAXN+1];
int ans[MAXN+1];
int pre[MAXN+1];
bool E[MAXN+1];
int ind[MAXN+1],num;
bool cmp(const qus &A,const qus &B)
{
    if(A.r!=B.r)return A.r<B.r;
    return A.l<B.l;
}
void build(int rt,int l,int r)
{
    T[rt].l=l,T[rt].r=r;T[rt].sum=0;
    if(l==r)return;
    int mid=(l+r)/2;
    build(rt*2,l,mid),build(rt*2+1,mid+1,r);
}
void modify(int rt,int pos,int x)
{
    if(!pos)return;
    if(T[rt].l==T[rt].r&&T[rt].l==pos)
    {
        T[rt].sum=x;
        return;
    }
    int mid=(T[rt].l+T[rt].r)/2;
    if(mid>=pos)modify(rt*2,pos,x);
    else modify(rt*2+1,pos,x);
    T[rt].sum=T[rt*2].sum+T[rt*2+1].sum;
}
int answer(int rt,int l,int r)
{
    if(T[rt].l==l&&T[rt].r==r)
        return T[rt].sum;
    int mid=(T[rt].l+T[rt].r)/2;
    if(mid>=r)return answer(rt*2,l,r);
    else if(mid<l)return answer(rt*2+1,l,r);
    else return answer(rt*2,l,mid)+answer(rt*2+1,mid+1,r);
}
int main()
{
    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",&B[i].l,&B[i].r),B[i].id=i;
    build(1,1,n);
    sort(B+1,B+m+1,cmp);
    memset(E,false,sizeof(E));
    for(int i=1;i<=n;i++)
    {
        if(E[A[i]])
            pre[i]=ind[A[i]];
        E[A[i]]=true;
        ind[A[i]]=i;
    }
    num=1;
    for(int i=1;i<=n;i++)
    {
        modify(1,pre[i],1);
        modify(1,pre[pre[i]],0);
        if(i==B[num].r)
        {
            int j;
            for(j=num;B[j].r==i&&j<=m;j++)
                ans[B[j].id]=answer(1,B[j].l,B[j].r);
            num=j;
        }
    }
    for(int i=1;i<=m;i++)
        printf("%d\n",ans[i]);
    return 0;
 }