原题
其实就是利用树链剖分建图,然后在线段树上实现。树链剖分中的每一个树枝都可以在线段数中有对应的。然后,至于算出‘城市u所管辖下的所有城市对’的花费之和,其实就是每一个子城市(v)的贡献=路程wsiz[v](siz[u]-siz[v])(这个很容易证),可以在线段树中用2个数字维护。一个维护wsiz[v],另一个维护wsiz[v]*siz[v],然后就是区间求和,区间改变。
#include<bits/stdc++.h>
#define lson k<<1,l,mid
#define rson k<<1|1,mid+1,r
#define ll long long
using namespace std;
const int MX=5e4+9;
const int mod=2019;
struct node{
int u,v,next;
}edge[MX];
int n,q,u,v,futh,cnt=0,tot=0;
int head[MX],fa[MX],siz[MX],tp[MX],son[MX],de[MX],po[MX],st[MX],ed[MX];
int w,a[MX],t1[MX<<2],t2[MX<<2],laze[MX<<2],s1[MX<<2],s2[MX<<2];
char s[20];
void add(int u,int v){
edge[cnt].v=u;
edge[cnt].next=head[v];
head[v]=cnt++;
}
void dfs1(int u,int fath,int deth){
fa[u]=fath,de[u]=deth,son[u]=-1,siz[u]=1;
for( int i=head[u] ; ~i ; i=edge[i].next ){
if( edge[i].v==fath )
continue;
dfs1(edge[i].v,u,deth+1);
siz[u]+=siz[edge[i].v];
if( son[u]==-1 || siz[son[u]]<siz[edge[i].v] )
son[u]=edge[i].v;
}
}
void dfs2(int u,int top){
st[u]=++tot;
po[tot]=u;
tp[u]=top;
if( son[u]>0 )
dfs2(son[u],top);
for( int i=head[u] ; ~i ; i=edge[i].next )
if( edge[i].v!=son[u] )
dfs2(edge[i].v,edge[i].v);
ed[u]=tot;
return ;
}
void pushup(int k){
t1[k]=(t1[k<<1]+t1[k<<1|1])%mod;
t2[k]=(t2[k<<1]+t2[k<<1|1])%mod;
}
void pushdown(int k){
if( laze[k] ){
t1[k<<1]=(t1[k<<1]+(laze[k]*s1[k<<1])%mod)%mod;
t1[k<<1|1]=(t1[k<<1|1]+(laze[k]*s1[k<<1|1])%mod)%mod;
t2[k<<1]=(t2[k<<1]+(laze[k]*s2[k<<1])%mod)%mod;
t2[k<<1|1]=(t2[k<<1|1]+(laze[k]*s2[k<<1|1])%mod)%mod;
laze[k<<1]=(laze[k<<1]+laze[k])%mod;
laze[k<<1|1]=(laze[k<<1|1]+laze[k])%mod;
laze[k]=0;
}
}
void build(int k,int l,int r){
laze[k]=0;
if( l==r ){
s1[k]=siz[po[l]]%mod;
s2[k]=s1[k]*s1[k]%mod;
t1[k]=s1[k]*a[po[l]]%mod;
t2[k]=s2[k]*a[po[l]]%mod;
return ;
}
int mid=(l+r)>>1;
build(lson);
build(rson);
pushup(k);
s1[k]=(s1[k<<1]+s1[k<<1|1])%mod;
s2[k]=(s2[k<<1]+s2[k<<1|1])%mod;
}
void update(int k,int l,int r,int L,int R,int w){
if( L<=l && r<=R ){
t1[k]=(t1[k]+w*s1[k]%mod)%mod;
t2[k]=(t2[k]+w*s2[k]%mod)%mod;
laze[k]=(laze[k]+w)%mod;
return ;
}
pushdown(k);
int mid=(l+r)>>1;
if( L<=mid )
update(lson,L,R,w);
if( mid<R )
update(rson,L,R,w);
pushup(k);
}
int que1(int k,int l,int r,int L,int R){
if( L<=l && r<=R )
return t1[k];
pushdown(k);
int mid=(l+r)>>1;
int ans=0;
if( L<=mid )
ans+=que1(lson,L,R);
if( mid<R )
ans+=que1(rson,L,R);
return ans%mod;
}
int que2(int k,int l,int r,int L,int R){
if( L<=l && r<=R )
return t2[k];
pushdown(k);
int mid=(l+r)>>1;
int ans=0;
if( L<=mid )
ans+=que2(lson,L,R);
if( mid<R )
ans+=que2(rson,L,R);
return ans%mod;
}
void vs(int u,int v,ll w){
int f1=tp[u],f2=tp[v];
while( f1!=f2 ){
if( de[f1]>de[f2] ){
swap(f1,f2);
swap(u,v);
}
update(1,1,n,st[f2],st[v],w); // 走一次,更新一次
v=fa[f2];
f2=tp[v];
}
if( u==v )
return ;
if( st[u]>st[v] )
swap(u,v);
update(1,1,n,st[u]+1,st[v],w);
return ;
}
int main()
{
// freopen("input.txt","r",stdin);
scanf("%d %d",&n,&q);
memset(head,-1,sizeof(head));
for( int i=1 ; i<=n-1 ; i++ ){
scanf("%d %d",&futh,&w);
a[i+1]=w;
add(i+1,futh);
}
dfs1(1,0,1);
dfs2(1,1);
build(1,1,n);
while( q-- ){
scanf("%s",s);
if( s[0]=='I' ){
scanf("%d %d %d",&u,&v,&w);
vs(u,v,w%mod);
}
else{
scanf("%d",&u);
if( siz[u]==1 )
printf("0\n");
else{
int q1=que1(1,1,n,st[son[u]]+1,ed[u]),q2=que2(1,1,n,st[son[u]]+1,ed[u]); //记住,例如2到3,改变的其实只有3,没有2(谁的深度低,谁被改变)
int ans=(q1*siz[u]-q2+mod)%mod;
printf("%d\n",ans%mod);
}
}
}
fclose(stdin);fclose(stdout);
return 0;
}