POJ 2763 Housewife Wind

连通图任意两点间路径唯一，那么肯定是树，不然就会有圈

 

如果没有动态的修改边权的操作，就是简单的LCA问题了

 

现在加入动态操作后，如果修改(u,v)这条边，且不妨设u是v的父节点，那么以v为根的子树中所有节点的dis值都会改变

我们可以用树状数组来维护这个改变的del值

 

首先，将LCA转化成RMQ问题，这样我们求得了一个dfs顺序的欧拉序列，和对应的深度序列，还有一个每个节点第一次被访问的时间h[i]，我们再多记录一个每个节点它的子树被访问完时的时间end[i]，这样，修改(u,v)这条边，且u是v的父节点的话，处于[h[v],end[v]]这个时间段的所有节点的dis值都改变了del，用树状数组维护就行,update(h[v],del),update(end[v]+1,-del)，一开始end[v]后面没有+1，wa了好久=。=

 

代码：

#include<iostream> #include<memory.h> #include<string> #include<cstdio> #include<algorithm> #include<math.h> #include<stack> #include<queue> #include<vector> #include<map> using namespace std; const int MAX=100005; struct node { int u,v,w,next; }g[MAX*5]; int adj[MAX],dis[MAX],f[MAX*2],h[MAX],level[MAX*2],end[MAX]; int c[MAX*2]; int M[MAX*2][17]; int n,q,s,e,Time; void dfs(int u,int w,int d) { f[++Time]=u; level[Time]=d; h[u]=Time; dis[u]=w; for(int i=adj[u];i!=-1;i=g[i].next) { if(!h[g[i].v]) { dfs(g[i].v,w+g[i].w,d+1); f[++Time]=u; level[Time]=d; } } end[u]=Time; } void preprocess() { int i,j; for(i=1;i<=Time;i++) M[i][0]=i; for(j=1;(1<<j)<=Time;j++) { for(i=1;i+(1<<j)-1<=Time;i++) { if(level[M[i][j-1]]<level[M[i+(1<<(j-1))][j-1]]) M[i][j]=M[i][j-1]; else M[i][j]=M[i+(1<<(j-1))][j-1]; } } } void update(int idx,int val) { while(idx<=Time) { c[idx]+=val; idx+=idx&(-idx); } } int read(int idx) { int sum=0; while(idx>0) { sum+=c[idx]; idx-=idx&(-idx); } return sum; } int main() { int i,j,k,w,t,ancestor; while(scanf("%d%d%d",&n,&q,&s)!=EOF) { memset(adj,-1,sizeof(adj)); memset(h,0,sizeof(h)); memset(c,0,sizeof(c)); e=Time=0; for(k=1;k<n;k++) { scanf("%d%d%d",&i,&j,&w); g[e].u=i; g[e].v=j; g[e].w=w; g[e].next=adj[i]; adj[i]=e++; g[e].u=j; g[e].v=i; g[e].w=w; g[e].next=adj[j]; adj[j]=e++; } dfs(s,0,0); preprocess(); while(q--) { scanf("%d",&k); if(k==0) { scanf("%d",&i); int ii=min(h[i],h[s]); int jj=max(h[i],h[s]); t=(int)(log((jj-ii+1)*1.0)/log(2.0)); if(level[M[ii][t]]<level[M[jj-(1<<t)+1][t]]) ancestor=f[M[ii][t]]; else ancestor=f[M[jj-(1<<t)+1][t]]; //cout<<"ancestor="<<ancestor<<endl; printf("%d/n",dis[i]+dis[s]-2*dis[ancestor]+read(h[i])+read(h[s])-2*read(h[ancestor])); s=i; } else { scanf("%d%d",&i,&j); int del=j-g[(i-1)*2].w; int u=g[(i-1)*2].u; int v=g[(i-1)*2].v; g[(i-1)*2].w=j; if(h[v]<h[u]) { update(h[u],del); update(end[u]+1,-del); } else { update(h[v],del); update(end[v]+1,-del); } } } } return 0; }