HDU 5044 Tree 树链剖分+差分

题意:n点m条边的带权树,点边都带权，Q次操作.
op1:将u,v路径上所有点的权值加k.
op2:将u,v路径上所有边的权值加k.


n,m<=1e5,输出所有操作过后的每个点的权值和每条边的权值.

操作为区间+,查询只要最后一次,可以不用线段树,破树成链以后,用两个数组差分一下就好了.

#pragma comment(linker, "/STACK:1024000000,1024000000") 
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=5e5+20;
const int M=6e5+20;
const int inf=0x3f3f3f3f;
int n,Q,cnt,sz;
ll w1[N],w2[N];
int dep[N],size[N],head[N],fa[N];
int pos[N],bl[N],son[N];
struct edge{int to,next,w;}e[M];
struct Edge{int u,v,w;}a[M];
void add_edge(int u,int v,int w){e[++cnt].to=v,e[cnt].w=w,e[cnt].next=head[u],head[u]=cnt;}
void init()
{
	sz=cnt=0;
	memset(w1,0,sizeof(w1));
	memset(w2,0,sizeof(w2));
	memset(head,-1,sizeof(head));
	scanf("%d%d",&n,&Q);
	int x,y,w;
	for(int i=1;i<=n-1;i++)
	{
		scanf("%d%d",&x,&y);
		w=0;
		add_edge(x,y,w),add_edge(y,x,w);
		a[i].u=x,a[i].v=y,a[i].w=w;
	}
}
void dfs(int u)
{
	size[u]=1;
	for(int i=head[u];i!=-1;i=e[i].next)
	{
		int v=e[i].to;
		if(v==fa[u])	continue;
		w2[v]=e[i].w;//w[v]:v-fa[v]'scost 
		dep[v]=dep[u]+1,fa[v]=u;
		dfs(v);
		size[u]+=size[v];
	}
}
void dfs(int x,int chain)
{
	int k=0;sz++;
	pos[x]=sz;//time
	bl[x]=chain;
	for(int i=head[x];i!=-1;i=e[i].next)//find heavy son
	{
		int v=e[i].to;
		if(dep[v]>dep[x]&&size[v]>size[k])
			k=v;
	}
	son[x]=k;
	if(k==0){son[x]=x;return;}
	dfs(k,chain);
	for(int i=head[x];i!=-1;i=e[i].next)
		if(dep[e[i].to]>dep[x]&&k!=e[i].to)
			dfs(e[i].to,e[i].to);//another link
}

//
void update(int l,int r,int val,int op)
{
	if(op==1)
		w1[l]+=val,w1[r+1]-=val;
	else
		w2[l]+=val,w2[r+1]-=val;
}
int solve1(int x,int y,int val)
{
	while(bl[x]!=bl[y])
	{
		if(dep[bl[x]]<dep[bl[y]]) swap(x,y);
		update(pos[bl[x]],pos[x],val,1);
		x=fa[bl[x]];
	}
	if(pos[x]>pos[y]) swap(x,y);
	update(pos[x],pos[y],val,1);
}
int solve2(int x,int y,int val)
{
	while(bl[x]!=bl[y])
	{
		if(dep[bl[x]]<dep[bl[y]]) swap(x,y);
		update(pos[bl[x]],pos[x],val,2);
		x=fa[bl[x]];
	}
	if(pos[x]>pos[y]) swap(x,y);
	update(pos[son[x]],pos[y],val,2);//#
}
void solve()
{
	int x,y,z;
	char op[10];
	while(Q--)
	{
		scanf("%s",op);
		scanf("%d%d%d",&x,&y,&z);
		if(op[3]=='1')
			solve1(x,y,z);
		else
			solve2(x,y,z);
	}
	for(int i=2;i<=n;i++)
		w1[i]+=w1[i-1],w2[i]+=w2[i-1];
		
	bool flag=false;
	for(int i=1;i<=n;i++)
	{
		if(flag)
			printf(" ");
		else
			flag=true;
		printf("%I64d",w1[pos[i]]);
			
	}
	flag=false;
	printf("\n");
	for(int i=1;i<n;i++)
	{
		int u=a[i].u,v=a[i].v;
		if(dep[u]<dep[v])
			swap(u,v);
		if(flag)
			printf(" ");
		else
			flag=true;
		printf("%I64d",w2[pos[u]]);
	}
	printf("\n");
}
int main()
{
	int T,cas=0;
	scanf("%d",&T);
	while(T--)
	{
		init();
		dfs(1);
		dfs(1,1);
		printf("Case #%d:\n",++cas);
		solve();
	}
	return 0;
}