单独练习的时候,做到最小生成树处理出来后,就不知道怎么处理了。关键是两个子树间的距离,不能在有效的时间复杂度内解决。
实际上,对于一个最小生成树,两个子树间距离可以用dp来做。
假设从root开始向下搜索,那么dp[root][u]表示root到u和以u为根节点的子树的最小距离,此处距离即是取的不在最小生成树中的原图的边。那么很容易发现是一个简单的树形DP过程。
然后,如果对于两个子树呢?两个子树是的并集是最小生成树,设两个子树的连接边为(u,v),那么此时两个子树最小距离为dp[k][v],k为u或者以u为根节点的子节点。
实际上,这种方法应该可以扩展到最小生成树的一些其他题目?
调代码的时候出现一些小插曲,prime算法携程dijkstra一直没改出来……prime算法加新点进去后,dis更新为dis[v] = min(dis[v], g[u][v]),Dijkstra为dis[v] = min(dis[v], dis[u] + g[u][v])
失分点/学习地方:
1. 两个子树距离
2. 最小生成树prime算法
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <string>
#include <algorithm>
#include <iostream>
#include <vector>
using namespace std;
#define inf (1000000007)
#define LL long long
const int MAXN = 3000 + 5;
int n, m;
int g[MAXN][MAXN], dp[MAXN][MAXN], best[MAXN][MAXN];
LL mst;
vector<int>e[MAXN];
///prime use
int vis[MAXN], dis[MAXN], pre[MAXN];
void prime()
{
for(int i = 0 ; i < n ; i++) vis[i] = 0, dis[i] = inf, pre[i] = -1;
dis[0] = 0;
vis[0] = 1;
for(int i = 1 ; i < n ; i++) {
if(dis[i] > dis[0] + g[0][i]) {
dis[i] = dis[0] + g[0][i];
pre[i] = 0;
}
}
mst = 0;
for(int T = 2 ; T <= n ; T++) {
int mindis = inf, re = -1;
for(int i = 0 ; i < n ; i++) {
if(vis[i] == 0 && dis[i] < mindis) {
mindis = dis[i];
re = i;
}
}
mst += mindis;
vis[re] = 1;
if(pre[re] != -1) {
e[pre[re]].push_back(re);
e[re].push_back(pre[re]);
}
for(int i = 0 ; i < n ; i++) {
if(vis[i] == 0 && dis[i] > g[re][i]) {
dis[i] = g[re][i];
pre[i] = re;
}
}
}
}
int dfs1(int u, int fa, int root)
{
// printf("u = %d, fa = %d, root = %d\n", u, fa, root);
for(int i = 0 ; i < (int)e[u].size() ; i++) {
int v = e[u][i];
if(v == fa) continue;
dp[root][u] = min(dp[root][u], dfs1(v, u, root));
}
if(fa != root) dp[root][u] = min(dp[root][u], g[root][u]);
return dp[root][u];
}
int dfs2(int u, int fa, int root)
{
int ans = dp[u][root];
for(int i = 0 ; i < (int)e[u].size() ; i++) {
int v = e[u][i];
if(v == fa) continue;
ans = min(ans, dfs2(v, u, root));
}
return ans;
}
int main()
{
while(scanf("%d%d", &n, &m) != EOF && n + m) {
for(int i = 0 ; i <= n ; i++) for(int j = 0 ; j <= n ; j++) g[i][j] = dp[i][j] = best[i][j] = inf;
for(int i = 0 ; i <= n ; i++) e[i].clear();
for(int i = 1 ; i <= m ; i++) {
int u, v, w; scanf("%d%d%d", &u, &v, &w);
g[u][v] = g[v][u] = w;
}
prime();
// printf("mst = %d\n", mst);
// puts("pre");
// for(int i = 0 ; i < n ; i++) printf("%d ", pre[i]);
// puts("\npre");
for(int i = 0 ; i < n ; i++) dfs1(i, -1, i);
for(int i = 0 ; i < n ; i++) {
for(int j = 0 ; j < (int)e[i].size() ; j++){
int v = e[i][j];
best[v][i] = dfs2(v, i, i);
}
}
// printf("\ndp\n");
// for(int i = 0 ; i < n ; i++) {
// for(int j = 0 ; j < n ; j++) printf("%d ", dp[i][j]);
// printf("\n");
// }
// puts("dp");
// printf("best\n");
// for(int i = 0 ; i < n ; i++) {
// for(int j = 0 ; j < n ; j++) printf("%d ", best[i][j]);
// printf("\n");
// }
// puts("best");
double res = 0;
int q; scanf("%d", &q);
for(int i = 0 ; i < q ; i++) {
int u, v, w; scanf("%d%d%d", &u, &v, &w);
int ok = 0;
for(int j = 0 ; j < (int)e[u].size() ; j++) {
if(e[u][j] == v) {
ok = 1;
break;
}
}
// printf("ok = %d, g[u][v] = %d\n", ok, g[u][v]);
if(ok == 0) res += mst;
else {
double res1 = 1.0 * mst - g[u][v] + w;
double res2 = 1.0 * mst - g[u][v] + best[u][v];
// printf("res1 = %d, res2 = %d\n", res1, res2);
res += min(res1, res2);
}
}
printf("%.4f\n", res / (1.0 * q));
}
return 0;
}