题意:一棵树。有两种操作,对两个点uv路径上的所有点加权和对路径上的所有边加权。最后输出所有点的权和所有边的权。
思路:树链剖分+树状数组。感觉这个题出出来就是为了报复社会的,我一共跪了25次。。又是扩栈又是输入挂又是内联,最后只有用c++交能过,g++爆栈。不过话说回来这个题也挺模版的,都是树链剖分的基本操作。因为没学过对边的处理,我把边绑在点上了,更改查询边时,去操作边向远离根方向连接的点。另外树状数组不能区间更新,所以每次更新在起点加上k,终点+1减去k。
#pragma comment(linker, "/STACK:1024000000,1024000000") #include<iostream> #include<cmath> #include<queue> #include<map> #include<set> #include<vector> #include<algorithm> #include<string.h> #include<cstdio> using namespace std; #define maxn 100010 #define ll long long int head[maxn]; int next[maxn*2]; int to[maxn*2]; int e2v[maxn]; //第i条边向下接着那个哪个点 int edgeu[maxn]; int edgev[maxn]; int siz[maxn]; //以i为根的子树的节点数 int dep[maxn]; //节点深度 int top[maxn]; //所在链的顶端节点 int fa[maxn]; //父节点 int son[maxn]; //重儿子 int w[maxn]; //节点在线段树中的位置 int w_1[maxn]; //上面的反函数 bool vis[maxn]; int pos; int end; int n,m,p; ll cv[maxn]; ll ce[maxn]; ll ans1[maxn]; ll ans2[maxn]; void scanf_(int &num){ char in; bool neg=false; while(((in=getchar()) > '9' || in<'0') && in!='-') ; if(in=='-') { neg=true; while((in=getchar()) >'9' || in<'0'); } num=in-'0'; while(in=getchar(),in>='0'&&in<='9') num*=10,num+=in-'0'; if(neg) num=0-num; } void init(){ memset(siz,0,sizeof(siz)); memset(top,0,sizeof(top)); memset(son,0,sizeof(son)); memset(head,-1,sizeof(head)); memset(cv,0,sizeof(cv)); memset(ce,0,sizeof(ce)); pos=0; end=0; } inline void addedge(int u,int v){ to[end]=v,next[end]=head[u],head[u]=end++; to[end]=u,next[end]=head[v],head[v]=end++; } void dfs(int x,int pre){ vis[x]=1; dep[x]=dep[pre]+1; fa[x]=pre; for(int i=head[x];i!=-1;i=next[i]){ if(to[i]==pre)continue; if(vis[to[i]])continue; dfs(to[i],x); siz[x]+=siz[to[i]]; if(son[x]==0)son[x]=to[i]; else if(siz[son[x]]<siz[to[i]])son[x]=to[i]; } siz[x]++; } void dfs2(int x,int pre){ vis[x]=1; pos++; w[x]=pos; w_1[pos]=x; if(son[fa[x]]==x)top[x]=top[fa[x]]; else top[x]=x; //先递归重儿子 if(son[x]==0)return; dfs2(son[x],x); for(int i=head[x];i!=-1;i=next[i]){ if(to[i]==pre)continue; if(to[i]==son[x])continue; if(vis[to[i]])continue; dfs2(to[i],x); } } inline int lowbit(int x){ return x&(-x); } ll sum(int end,int type){ if(type==1){ ll sum=0; while(end){ sum+=cv[end]; end-=lowbit(end); } return sum; }else{ ll sum=0; while(end){ sum+=ce[end]; end-=lowbit(end); } return sum; } } void update(int pos,ll num,int type){ if(type==1){ while(pos<=n+1){ cv[pos]+=num; pos+=lowbit(pos); } }else{ while(pos<=n+1){ ce[pos]+=num; pos+=lowbit(pos); } } } int main(){ int t; cin>>t; for(int cas=1;cas<=t;cas++){ cin>>n>>m; init(); for(int i=1;i<n;i++){ scanf_(edgeu[i]); scanf_(edgev[i]); addedge(edgeu[i],edgev[i]); } memset(vis,0,sizeof(vis)); dfs(1,0); memset(vis,0,sizeof(vis)); dfs2(1,0); for(int i=1;i<n;i++){ if(fa[edgeu[i]]==edgev[i]){ e2v[i]=edgeu[i]; }else{ e2v[i]=edgev[i]; } } char op[8]; for(int i=1;i<=m;i++){ scanf("%s",op); int u,v,k; scanf_(u); scanf_(v); scanf_(k); if(op[3]=='1'){ while(top[u]!=top[v]){ if(dep[top[u]]<dep[top[v]])swap(u,v); update(w[top[u]],k,1); update(w[u]+1,-k,1); u=fa[top[u]]; } if(w[u]>w[v])swap(u,v); update(w[u],k,1); update(w[v]+1,-k,1); }else{ while(top[u]!=top[v]){ if(dep[top[u]]<dep[top[v]])swap(u,v); update(w[top[u]],k,2); update(w[u]+1,-k,2); u=fa[top[u]]; } if(w[u]>w[v])swap(u,v); update(w[u]+1,k,2); //注意少更新一个点 update(w[v]+1,-k,2); } } printf("Case #%d:\n",cas); ll t=0; for(int i=1;i<=n;i++){ t+=sum(i,1); t-=sum(i-1,1); ans1[i]=t; } t=0; for(int i=1;i<=n;i++){ t+=sum(i,2); t-=sum(i-1,2); ans2[i]=t; } for(int i=1;i<=n;i++){ printf("%I64d",ans1[w[i]]); if(i!=n)printf(" "); } printf("\n"); for(int i=1;i<n;i++){ printf("%I64d",ans2[w[e2v[i]]]); if(i!=n-1)printf(" "); } printf("\n"); } return 0; }