为什么全 WA 了 QWQ
查看原帖
为什么全 WA 了 QWQ
661044
_saltFish_楼主2022/8/9 16:54
#include<iostream>
#define ll long long
using namespace std;
const int N(1e5+5);
int n,m;
int h[N],to[N<<1],nxt[N<<1],tot;
inline void egde(int u,int v){
	nxt[++tot]=h[u];to[h[u]=tot]=v;
	nxt[++tot]=h[v];to[h[v]=tot]=u;
}
int fa[N],dep[N],siz[N],son[N],dfn[N],rnk[N],top[N],cnt;
void cut(int u,int f){
	fa[u]=f;dep[u]=dep[f]+1;siz[u]=1;
	for(int i=h[u];i;i=nxt[i]){
		int v=to[i];
		if(v==f) continue;
		cut(v,u);
		siz[u]+=siz[v];
		if(siz[son[u]]<siz[v]) son[u]=v;
	}
}
void dfs(int u,int Top){
	top[u]=Top;dfn[++cnt]=u;rnk[u]=cnt;
	if(!son[u]) return ;
	dfs(son[u],Top);
	for(int i=h[u];i;i=nxt[i])
		if(to[i]!=fa[u]&&to[i]!=son[u])
			dfs(to[i],to[i]);
}
#define ls (u<<1)
#define rs (u<<1|1)
#define mid (l+r>>1)
ll sum[N<<2],add[N<<2],val[N];
void build(int u,int l,int r){
	if(l==r){
		sum[u]=val[dfn[l]];
		return ;
	}
	build(ls,l,mid),build(rs,mid+1,r);
	sum[u]=sum[ls]+sum[rs];
}
void pushdown(int u,int len){
	if(add[u]){
		sum[ls]+=1ll*(len-(len>>1))*add[u];
		sum[rs]+=1ll*(len>>1)*add[u];
		add[ls]+=add[u];add[rs]+=add[u];
		add[u]=0;
	}
}
void change(int u,int l,int r,int L,int R,ll k){
	if(L<=l&&r<=R){
		sum[u]+=1ll*(r-l+1)*k;add[u]+=k;
		return ;
	}
	pushdown(u,r-l+1);
	if(L<=mid) change(ls,l,mid,L,R,k);
	if(R>mid) change(rs,mid+1,r,L,R,k);
	sum[u]=sum[ls]+sum[rs];
}
void update(int u,int l,int r,int p,ll k){
	if(l==r){
		sum[u]+=k;return ;
	}
	pushdown(u,r-l+1);
	if(p<=mid) update(ls,l,mid,p,k);
	else update(rs,mid+1,r,p,k);
	sum[u]=sum[ls]+sum[rs];
}
ll query(int u,int l,int r,int L,int R){
	if(L<=l&&r<=R) return sum[u];
	pushdown(u,r-l+1);
	ll res=0;
	if(L<=mid) res+=query(ls,l,mid,L,R);
	if(R>mid) res+=query(rs,mid+1,r,L,R);
	return res;
}
void chaT(int x,ll a){
	change(1,1,n,rnk[x],rnk[x]+siz[x]-1,a);
}
ll qryR(int x,int y){
	ll res=0;
	while(top[x]!=top[y]){
		if(dep[top[x]]<dep[top[y]]) swap(x,y);
		res+=query(1,1,n,rnk[top[x]],rnk[x]);
		x=fa[top[x]];
	}
	if(dep[x]>dep[y]) swap(x,y);
	res+=query(1,1,n,rnk[x],rnk[y]);
	return res;
}
int main(){
	#ifdef ytxy
	freopen("in.txt","r",stdin);
	#endif
	ios::sync_with_stdio(0);
	cin.tie(0),cout.tie(0);
	cin>>n>>m;
	for(int i=1;i<=n;i++) cin>>val[i];
	for(int i=1,u,v;i<n;i++){
		cin>>u>>v;
		egde(u,v);
	}
	cut(1,1);dfs(1,1);
	build(1,1,n);
	while(m--){
		int op,x;ll a;
		cin>>op>>x;
		if(op==1){
			cin>>a;
			update(1,1,n,x,a);
		}
		else if(op==2){
			cin>>a;
			chaT(x,a);
		}
		else if(op==3){
			cout<<qryR(x,1)<<'\n';
		}
	}
}

调了好久了 qwq

2022/8/9 16:54
加载中...