2023NOIP A层联测32 sakuya

题目大意

有一棵有$n$个节点的树，每条边有一个边权$w$。有$m$个特殊点，将这些点记为集合$A$。

将$A$中的元素随机打乱得到序列$a$，求$\sum _{i=2}^{m}d(a_{i-1},a_i)$的期望值模$998244353$后的值，其中$d(x,y)$表示$x$到$y$的边权和。

有$q$次修改，每次修改会将与$x$相连的边的权值增加$k$。求每次修改后上述式子的期望值。

$1\le n\le 5\times 10^5,m\le n,1\le q\le 5\times 10^5$

$1\le w,k\le 10^9$

题解

对于每组特殊点$x,y$，我们考虑有多少种方案会计算到$d(x,y)$的贡献。在确定$x,y$在$a$中相邻之后，其他$m-2$个数有$(m-2)!$种放法，$x,y$中较前的数可以放在第一个到第$m-1$个位置上，确定了前一个数，则后一个数也确定了，而这两个数的顺序可以为$x,y$或者$y,x$，所以还要乘$2$，也就是说有$2(m-2)!\times (m-1)=2(m-1)!$种方案会计算到$d(x,y)$的贡献。而题目要求的是期望值，总共有$m!$种方案，那么$d(x,y)$对答案的贡献为$\frac{2(m-1)!}{m!}\times d(x,y)=\frac{2}{m}\times d(x,y)$。

下面，我们要求每条边被多少$d(x,y)$计算过，这用一个$dfs$即可算出，记这个值为$t{d}_i$。然后，求出所有边$i$的$w_i$与$t{d}_i$之积的和，也就是$\sum _i{w}_i\times t{d}_i$，$\frac{m}{2}\times \sum _i{w}_i\times t{d}_i$即为答案。

我们考虑每次修改对答案的贡献。设与$i$相连的边的$td$值之和为$t{w}_i$，则每次修改会让$\sum _i{w}_i\times t{d}_i$增加$k\times t{w}_i$。那么，我们可以$O(1)$修改。因为题目只需要求答案，所以我们不需要真的去修改$w_i$。

时间复杂度为$O(n+q)$。

code

#include<bits/stdc++.h>
using namespace std;
const int N=500000;
const long long mod=998244353;
int n,m,q,z[N+5],siz[N+5];
long long ans,pt,w[N+5],td[N+5],tw[N+5];
vector<pair<int,int>>g[N+5];
long long mi(long long t,long long v){if(!v) return 1;long long re=mi(t,v/2);re=re*re%mod;if(v&1) re=re*t%mod;return re;
}
void dfs(int u,int fa){siz[u]=z[u];for(auto p:g[u]){int v=p.first,id=p.second;if(v==fa) continue;dfs(v,u);siz[u]+=siz[v];td[id]=1ll*(m-siz[v])*siz[v]%mod;}
}
int main()
{
//	freopen("sakuya.in","r",stdin);
//	freopen("sakuya.out","w",stdout);scanf("%d%d",&n,&m);for(int i=1,x,y;i<n;i++){scanf("%d%d%lld",&x,&y,&w[i]);g[x].push_back({y,i});g[y].push_back({x,i});}for(int i=1,x;i<=m;i++){scanf("%d",&x);z[x]=1;}dfs(1,0);for(int i=1;i<n;i++){ans=(ans+td[i]*w[i])%mod;}for(int i=1;i<=n;i++){for(auto p:g[i]){tw[i]=(tw[i]+td[p.second])%mod;}}scanf("%d",&q);long long tq=mi(m,mod-2)*2%mod;for(int o=1,x,k;o<=q;o++){scanf("%d%d",&x,&k);ans=(ans+tw[x]*k)%mod;pt=ans*tq%mod;printf("%lld\n",pt);}return 0;
}