题意:一棵树。有两种操作,对两个点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;
}