Wannafly挑战赛21 E 未来城市规划

Wannafly挑战赛21 E 未来城市规划

---

https://www.nowcoder.com/acm/contest/159/E

数据结构好题orz

先看看一条边在一个子树里的贡献，就是这条边的权值×一边的点数×另一边的点数

设子树的树根是\(x\)，这条边连接的深度较大的点是\(y\)。

那么贡献就是\(W(siz_y)(siz_x-siz_y)\)

\(W(siz_xsiz_y-siz_y^2)\)

边权放到点上，对每个点维护\(Wsiz\)和\(Wsiz^2\)即可。

注意nowcoder的clang++非常锅锅，尽量开全局变量，不然main里的n可能就随便从5变成67了（调了1h mmp）

#include<bits/stdc++.h>
#define il inline
#define vd void
#define mod 2019
typedef long long ll;
il int gi(){
	int x=0,f=1;
	char ch=getchar();
	while(!isdigit(ch)){
		if(ch=='-')f=-1;
		ch=getchar();
	}
	while(isdigit(ch))x=x*10+ch-'0',ch=getchar();
	return x*f;
}
int W[50010];
int fir[50010],dis[50010],nxt[50010],id;
il vd link(int a,int b){nxt[++id]=fir[a],fir[a]=id,dis[id]=b;}
int dfn[50010],redfn[50010],siz[50010],son[50010],dep[50010],fa[50010];
il vd dfs1(int x){
	siz[x]=1;
	for(int i=fir[x];i;i=nxt[i]){
		dep[dis[i]]=dep[x]+1;
		dfs1(dis[i]);
		siz[x]+=siz[dis[i]];
		if(siz[dis[i]]>siz[son[x]])son[x]=dis[i];
	}
}
int top[50010];
il vd dfs2(int x,int tp){
	dfn[x]=++dfn[0];redfn[dfn[0]]=x;
	top[x]=tp;
	if(son[x])dfs2(son[x],tp);
	for(int i=fir[x];i;i=nxt[i])if(dis[i]!=son[x])dfs2(dis[i],dis[i]);
}
int s1[200010],s2[200010],S1[200010],S2[200100],lz[200010];
#define mid ((l+r)>>1)
il vd Inc(int x,const int&t){
	S1[x]=(S1[x]+s1[x]*t)%mod;
	S2[x]=(S2[x]+s2[x]*t%mod)%mod;
	lz[x]=(lz[x]+t)%mod;
}
il vd down(const int&x){if(lz[x])Inc(x<<1,lz[x]),Inc(x<<1|1,lz[x]),lz[x]=0;}
il vd upd(const int&x){
	S1[x]=S1[x<<1]+S1[x<<1|1];
	S2[x]=S2[x<<1]+S2[x<<1|1];
}
il vd build(int x,int l,int r){
	if(l==r){
		int p=redfn[l];
		s1[x]=siz[p],s2[x]=siz[p]*siz[p]%mod;
		S1[x]=s1[x]*W[p]%mod,S2[x]=s2[x]*W[p]%mod;
		return;
	}
	build(x<<1,l,mid),build(x<<1|1,mid+1,r);
	upd(x);
	s1[x]=(s1[x<<1]+s1[x<<1|1])%mod;
	s2[x]=(s2[x<<1]+s2[x<<1|1])%mod;
}
il vd update(int x,int l,int r,const int&L,const int&R,const int&d){
	if(L<=l&&r<=R)return Inc(x,d);
	down(x);
	if(L<=mid)update(x<<1,l,mid,L,R,d);
	if(mid<R)update(x<<1|1,mid+1,r,L,R,d);
	upd(x);
}
il std::pair<int,int> operator +(const std::pair<int,int>&a,const std::pair<int,int>&b){return std::make_pair(a.first+b.first,a.second+b.second);}
il std::pair<int,int> query(int x,int l,int r,const int&L,const int&R){
	if(L<=l&&r<=R)return std::make_pair(S1[x],S2[x]);
	down(x);
	if(L<=mid)
		if(mid<R)return query(x<<1,l,mid,L,R)+query(x<<1|1,mid+1,r,L,R);
		else return query(x<<1,l,mid,L,R);
	else return query(x<<1|1,mid+1,r,L,R);
}
int n,q;
int main(){
    n=gi(),q=gi();
	for(int i=2;i<=n;++i)fa[i]=gi(),W[i]=gi(),link(fa[i],i);
	dfs1(1),dfs2(1,1);
	build(1,1,n);
	char opt[2];int u,v,w;
	while(q--){
		scanf("%s",opt);
		if(opt[0]=='I'){
			u=gi(),v=gi(),w=gi();
			if(u==v)continue;
			while(top[u]!=top[v]){
				if(dep[top[u]]>dep[top[v]])update(1,1,n,dfn[top[u]],dfn[u],w),u=fa[top[u]];
				else update(1,1,n,dfn[top[v]],dfn[v],w),v=fa[top[v]];
			}
			if(dfn[u]>dfn[v])std::swap(u,v);
			update(1,1,n,dfn[u]+1,dfn[v],w);
		}else{
			w=gi();
			if(siz[w]==1){puts("0");continue;}
			std::pair<int,int>A=query(1,1,n,dfn[w]+1,dfn[w]+siz[w]-1);
			A.first%=mod,A.second%=mod;
			printf("%d\n",(A.first*siz[w]-A.second+mod)%mod);
		}
	}
	return 0;
}