2588: Spoj 10628. Count on a tree

来源：互联网发布：张正隆雪白血红知乎编辑：程序博客网时间：2024/05/21 17:17

竟然1A了，完了这个月的RP用光了。

树上主席树（听起来好奇怪）。

在每个节点建一颗主席树，保存其到根节点的权值。

然后询问就是sum(u)+sum(v)-sum(lca(u,v))-sum(fa(lca(u,v)))，至于这个前缀和是什么，大概就是线性序列的那个前缀差不多，不过搞成根节点到当前节点链上的前缀和了。

#include<iostream>#include<cstdio>#include<cstring>#include<algorithm>using namespace std;inline int read(){int x=0,f=1;char ch;while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}return x*f;}const int N=100010;struct Node{int l,r,w;}tr[N*20];int F[N*2],dep[N*2],st[N*2][20],cnt,pos[N],n;int root[N],sz,dfn[N],dfs_clock,fa[N],rk[N];int a[N],b[N],rank[N];struct Edge{int to,next;}e[N*2];int head[N],edge;void ins(int u,int v){edge++;e[edge].to=v;e[edge].next=head[u];head[u]=edge;}inline bool cmp(int i,int j){return a[i]<a[j];}void dfs(int u,int depth){F[++cnt]=u;dep[cnt]=depth;pos[u]=cnt;st[cnt][0]=cnt;dfn[u]=++dfs_clock;rk[dfs_clock]=u;for(int i=head[u];i;i=e[i].next){int v=e[i].to;if(v==fa[u])continue;fa[v]=u;dfs(v,depth+1);F[++cnt]=u;dep[cnt]=depth;st[cnt][0]=cnt;}}void rmq_init(){for(int j=1;(1<<j)<2*n;j++)for(int i=1;i+(1<<j)-1<2*n;i++){int l=st[i][j-1],r=st[i+(1<<(j-1))][j-1];if(dep[l]<dep[r])st[i][j]=l;else st[i][j]=r;}}int rmq(int l,int r){int k=0;while(1<<(k+1)<=r-l+1)k++;l=st[l][k];r=st[r-(1<<k)+1][k];if(dep[l]<dep[r])return l;return r;}int lca(int u,int v){int l=pos[u],r=pos[v];if(l>r)swap(l,r);return F[rmq(l,r)];}void update(int &i,int l,int r,int x){tr[++sz]=tr[i];i=sz;tr[i].w++;if(l==r)return;int mid=l+r>>1;if(x<=mid)update(tr[i].l,l,mid,x);else update(tr[i].r,mid+1,r,x);}int query(int a,int b,int A,int B,int l,int r,int k){if(l==r)return l;int tmp=tr[tr[a].l].w+tr[tr[b].l].w-tr[tr[A].l].w-tr[tr[B].l].w;int mid=l+r>>1;if(k<=tmp)return query(tr[a].l,tr[b].l,tr[A].l,tr[B].l,l,mid,k);else return query(tr[a].r,tr[b].r,tr[A].r,tr[B].r,mid+1,r,k-tmp);}int main(){int m;scanf("%d%d",&n,&m);for(int i=1;i<=n;i++)scanf("%d",&a[i]),rank[i]=i;sort(rank+1,rank+1+n,cmp);for(int i=1;i<=n;i++)b[rank[i]]=i;int u,v;for(int i=1;i<n;i++){scanf("%d%d",&u,&v);ins(u,v);ins(v,u);}int rt=1+n>>1;dfs(rt,0);rmq_init();for(int i=1;i<=n;i++){int t=rk[i];root[i]=root[dfn[fa[t]]];update(root[i],1,n,b[t]);}int k,last=0;for(int i=1;i<=m;i++){scanf("%d%d%d",&u,&v,&k);u^=last;int w=lca(u,v);last=a[rank[query(root[dfn[u]],root[dfn[v]],root[dfn[w]],root[dfn[fa[w]]],1,n,k)]];printf("%d",last);if(i!=m)printf("\n");}return 0;}

为什么我的代码都这么短（斜眼）

0 0