bzoj 1803(DFS序+主席树)

传送门
题解:按DFS序下标建主席树,然后就不用说什么了吧。。。
注意:如果询问/插入的[l,r]是[0,1e9]的话,就需要多开数组(MAXN*32),而且效率不太高,所以建议先离散化,这样就可以在[0,MAXN]的范围内询问/插入,而且时间可以缩短10%左右。

#include
using namespace std;
const int MAXN=1e5+4;
int n,q,a[MAXN],b[MAXN],in[MAXN],out[MAXN],tot=0;
int root[MAXN]={0},lc[MAXN*18],rc[MAXN*18],sum[MAXN*18],rk[MAXN*18],tim=0;
int head[MAXN],edge=0;
struct EDGE {
    int v,nxt;
}e[MAXN<<1];
inline 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*10+c-'0',c=getchar();
    return x;
}
inline void adde(int u,int v) {
    e[edge].nxt=head[u],e[edge].v=v,head[u]=edge++;
    e[edge].nxt=head[v],e[edge].v=u,head[v]=edge++;
}
void insert(int pre,int &rt,int l,int r,int val,int id) {
    rt=++tim,sum[rt]=sum[pre]+1;
    if (l==r) {
        rk[rt]=id;
        return ;
    }
    lc[rt]=lc[pre],rc[rt]=rc[pre];
    int mid=(l+r)>>1;
    if (val<=mid) insert(lc[pre],lc[rt],l,mid,val,id);
    else insert(rc[pre],rc[rt],mid+1,r,val,id);
}
int query(int pre,int now,int l,int r,int k) {
    if (l==r) return rk[now];
    int mid=(l+r)>>1,temp=sum[lc[now]]-sum[lc[pre]];
    if (k<=temp) return query(lc[pre],lc[now],l,mid,k);
    else return query(rc[pre],rc[now],mid+1,r,k-temp);
}
void dfs(int p,int fa) {
    in[p]=++tot;
    insert(root[in[p]-1],root[in[p]],0,MAXN,a[p],p);
    for (int i=head[p];~i;i=e[i].nxt) {
        int v=e[i].v;
        if (v^fa)
            dfs(v,p);
    }
    out[p]=tot;
}
int main() {
//  freopen("bzoj 1803.in","r",stdin);
    memset(head,-1,sizeof(head));
    n=read();
    for (register int i=1;i<=n;++i) b[i]=a[i]=read();
    sort(b+1,b+n+1);
    int m=unique(b+1,b+n+1)-b-1;
    for (register int i=1;i<=n;++i) a[i]=lower_bound(b+1,b+m+1,a[i])-b;
    for (register int i=1;iint u=read(),v=read();
        adde(u,v);
    }
    dfs(1,0);
    q=read();
    for (register int r=0;r<q;++r) {
        int pos=read(),k=read();
        printf("%d\n",query(root[in[pos]-1],root[out[pos]],0,MAXN,k));
    }
    return 0;
}

你可能感兴趣的:(数据结构)