50pts求调
查看原帖
50pts求调
254491
橙橙like海绵楼主2022/12/10 15:29

错误点#2 #7 #8 #9 #10

#include<bits/stdc++.h>
#define ll long long 
#define int long long
using namespace std;
const int N=5e5+10;
const int inf=0x3f3f3f3f;
int n,m;
ll sum[N],add[N];
int dep[N],top[N],sz[N],son[N],fa[N];
int dfn[N],tot;
int h[N],cnt;
struct node{
	int nxt,to;
}e[N<<2];
ll read(){
	ll s=0,w=1;
	char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-') w=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){s=s*10+ch-'0';ch=getchar();}
	return s*w;
}
void adde(int u,int v){
	e[++cnt].nxt=h[u];
	e[cnt].to=v;
	h[u]=cnt;
}
void dfs1(int u,int f){
	fa[u]=f;dep[u]=dep[f]+1;sz[u]=1;
	for(int i=h[u];i;i=e[i].nxt){
		int v=e[i].to;
		if(v==f) continue;
		dfs1(v,u);
		sz[u]+=sz[v];
		if(sz[v]>sz[son[u]]) son[u]=v;
	}
}
void dfs2(int u,int topf){
	top[u]=topf;dfn[u]=++tot;
	if(!son[u]) return ;
	dfs2(son[u],topf);
	for(int i=h[u];i;i=e[i].nxt){
		int v=e[i].to;
		if(v==son[u]||v==fa[u]) continue;
		dfs2(v,v);
	}
}
void Add(int k,int l,int r,ll v){
	add[k]+=1ll*v;
	sum[k]+=1ll*(r-l+1)*v;
}
void push_up(int k){sum[k]=1ll*sum[k<<1]+1ll*sum[k<<1|1];}
void push_down(int k,int l,int r,int mid){
	if(!add[k]) return ;
	Add(k<<1,l,mid,add[k]);
	Add(k<<1|1,mid+1,r,add[k]);
	add[k]=0;
}
void build(int k,int l,int r){
	if(l==r){
		sum[k]=0;
		return ;
	}
	int mid=(l+r)>>1;
	build(k<<1,l,mid);
	build(k<<1|1,mid+1,r);
	push_up(k);
}
void change(int k,int l,int r,int x,int y,ll v){
	if(x<=l&&r<=y){Add(k,l,r,v);return ;}
	int mid=(l+r)>>1;
	push_down(k,l,r,mid);
	if(x<=mid) change(k<<1,l,mid,x,y,v);
	if(mid<y) change(k<<1|1,mid+1,r,x,y,v);
	push_up(k);
}
ll query(int k,int l,int r,int x,int y){
	if(x<=l&&r<=y){return sum[k];}
	int mid=(l+r)>>1;
	push_down(k,l,r,mid);
	ll res=0;
	if(x<=mid) res+=query(k<<1,l,mid,x,y);
	if(mid<y) res+=query(k<<1|1,mid+1,r,x,y);
	return res;
}
void crange(int u,int v,int k){
	if(top[u]!=top[v]){
		if(dep[top[u]]<dep[top[v]]) swap(u,v);
		change(1,1,n,dfn[top[u]],dfn[u],k);
		u=fa[top[u]];
	}
	if(dep[u]>dep[v]) swap(u,v);
	change(1,1,n,dfn[u],dfn[v],k);
}
signed main(){
	n=read();
	int u,v,d;
	for(int i=1;i<n;i++){
		u=read()+1;v=read()+1;
		adde(u,v);adde(v,u);
	}
	m=read();
	dfs1(1,0);
	dfs2(1,1);
	build(1,1,n);
	char ch;
	while(m--){
		scanf("%s",&ch);
		if(ch=='A'){
			u=read()+1;v=read()+1;d=read();
			crange(u,v,d);
		}
		else{
			u=read()+1;
			printf("%lld\n",query(1,1,n,dfn[u],dfn[u]+sz[u]-1));			
		} 
	}
	return 0;
}
2022/12/10 15:29
加载中...