500D New Year Santa Network (边权期望+树形DP)
New Year is coming in Tree World! In this world, as the name implies, there are n cities connected by n - 1 roads, and for any two distinct cities there always exists a path between them. The cities are numbered by integers from 1 to n, and the roads are numbered by integers from 1 to n - 1. Let's define d(u, v) as total length of roads on the path between city u and city v.
As an annual event, people in Tree World repairs exactly one road per year. As a result, the length of one road decreases. It is already known that in the i-th year, the length of the ri-th road is going to become wi, which is shorter than its length before. Assume that the current year is year 1.
Three Santas are planning to give presents annually to all the children in Tree World. In order to do that, they need some preparation, so they are going to choose three distinct cities c1, c2, c3 and make exactly one warehouse in each city. The k-th (1 ≤ k ≤ 3) Santa will take charge of the warehouse in city ck.
It is really boring for the three Santas to keep a warehouse alone. So, they decided to build an only-for-Santa network! The cost needed to build this network equals to d(c1, c2) + d(c2, c3) + d(c3, c1) dollars. Santas are too busy to find the best place, so they decided to choose c1, c2, c3 randomly uniformly over all triples of distinct numbers from 1 to n. Santas would like to know the expected value of the cost needed to build the network.
However, as mentioned, each year, the length of exactly one road decreases. So, the Santas want to calculate the expected after each length change. Help them to calculate the value.
The first line contains an integer n (3 ≤ n ≤ 105) — the number of cities in Tree World.
Next n - 1 lines describe the roads. The i-th line of them (1 ≤ i ≤ n - 1) contains three space-separated integers ai, bi, li (1 ≤ ai, bi ≤ n, ai ≠ bi, 1 ≤ li ≤ 103), denoting that the i-th road connects cities ai and bi, and the length of i-th road is li.
The next line contains an integer q (1 ≤ q ≤ 105) — the number of road length changes.
Next q lines describe the length changes. The j-th line of them (1 ≤ j ≤ q) contains two space-separated integers rj, wj (1 ≤ rj ≤ n - 1,1 ≤ wj ≤ 103). It means that in the j-th repair, the length of the rj-th road becomes wj. It is guaranteed that wj is smaller than the current length of the rj-th road. The same road can be repaired several times.
Output q numbers. For each given change, print a line containing the expected cost needed to build the network in Tree World. The answer will be considered correct if its absolute and relative error doesn't exceed 10 - 6.
3
2 3 5
1 3 3
5
1 4
2 2
1 2
2 1
1 1
14.0000000000
12.0000000000
8.0000000000
6.0000000000
4.0000000000
6
1 5 3
5 3 2
6 1 7
1 4 4
5 2 3
5
1 2
2 1
3 5
4 1
5 2
19.6000000000
18.6000000000
16.6000000000
13.6000000000
12.6000000000
Consider the first sample. There are 6 triples: (1, 2, 3), (1, 3, 2), (2, 1, 3), (2, 3, 1), (3, 1, 2), (3, 2, 1). Because n = 3, the cost needed to build the network is always d(1, 2) + d(2, 3) + d(3, 1) for all the triples. So, the expected cost equals to d(1, 2) + d(2, 3) + d(3, 1).
题意:有n个顶点,给你n-1条边,每条边有权重w,给你q次询问,第i条边权重减少到y,(前面的减少是会影响后面的询问的)问你从n个顶点任选3个并算出路程的期望。
思路:求任意选3个点 c1,c2,c3 的 d(c1,c2) + d(c2,c3) + d(c1,c3) 的期望。
只要计算每一条边在所有情况下[ C(3,n ) ]使用的次数cnt便可以知道总路程d = sigma(cnt[i] * w[i]),那么exp = d / C(3,n) 。q次询问更改边权就是减去之前的边期望, 加上新的边权求出来的期望 。注意要用 , cnt[i]*w[i] / C(3,n) 处理好边, 即边期望。若果先求出总路程再除C(3,n)的话数据过大会溢出。
首先选法有[C(3,n)] = n*(n-1)*(n-2)/6种,每条边设其两边的节点数分别为a和b,那么这条边用到的情况为a*(a-1)*b/2+b*(b-1)*a/2种,而这三个节点间的每条边都需要用两次。
代码:
#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
const int N = 100005;
vector<pair<int,int> > g[N];
int cnt[N],n,w[N];
double C3,ans=0;
int dfs(int u,int fa)
{
int tot=0;
for(int i=0; i<g[u].size(); i++)
{
int v=g[u][i].first,id=g[u][i].second;
if(v == fa) continue;
cnt[id]=dfs(v,u);//存的是id边一个方向上的节点数
tot+=cnt[id];
}
return tot+1;
}
double cal(int i)
{
double a=cnt[i],b=n-cnt[i];//这里要用double,不然会数据过大出错
return 2.0*w[i]/C3*(a*(a-1)*b/2.0+b*(b-1)*a/2.0);
}
int main()
{
int a,b,q,x,y;
scanf("%d",&n);
for(int i=1; i<n; i++)
{
scanf("%d%d%d",&a,&b,&w[i]);
g[a].push_back(make_pair(b,i));
g[b].push_back(make_pair(a,i));
}
C3=1.0*n*(n-1)*(n-2)/6;
int m=dfs(1,0);
for(int i=1; i<n; i++)
ans+=cal(i);
scanf("%d",&q);
for(int i=0; i<q; i++)
{
scanf("%d%d",&x,&y);
ans-=cal(x);
w[x]=y;
ans+=cal(x);
printf("%.9lf\n",ans);
}
}

浙公网安备 33010602011771号