题目描述:给出一棵有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;
}