城市管理 (2018佛山市选拔赛)

题目描述：给出一棵有n个节点的带边权的有根树，然后有q次操作，维护两个功能：

1.修改：将点u，v的路径上的边边权统一加w

2.查询：询问以i号节点为根的子树的点的两两之间路径和，输出对2018取模的结果

20%数据，n，q<=200;

40%数据n,q<=5000;

100%数据，n,q<=50000;

20%做法：

对于每个修改，我们可以暴力改，但是对于每个询问我们不能枚举两个点再暴力跑路径，考虑在求路径的方法上进行优化，那么应该怎么做呢：不妨设点u，v的lca是点i，那么我们可以用一个神奇的方法求u，v的路径和：对于每一个节点维护一个值，就是从它到根节点的路径和，如图。

求节点6到节点9的路径可以这么求：先求出节点6,9的lca，发现他们的lca是3，然后取出节点9,节点5到根节点的路径和，分为12和14，把他们加起来，发现值刚刚好是它们的路径和加上它们的lca到根节点路径的两倍，所以再减去lca到路径和的两倍就好了。这样，时间复杂度是3次方级别的，如果写的是倍增lca则再乘个log。

40%做法：

我们换一种思路，能不能不枚举点对呢？答案是可以的，我们考虑对于i的子树，某一条边的单独贡献：

如图，我们考虑红色这条边的贡献 ，我们发现，如果点对中的两个点都是子树u内的节点，那么这条边是不会对那个点对产生影响的，同理，如果点对中的两个点都是在子树u的外面，这条边依然不会产生影响，那么对于一条边，它有且只有对那些点对中：一个点在u的子树内，一个点在u的子树外的情况产生影响，假设u的子树大小为size【u】，i的子树大小为size【i】，那么在u的子树中选点的方案数是size【u】种，在u的子树外选点的方案数是（size【i】-size【u】）种，那么让u子树内的点和子树外的点两两配对，一共有size【u】*（size【i】-size【u】）种方案，那么这条边的贡献就是，这条边的权值val乘上size【u】*（size【i】-size【u】）这样我们先dfs一遍，求出以每个点为根时的子树的大小，修改边的时候依然暴力改，求边的时候对每条边像上面那样单独考虑贡献，就可以过40%的数据了，时间复杂度：。

100%做法：

首先我们意识到不能直接暴力改边，由于是链上的修改，我们很容易想到树剖，那么，我们应该用数据结构维护一些什么信息才能让计算答案，变得更简便呢：考虑如果每次询问都是询问1号节点，那么节点u连向父亲的边的贡献就是val*size【u】*（n-size【u】）而如果询问的是i号节点，那么贡献是val*size【u】*（i-size【u】）=val*size【u】*（n-size【u】）-（n-size【i】）*val*size【u】，而val*size【u】*（n-size【u】）和val*size【u】两个值都是和询问的i节点无关的，所以我们只需要维护这两个值，假设sum1=val*size【u】*（n-size【u】），sum2=val*size【u】，那么这条边对i的子树的贡献就是sum1-（n-size【i】）*sum2。这样，这道题就完美的解决了,时间复杂度：。

                                           推荐番：《末日时在做什么？有没有空？可以来拯救吗？》

代码：

#include <iostream>
#include <cmath>
#include <algorithm>
#include <cstdio>
using namespace std;
int dep[100005],fa[100005],top[100005],siz[100005],line[100005],son[100005],vvv[100005];
int p,n,q,to[100005],pp[100005],next[100005],dui[100005],val[100005],summax[100005],summin[100005];
int x,u,v,w,rr;
struct data
{int mark,sum1,sum2;};
data tre[500005];
char ch;
void dfs1(int k)
{
	int pu=pp[k],ma=0,wei=0;siz[k]=1;
	while (pu>0)
	{
		dep[to[pu]]=dep[k]+1;
		fa[to[pu]]=k;
		dfs1(to[pu]);
		siz[k]+=siz[to[pu]];
		if (siz[to[pu]]>ma){ma=siz[to[pu]];wei=to[pu];}
		pu=next[pu];
	}
	son[k]=wei;
}
void dfs2(int k)
{
	p++;line[p]=k;dui[k]=p;
	if (son[k]!=0)
	{
		top[son[k]]=top[k];
	    dfs2(son[k]);
    }
	int pu=pp[k];
	while (pu>0)
	{
		if (to[pu]!=son[k])
		{
			top[to[pu]]=to[pu];
			dfs2(to[pu]);
		}
		pu=next[pu];
	}
}
void build(int root,int l,int r)
{
	if (l==r)
	{
	    tre[root].sum1=(val[l]*((siz[line[l]]*((n-siz[line[l]])%2018))%2018))%2018;
	    tre[root].sum2=(val[l]*siz[line[l]])%2018;
		return;
	}
	build(root+root,l,(l+r)/2);
	build(root+root+1,(l+r)/2+1,r);
	tre[root].sum1=(tre[root+root].sum1+tre[root+root+1].sum1)%2018;
	tre[root].sum2=(tre[root+root].sum2+tre[root+root+1].sum2)%2018;
}
void fang(int root,int l,int r)
{
	int mid=(l+r)/2;
	tre[root+root].sum1=(tre[root].mark*(summax[mid]-summax[l-1]+2018)+tre[root+root].sum1)%2018;
	tre[root+root+1].sum1=(tre[root].mark*(summax[r]-summax[mid]+2018)+tre[root+root+1].sum1)%2018;
	tre[root+root].sum2=(tre[root].mark*(summin[mid]-summin[l-1]+2018)+tre[root+root].sum2)%2018;
	tre[root+root+1].sum2=(tre[root].mark*(summin[r]-summin[mid]+2018)+tre[root+root+1].sum2)%2018;
	tre[root+root].mark=(tre[root].mark+tre[root+root].mark)%2018;
	tre[root+root+1].mark=(tre[root].mark+tre[root+root+1].mark)%2018;
	tre[root].mark=0;
}
void gai(int root,int l,int r,int h,int t,int vv)
{
	if (h>t)return;
	if (h<=l && t>=r)
	{
		tre[root].sum1=(tre[root].sum1+vv*(summax[r]-summax[l-1]+2018))%2018;
	    tre[root].sum2=(vv*(summin[r]-summin[l-1]+2018)+tre[root].sum2)%2018;
	    tre[root].mark=(tre[root].mark+vv)%2018;
	    return;
	}
	if (tre[root].mark>0)fang(root,l,r);
	int mid=(l+r)/2;
	if (h<=mid)gai(root+root,l,mid,h,t,vv);
	if (t>=mid+1)gai(root+root+1,mid+1,r,h,t,vv);
	tre[root].sum1=(tre[root+root].sum1+tre[root+root+1].sum1)%2018;
	tre[root].sum2=(tre[root+root].sum2+tre[root+root+1].sum2)%2018;
}
int askk(int root,int l,int r,int h,int t,int sizz)
{
	if (h>t)return 0;
	if (h<=l && t>=r)
	{
		return (tre[root].sum1-(tre[root].sum2*((n-sizz)%2018))%2018+2018)%2018;
	}
	if (tre[root].mark>0)fang(root,l,r);
		int mid=(l+r)/2,ss=0;
	if (h<=mid)ss+=askk(root+root,l,mid,h,t,sizz);
	if (t>=mid+1) ss+=askk(root+root+1,mid+1,r,h,t,sizz);
	tre[root].sum1=(tre[root+root].sum1+tre[root+root+1].sum1)%2018;
	tre[root].sum2=(tre[root+root].sum2+tre[root+root+1].sum2)%2018;
	return ss%2018;
}
void gg(int u,int v,int vv)
{
	while (top[u]!=top[v])
	{
		if (dep[top[u]]<dep[top[v]])swap(u,v);
		gai(1,1,n,dui[top[u]],dui[u],vv); 
		u=fa[top[u]];
	}
	gai(1,1,n,min(dui[u],dui[v])+1,max(dui[u],dui[v]),vv);
} 
int main()
{
	cin>>n>>q;
	for (int i=2;i<=n;i++)
	{
		scanf("%d %d",&x,&v);
		p++;to[p]=i;next[p]=pp[x];vvv[i]=v;pp[x]=p;
	}
	dep[1]=1;
	dfs1(1);
	p=0;
	top[1]=1;
	dfs2(1);
	for (int i=1;i<=n;i++)summax[i]=(summax[i-1]+siz[line[i]]*((n-siz[line[i]])%2018)%2018)%2018;
	for (int i=1;i<=n;i++)summin[i]=(summin[i-1]+siz[line[i]])%2018;
	for (int i=1;i<=n;i++)val[i]=vvv[line[i]];
	build(1,1,n);
	for (int i=1;i<=q;i++)
	{
		ch='$';
		while (ch!='I' && ch!='A')scanf("%c",&ch);
		scanf("%c",&ch);scanf("%c",&ch);
		if (ch=='C')
		{
			scanf("%d %d %d",&u,&v,&w);
			gg(u,v,w);
		}
		else if (ch='K')
		{
			scanf("%d",&rr);
			printf("%d\n",askk(1,1,n,dui[rr]+1,dui[rr]+siz[rr]-1,siz[rr]));
		}
	}
	return 0;
}