Description
给一棵有n个节点的树,有q次操作,每次操作将改变某条边边权,问改变后,从n个点中等概率任取三点,将这三点用最短路连接起来的距离和的期望
Input
第一行为一整数n表示点数,之后n-1行每行三个整数u,v,w表示u点和v点之间有一条边权为w的边,然后输入一个整数q表示操作次数,之后q行每行两个整数r,w表示将第r条边的边权该为w
Output
对于每次操作,输出操作后的期望
Sample Input
3
2 3 5
1 3 3
5
1 4
2 2
1 2
2 1
1 1
Sample Output
14.0000000000
12.0000000000
8.0000000000
6.0000000000
4.0000000000
Solution
从n个点中选三个点的方法数有n*(n-1)(n-2)/6种,假设某条边两边节点数分别为x和y的话,那么经过这条边的方法数有x(x-1)/2*y+y*(y-1)/2*x种,所有首先做一遍树形DP求出以某个点为根节点的子树中节点个数以及尚未改变边权时的期望,对于每次修改,例如将第i条边的边权改成ww,原边权为w,第i条边两边点数为x,y,那么对答案的影响就是(x * (x-1)* y+y * (y-1) * x) * (ww-w)
Code
#include<cstdio>
#include<iostream>
#include<cstring>
#include<vector>
using namespace std;
#define maxn 111111
int n,edge[maxn][3],num[maxn];
vector<int>g[maxn];
void dfs(int u,int fa)
{
num[u]=1;
for(int i=0;i<g[u].size();i++)
{
int v=g[u][i];
if(v==fa)continue;
dfs(v,u);
num[u]+=num[v];
}
}
int main()
{
int n,q;
while(~scanf("%d",&n))
{
for(int i=1;i<=n;i++)g[i].clear();
for(int i=1;i<n;i++)
{
scanf("%d%d%d",&edge[i][0],&edge[i][1],&edge[i][2]);
int u=edge[i][0],v=edge[i][1],w=edge[i][2];
g[u].push_back(v),g[v].push_back(u);
}
dfs(1,1);
double ans=0;
for(int i=1;i<n;i++)
{
int u=edge[i][0],v=edge[i][1],w=edge[i][2];
int x=min(num[u],num[v]),y=n-x;
ans+=(1.0*x*(x-1)*y+1.0*y*(y-1)*x)*w;
}
scanf("%d",&q);
while(q--)
{
int r,w;
scanf("%d%d",&r,&w);
int x=min(num[edge[r][0]],num[edge[r][1]]),y=n-x;
ans+=(1.0*x*(x-1)*y+1.0*y*(y-1)*x)*(w-edge[r][2]);
edge[r][2]=w;
printf("%.10lf\n",ans*6/n/(n-1)/(n-2));
}
}
}