绿色
题意简述
题解
首先,每次修改完点权后,重新考虑一遍所有路径显然是不现实的,所以我们考虑求出经过每个点的两端同色的简单路径数,这样权值和容易统计和修改。
接下来分析仙人掌上的简单路径性质。一条简单路径上的边,可以分为桥边和环上的边。过桥只有一种方法,而过同一个环有两个方向的环边可选。
现在,我们选取一个关键点,强制简单路径必须通过该点。如果关键点在一个环上,且该简单路径在该环上的第一个和最后一个点均不是关键点,那么该环只有一个方向的环边可选。
建立原仙人掌的圆方树。
如果定义方点贡献 2 种方案,圆点贡献 1 种方案,一条路径贡献方案数为所有点贡献方案数之积,那么原仙人掌上两个同色点间的路径数转化为对应圆点间的简单路径方案数。
在所有的树上简单路径中,
如果它通过关键点,则贡献值为该路径的方案数;
如果它不通过关键点,但是通过关键点的某相邻方点,则贡献该路径方案数的一半。
考虑用何算法求出经过每个点的两端同色的简单路径数(记过uuu的简单路径数为coe[u]coe[u]coe[u]):
法一:暴力枚举
按题意枚举所有同色点对,可解决n≤100n≤100n≤100的子任务。
法二:树形DP
首先枚举每一种颜色,每次只考虑当前颜色的同色点对的贡献。
对于树上任意一点uuu,我们可以求出:
dp[u]=dp[u]=dp[u]= uuu的子树内当前色的圆点到uuu的路径数
up[u]=up[u]=up[u]= uuu的子树外当前色的圆点到uuu的路径数
(换根DP)
那么过uuu的简单路径数coe[u]+=(dp[u]+up[u]+1)2−dp[u]2−dp[v]2−122coe[u]+=\frac{(dp[u]+up[u]+1)^2-dp[u]^2-dp[v]^2-1^2}{2}coe[u]+=2(dp[u]+up[u]+1)2−dp[u]2−dp[v]2−12
这样可以解决n≤105,k≤5n\leq10^5,k\leq5n≤105,k≤5的子任务
优化:建虚树
处理树上询问点两两间路径的并的问题,一个常见的套路是构建询问点的虚树,因为虚树中同一条边上的点受询问点的影响一致,把这些点一起处理可大大提高效率
于是,我们构建同色点的虚树,通过前缀和,预处理虚树每条边上压缩的点贡献的方案数,作为边的方案数。
对于过虚树节点的简单路径数,可以在树形dp时直接统计出来;
对于过虚树边上压缩掉的点的简单路径数,可以利用树上差分与前缀和快速处理。
那么接下来仅需考虑 过关键点相邻的方点的路径总方案数 即可:
计算通过了每个相邻方点的路径总方案数,再减去通过了它们之间相邻边的方案数。
不难发现,上述树上差分与前缀和的方法,已经预处理了通过各边的路径方案数。
此算法时间复杂度O(nlogn)O(n log n)O(nlogn)、空间复杂度O(n)O(n)O(n)
#include<bits/stdc++.h>
using namespace std;
const int mod=1e9+7;
const int inv2=(mod+1)/2;
namespace modular{int add(int a,int b){return a+b>=mod?a+b-mod:a+b;}int dec(int a,int b){return a-b<0?a-b+mod:a-b;}int mul(int a,int b){return 1ll*a*b%mod;}void Add(int &a,int b){a=add(a,b);}void Dec(int &a,int b){a=dec(a,b);}void Mul(int &a,int b){a=mul(a,b);}
}
using namespace modular;
inline int read(){int x=0,f=1;char ch=getchar();while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}return x*f;
}
const int N=1e6+10;
int n,m,k,pw[N];
int c[N],w[N];
vector<int> S[N];int cnt,head[N],to[N<<1],nxt[N<<1];
int coe[N],nd;
vector<int> e[N];
int in[N],out[N],dfn,fr[N];
bool is[N];
void addedge(int u,int v){to[++cnt]=v;nxt[cnt]=head[u];head[u]=cnt;
}
void build(int u,int pre){//建圆方树 in[u]=++dfn;for(int i=head[u];i;i=nxt[i]){if(!((i^1)^pre)) continue;//条件一定不能改 int v=to[i];if(!in[v]){fr[v]=u;build(v,i);//记来边而不是来点 }else if(in[v]<in[u]){++nd;e[nd].push_back(v);e[v].push_back(nd);for(int x=u;x!=v;x=fr[x]){is[x]=1;e[nd].push_back(x);e[x].push_back(nd);}}}
}int ct[N],fa[N],sz[N],son[N],dep[N],top[N];
void sub_dfs(int u,int f){in[u]=++dfn;sz[u]=1;fa[u]=f;dep[u]=dep[f]+1;for(int i=0;i<e[u].size();i++){int v=e[u][i];if(v==f) continue;ct[v]+=ct[u]+(u>n);sub_dfs(v,u);sz[u]+=sz[v];if(sz[v]>sz[son[u]]) son[u]=v;}out[u]=dfn;
}
void tp_dfs(int u,int tp){top[u]=tp;if(!son[u]) return;tp_dfs(son[u],tp);for(int i=0;i<e[u].size();i++){int v=e[u][i];if(v!=fa[u]&&v!=son[u]) tp_dfs(v,v);}
}
int LCA(int x,int y){while(top[x]!=top[y]){if(dep[top[x]]<dep[top[y]]) swap(x,y);x=fa[top[x]];}return dep[x]<dep[y]?x:y;
}
bool sub(int x,int y){return in[x]<=in[y]&&in[y]<=out[x];
}int ta[N],dot[N];
//ta[u]记录过u,且过边(u,fa[u])的路径数(用树上差分求出,所以work时的ta是差分数组,u子树内的ta和才是实际路径数)
//dot[u]记录过u,但不过边(u,fa[u])的路径数
vector<int> G[N];
int dp[N],up[N];
//dp[u]记录u子树内当前色的圆点到u的路径数
//up[u]记录u子树外当前色的圆点到虚树上u的父亲的路径数
bool o[N];
bool cmp(int x,int y){return in[x]<in[y];
}
void dfs1(int u){dp[u]=o[u];for(int i=0;i<G[u].size();i++){int v=G[u][i];dfs1(v);Add(dp[u],mul(pw[ct[v]-ct[u]],dp[v]));}
}
void dfs2(int u,int f){int c;if(f){c=mul(up[u],dp[u]);Mul(c,pw[ct[u]-ct[f]-(f>n)]);Add(ta[u],c);Dec(ta[f],c);Dec(dot[u],c);}c=mul(pw[ct[u]-ct[f]+(u>n)-(f>n)],up[u]);int S=mul(c,c)+o[u], all=add(dp[u],c);for(int i=0;i<G[u].size();i++){int v=G[u][i];c=mul(pw[ct[v]-ct[u]],dp[v]);Add(S,mul(c,c));}S=dec(mul(all,all),S);Mul(S,inv2);if(u>n) Mul(S,inv2);Add(dot[u],S);for(int i=0;i<G[u].size();i++){int v=G[u][i];Add(up[v],dp[u]);Add(up[v],mul(pw[ct[u]-ct[f]+(u>n)-(f>n)],up[u]));Dec(up[v],mul(pw[ct[v]-ct[u]],dp[v]));dfs2(v,u);}
}
void work(vector<int> &p){//对每种颜色的点建虚树,求解 sort(p.begin(),p.end(),cmp);static int ex[N],tim;tim++;for(int i=0;i<p.size();i++){int x=p[i];ex[x]=tim;o[x]=1;}int n=p.size(),x,y,z;for(int i=0;i+1<n;i++){int x=LCA(p[i],p[i+1]);if(ex[x]!=tim) ex[x]=tim,p.push_back(x);}sort(p.begin(),p.end(),cmp);static int s[N];int top=0;for(int i=0;i<p.size();i++){x=p[i];while(top>1&&!sub(s[top],x)){y=s[top],z=s[top-1];G[z].push_back(y);top--;} s[++top] = x; }while(top>1){y=s[top],z=s[top-1];G[z].push_back(y);top--;}dfs1(p[0]);dfs2(p[0],0);for(int i=0;i<p.size();i++){int x=p[i];up[x]=o[x]=0;G[x].clear();}p.clear();
}
void dfs(int u,int f){for(int i=0;i<e[u].size();i++){int v=e[u][i];if(v==f) continue;dfs(v,u);Add(ta[u],ta[v]);}coe[u]=mul(2,add(ta[u],dot[u]));for(int i=0;i<e[u].size();i++){int v=e[u][i];if(v!=f&&v>n) Add(coe[u],dot[v]);}
}
int main(){n=read();m=read();k=read();pw[0]=1;for(int i=1;i<=n;i++) pw[i]=add(pw[i-1],pw[i-1]);nd=n;for(int i=1;i<=n;i++){c[i]=read();S[c[i]].push_back(i);}for(int i=1;i<=n;i++) w[i]=read();cnt=1;for(int i=1;i<=m;i++){int u=read(),v=read();addedge(u,v);addedge(v,u);}build(1,0);for(int i=2;i<=n;i++){if(!is[i]){e[i].push_back(fr[i]);e[fr[i]].push_back(i);} }dfn=0; sub_dfs(1,0);tp_dfs(1,1);for(int i=1;i<=k;i++) work(S[i]);dfs(1,0);for(int i=2;i<=n;i++){int x=fa[i];if(x>n){int c=add(ta[x],dot[x]);Dec(c,ta[i]);Add(coe[i],c);}}for(int i=1;i<=n;i++) Mul(coe[i],inv2);//trick:前面先计算系数的2倍,最后再/2 int ans=0;for(int i=1;i<=n;i++)Add(ans,mul(w[i],coe[i]));printf("%d\n",ans);int Q=read();for(int i=1;i<=Q;i++){int x=read(),v=read();Add(ans,mul(v,coe[x]));printf("%d\n",ans);}return 0;
}
/*
4 5 2
1 2 1 2
1 2 3 4
1 2
1 2
2 3
2 4
3 4
1
3 1
*/