BZOJ1878: [SDOI2009]HH的项链[树状数组+离线 | 主席树]

 

题意： 询问区间不同种类颜色数

---

 [2016-11-15]

离线好厉害

对于每一个区间询问，一个数只考虑一次，那么考虑他最后出现的一次

将询问按r排序

从1到n扫描,用树状数组维护一个位置应不应该考虑(记不记入答案)，让每种颜色最后一个出现位置贡献

last[x]是x上一个出现的位置，每到一个a[i]，last位置-1，i位置+1，并更新last

然后对于所有r==i的询问计算答案

 

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
const int N=5e4+5,M=2e5+5,INF=1e6+5;
inline int read(){
    char c=getchar();int x=0,f=1;
    while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}
    while(c>='0'&&c<='9'){x=x*10+c-'0';c=getchar();}
    return x*f;
}
int n,m,a[N],last[INF];
struct data{
    int l,r,id,ans;
    bool operator <(const data &rhs)const{return r<rhs.r;}
}q[M];
inline bool cmp(data a,data b){return a.id<b.id;}
int c[N];
inline int lowbit(int x){return x&-x;}
inline void add(int p,int v){
    for(;p<=n;p+=lowbit(p)) c[p]+=v;
}
inline int sum(int p){
    int res=0;
    for(;p>0;p-=lowbit(p)) res+=c[p];
    return res;
}
int main(){
    n=read();
    for(int i=1;i<=n;i++) a[i]=read();
    
    m=read();
    for(int i=1;i<=m;i++) q[i].l=read(),q[i].r=read(),q[i].id=i;
    sort(q+1,q+1+m);
    int p=1;
    for(int i=1;i<=n;i++){
        if(last[a[i]]) add(last[a[i]],-1);
        add(i,1); //printf("%d %d %d  %d\n",i,a[i],last[a[i]],p);
        last[a[i]]=i;
        while(q[p].r==i) q[p].ans=sum(q[p].r)-sum(q[p].l-1),p++;
    }
    sort(q+1,q+1+m,cmp);
    for(int i=1;i<=m;i++) printf("%d\n",q[i].ans);
}

 

---

[2017-01-14]

用主席树做起来直观多了，特别在做了BZOJ3514之后这就是水题啊

last[i]表示i位置的数上一个出现位置，查询区间中last[i]<l的个数，序列建主席树，last权值线段树上就是[0...l-1]的权值和啊

注意权值从0开始注意权值从0开始注意权值从0开始

 

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define lc(x) t[x].l
#define rc(x) t[x].r
const int N=2e5+5,MX=1e6+5;
inline int read(){
    char c=getchar();int x=0,f=1;
    while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}
    while(c>='0'&&c<='9'){x=x*10+c-'0';c=getchar();}
    return x*f;
}
int n,Q,x,ql,qr,last[N],pos[MX];
struct node{
    int l,r,size;
}t[N*20];
int sz,root[N];
void ins(int &x,int l,int r,int p){
    t[++sz]=t[x];x=sz;
    t[x].size++;
    if(l==r) return;
    int mid=(l+r)>>1;
    if(p<=mid) ins(t[x].l,l,mid,p);
    else ins(t[x].r,mid+1,r,p);
}
int que(int x,int y,int l,int r,int ql,int qr){
    if(ql<=l&&r<=qr) return t[y].size-t[x].size;
    else{
        int mid=(l+r)>>1,ans=0;
        if(ql<=mid) ans+=que(lc(x),lc(y),l,mid,ql,qr);
        if(mid<qr) ans+=que(rc(x),rc(y),mid+1,r,ql,qr);
        return ans;
    }
} 

int main(){
    //freopen("in.txt","r",stdin);
    n=read();
    for(int i=1;i<=n;i++){
        x=read();
        last[i]=pos[x];
        pos[x]=i;
    }
    for(int i=1;i<=n;i++) root[i]=root[i-1],ins(root[i],0,n,last[i]);//,printf("last %d %d\n",i,last[i]);
    Q=read();
    while(Q--){
        ql=read();qr=read();
        printf("%d\n",que(root[ql-1],root[qr],0,n,0,ql-1));
    }
}