题目大意:给n个点,m条边,每条边权值c,现在要使这n个点连通。现在已知某条边要发生突变,再给q个三元组,每个三元组(a,b,c),(a,b)表示图中可能发生突变的边,该边一定是图中的边。c表示该边新的权值,c只可能比原来的权值大。给的q条边发生突变的概率是一样的。求突变后连通n个点最小代价期望值。
题目分析:如果没有那条突变的边,就是求一个mst。但是因为有一条边要突变,每条边突变的概率相同,都为1/q。所以要枚举所有的q条边,求出该边突变后最小生成树代价。q条边分2类:一类是生成树中的边,第二类不是生成树中的边。
第二类边很好处理,既然本身不是生成树中的边。那么这条边突变后权值变大,现在要使权值最小,这条边不选就是了,所以总代价依然是原图的mst。
第一类边发生突变的话,那么去掉这条边后原来的mst就被分成了2个子树,所以我们要找到解决办法就是寻找这2个子树之间的次短边(最小边是去掉的那条生成树中的边)。比较一下,如果这条次短边的权值小于这条突变的生成树上的边,那么就要换掉这条边,否则保留。
现在的关键是求去掉某条边生成树边后,2个子树的最小距离。很容易想到一个O(n^3)的算法:枚举每条生成树边(u,v),分别从u和v开始沿着生成树边遍历,求出两两点间最小值。可是复杂度未免太高。借助dp的思想,可以将这个过程复杂度降一个n。
考虑次过程的这样一个性质:求某个点i到以点j为根的树的最短距离=min(i到j所有子树的最短距离,i到j的最短距离),那么可以枚举起点i,从i点开始沿着mst的边dfs,每经过一条边,那么这条边可以将n个点分成2部分,一部分含i,另一部分不含i(废话),利用dfs的性质,对于从i出发沿着生成树的边遍历到的每个点,当要离开这个点的时候,保证其所有的子树都已经遍历完毕,那么i到j的子树的最短距离就确定了,那么以到达j的边为割边,i到j这颗子树的最短距离就有了。每一次从i点的dfs表示的是枚举割边后包含i的子树和不包含i的子树之间的最小距离。枚举每个点,就能得到对于所有生成树的边,dp[u][v]表示以(u,v)为割边的两颗子树的最短距离。
好NB的DP!
汉语组织的可能不是很好,具体画图吧,好容易懂的。
详情请见代码:
#include <iostream> #include<cstdio> #include<cstring> #include<algorithm> using namespace std; const int N = 3005; const int M = 1000005; const int inf = 0x3f3f3f3f; int dp[N][N],dis[N][N],cost[N][N],lowcost[N],pre[N]; bool used[N][N],flag[N]; int head[N],num; double ans; int n,m,q,mst; struct node { int to,next; }e_mst[M]; void build(int s,int e) { e_mst[num].to = e; e_mst[num].next = head[s]; head[s] = num ++; } void prim() { int i,j; mst = 0; memset(flag,false,sizeof(flag)); for(i = 0;i < n;i ++) { lowcost[i] = dis[0][i]; pre[i] = 0; } flag[0] = true; for(i = 1;i < n;i ++) { int minn = inf; int v; for(j = 0;j < n;j ++) { if(lowcost[j] < minn && flag[j] == false) { minn = lowcost[j]; v = j; } } mst += minn; used[pre[v]][v] = used[v][pre[v]] = true; build(pre[v],v); build(v,pre[v]); flag[v] = true; for(j = 0;j < n;j ++) { if(flag[j] == false && lowcost[j] > dis[v][j]) { lowcost[j] = dis[v][j]; pre[j] = v; } } } } int dfs(int cur,int u,int fa)//用cur更新cur点所在的子树和另外子树的最短距离 { int ret = inf; for(int i = head[u];~i;i = e_mst[i].next)//沿着生成树的边遍历 { if(e_mst[i].to == fa) continue; int tmp = dfs(cur,e_mst[i].to,u);//用cur更新的以当前边(u,e_mst[i].to)为割边的两个子树最短距离 ret = min(tmp,ret);//以(fa,u)为割边的2个子树的最短距离 dp[u][e_mst[i].to] = dp[e_mst[i].to][u] = min(dp[u][e_mst[i].to],tmp); } if(fa != cur)//生成树边不更新 ret = min(ret,dis[cur][u]); return ret; } void solve() { int i,j; int a,b,c; int sum = 0; memset(head,-1,sizeof(head)); num = 0; prim(); for(i = 0;i < n;i ++) dfs(i,i,-1); scanf("%d",&q); for(i = 0;i < q;i ++) { scanf("%d%d%d",&a,&b,&c); if(used[a][b]) { if(c < dp[a][b]) sum += (mst - dis[a][b] + c); else sum += (mst - dis[a][b] + dp[a][b]); } else sum += mst; } ans = (double)sum/(double)q; printf("%.4lf\n",ans); } int main() { int i,j,a,b,c; while(scanf("%d%d",&n,&m),(m + n)) { for(i = 0;i < n;i ++) for(j = 0;j < n;j ++) { dp[i][j] = inf; used[i][j] = false; if(i == j) dis[i][j] = 0; else dis[i][j] = inf; } for(i = 0;i < m;i ++) { scanf("%d%d%d",&a,&b,&c); dis[a][b] = dis[b][a] = c; } solve(); } return 0; } //921MS 80028K