bzoj 1878: [SDOI2009]HH的项链

好像只有我sb写了巨长巨慢的线段树。。。
将询问离线，记录每个颜色首先出现的位置和下一个出现的位置，维护颜色数前缀和，将询问按左端点排序。从1到n枚举，如果左端点是当前位置则查询sum[r]-sum[l-1] (sum[l-1]似乎等于0…)，最后对i到nex[i] 减一（就是忽略当前位置）。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int read()
{
    char ch=getchar();int f=0;
    while(ch<'0'||ch>'9') ch=getchar();
    while(ch>='0'&&ch<='9'){f=f*10+ch-'0';ch=getchar();}
    return f;
}
int n,m,a[50005],nex[50005],vis[1000005],ans[200005],tot=1,b[50005];bool first[50005];
struct data
{
    int l;
    int r;
    int id;
}c[200005];
int cmp(data x,data y)
{
    return x.l<y.l;
}
struct segtree
{
    int ls;
    int rs;
    long long sum;
    int mark;
}tree[200005];
void build(int p,int l,int r)
{
    if(l==r)
    {
        tree[p].sum=a[l];
        return;
    }
    int mid=l+r>>1;
    tree[p].ls=++tot;
    tree[p].rs=++tot;
    build(tree[p].ls,l,mid);
    build(tree[p].rs,mid+1,r);
    tree[p].sum=tree[tree[p].ls].sum+tree[tree[p].rs].sum;
}
void modify(int p,int x,int y,int l,int r,int k)
{
    int mid=l+r>>1;
    if(x==l&&r==y)
    {
        tree[p].mark+=k;
        tree[p].sum+=(r-l+1)*k;
        return;
    }
    tree[tree[p].ls].mark+=tree[p].mark;
    tree[tree[p].rs].mark+=tree[p].mark;
    tree[tree[p].ls].sum+=1LL*(mid-l+1)*tree[p].mark;
    tree[tree[p].rs].sum+=1LL*(r-mid)*tree[p].mark;
    tree[p].mark=0;
    if(y<=mid)
    modify(tree[p].ls,x,y,l,mid,k);
    else if(x>mid)
    modify(tree[p].rs,x,y,mid+1,r,k);
    else
    {
        modify(tree[p].ls,x,mid,l,mid,k);
        modify(tree[p].rs,mid+1,y,mid+1,r,k);
    }
    tree[p].sum=tree[tree[p].ls].sum+tree[tree[p].rs].sum;
}
int query(int p,int x,int y,int l,int r)
{

    int mid=l+r>>1;
    if(x==l&&r==y)
    {
        return tree[p].sum;
    }
    tree[tree[p].ls].mark+=tree[p].mark;
    tree[tree[p].rs].mark+=tree[p].mark;
    tree[tree[p].ls].sum+=1LL*(mid-l+1)*tree[p].mark;
    tree[tree[p].rs].sum+=1LL*(r-mid)*tree[p].mark;
    tree[p].mark=0;
    if(y<=mid)
    return query(tree[p].ls,x,y,l,mid);
    if(x>mid)
    return query(tree[p].rs,x,y,mid+1,r);
    return query(tree[p].ls,x,mid,l,mid)+query(tree[p].rs,mid+1,y,mid+1,r);
}
int main()
{
    n=read();
    for(int i=1;i<=n;i++)
    b[i]=read();
    for(int i=1;i<=n;i++)
    {
        if(!vis[b[i]])
        {
            vis[b[i]]=i;
            first[i]=1;
        }
        else
        {
            nex[vis[b[i]]]=i;
            vis[b[i]]=i;
        }
    }
    m=read();
    for(int i=1;i<=m;i++)
    {
        c[i].l=read();c[i].r=read();
        c[i].id=i;
    }
    sort(c+1,c+m+1,cmp);
    int now=1;
    for(int i=1;i<=n;i++)
    {
        a[i]=a[i-1]+(first[i]?1:0);
    }
    build(1,0,n);
    for(int i=1;i<=n;i++)
    {
        while(c[now].l==i)
        {
            ans[c[now].id]=query(1,c[now].r,c[now].r,0,n);
            now++;
        }
        if(nex[i])
        modify(1,i,nex[i]-1,0,n,-1);
        else
        modify(1,i,n,0,n,-1);
    }
    for(int i=1;i<=m;i++)
    {
        printf("%d\n",ans[i]);
    }
}