試題 算法提高 最小方差生成樹
資源限制
時間限制：1.0s 內存限制：256.0MB
問題描述
給定帶權無向圖，求出一顆方差最小的生成樹。
輸入格式
輸入多組測試數據。第一行爲N,M，依次是點數和邊數。接下來M行，每行三個整數U,V,W，代表連接U,V的邊，和權值W。保證圖連通。n=m=0標誌着測試文件的結束。
輸出格式
對於每組數據，輸出最小方差，四捨五入到0.01。輸出格式按照樣例。
樣例輸入
4 5
1 2 1
2 3 2
3 4 2
4 1 1
2 4 3
4 6
1 2 1
2 3 2
3 4 3
4 1 1
2 4 3
1 3 3
0 0
樣例輸出
Case 1: 0.22
Case 2: 0.00
數據規模與約定
1<=U,V<=N<=50,N-1<=M<=1000,0<=W<=50。數據不超過5組。
題解
最小方差樹的概念就是，在n個結點的圖中，挑n-1個邊，首先要使這個圖連通，然後，這n-1條邊的權值的方差要最小，方差的計算公式爲：
可以看出，方差與邊的平均值有關，但是有一個難點是，隨着邊的加入，邊的平均值會動態地變化，很不利於解題。
這裏有一個思路：在總的邊的權和已知的情況下求解最小的方差，這是容易實現的，因爲邊權總和已知，然後結點數量(n)已知的話，邊的數量也就知道了(n-1)，這樣邊的平均值就知道了，這樣各條邊與均值的平方就是一個不動的常數了。此時我們再求最小生成樹的過程中，邊的權值就不是原來的權值了，而是其權值與平均值差值的平方，然後按照最小生成樹的步驟求出最小的方差樹。
AC代碼如下：使用克魯斯卡爾算法求MST，使用到了並查集，代碼來自https://www.cnblogs.com/asuml/p/6798307.html
#include <cstdio>
#include <algorithm>
using namespace std;
double const MAX = 10000000000000.0;
int n, m, tmp[1005], fa[55];
double ans;
struct Edge
{
int u, v;
double w, val;
}e[1005];
bool cmp(Edge a, Edge b)
{
return a.w < b.w;
}
void UF_set(int n)
{
for(int i = 1; i <= n; i++)
fa[i] = i;
}
int Find(int x)
{
return x == fa[x] ? x : fa[x] = Find(fa[x]);
}
void Union(int a, int b)
{
int r1 = Find(a);
int r2 = Find(b);
if(r1 != r2)
fa[r2] = r1;
}
void Kruskal(int sum)
{
UF_set(n);
int cnt = 0;
double f_all = 0;
double all = 0;
double ave = sum * 1.0 / (n - 1);
for(int i = 0; i < m; i++)
e[i].w = (e[i].val - ave) * (e[i].val - ave);
sort(e, e + m, cmp);
for(int i = 0; i < m; i++)
{
int u = e[i].u;
int v = e[i].v;
if(Find(u) != Find(v))
{
Union(u, v);
f_all += e[i].w;
all += e[i].val;
cnt ++;
}
if(cnt == n - 1)
break;
}
if((int)all == sum)
ans = min(ans, f_all);
}
int main()
{
int ca = 1;
while(scanf("%d %d", &n, &m) != EOF && (m + n))
{
// if(n == 1 || n == 2)
// {
// printf("0.00\n");
// continue;
// }
int minv = 0;
int maxv = 0;
ans = MAX;
for(int i = 0; i < m; i++)
{
scanf("%d %d %lf", &e[i].u, &e[i].v, &e[i].val);
tmp[i] = e[i].val;
}
sort(tmp, tmp + m);
for(int i = 0; i < n - 1; i++)
minv += tmp[i];
for(int i = m - 1; i > m - n; i--)
maxv += tmp[i];
for(int i = minv; i <= maxv; i++)
Kruskal(i);
ans = ans / (n - 1);
printf("Case %d: %.2f\n", ca++, ans);
}
}
可以看出，先按邊的權值從大到小枚舉，然後取其中n-1條邊，這n-1條邊的最大值就是從後往前取，最小值是從前往後取，然後最小生成樹的代價就會在這兩者之間，一一枚舉，如果找到了有這樣一個MST，那麼求該情況下最小的方差，最後在所有可能的方差中取最小的。
PS。編者有一個做法，沒有AC，功能上不大對，還是想與大家分享：
先選定一個點(a)，用其一條邊作爲出發邊，然後使用Prim算法求最小生成樹，配合優先隊列，每次從隊列取邊的時候，更新目前選定邊的均值，然後每次從隊列拿出的是與均值差最小的邊，直到訪問所有結點。這樣還需要從a的其他邊出發遍歷，因爲不知道是哪一條邊構成的MST，但是肯定有一條邊是屬於MST的，不然結點a就孤立了。然後取這幾種情況中方差的最小值。
代碼如下：
#include<bits/stdc++.h>
using namespace std;
struct edge {
int from;
int to;
int we;
edge(int _from = 0, int _to = 0, int _we = 0) { from = _from; to = _to; we = _we; }
};
vector<edge> V[51];
vector<edge> choose;
bool vis[51];
int N, M;
float fenzi, fenmu, shang;
float ans;
struct cmp {
bool operator()(edge a, edge b) {
return abs(a.we - shang) > abs(b.we - shang);
}
};
void prim() {
ans = 1e6;
for (int i = 0; i < V[1].size(); i++) {
priority_queue<edge, vector<edge>, cmp > Q;
memset(vis, 0, sizeof(vis));
fenzi = 0;
fenmu = 0;
choose.clear();
int code = 1;
Q.push(V[1][i]);
vis[1] = 1;
while (!Q.empty()) {
edge tmp = Q.top();
Q.pop();
if (code) {
code = 0;
for (int j = 0; j < V[1].size(); j++)
if (j != i)
Q.push(V[1][j]);
}
if (vis[tmp.to])
continue;
vis[tmp.to] = 1;
choose.push_back(tmp);
fenmu++;
fenzi += tmp.we;
shang = fenzi / fenmu;
for (int j = 0; j < V[tmp.to].size(); j++) {
Q.push(V[tmp.to][j]);
}
}
float h = 0;
for (int j = 0; j < choose.size(); j++) {
h += (choose[j].we - shang)*(choose[j].we - shang);
}
ans = min(ans, h / fenmu);
}
}
int main() {
int a, b, c;
int ccase = 0;
while (scanf("%d%d", &N, &M) && N&&M) {
ccase++;
for (int i = 0; i < 51; i++)
V[i].clear();
for (int i = 0; i < M; i++) {
scanf("%d%d%d", &a, &b, &c);
V[a].push_back(edge(a, b, c));
V[b].push_back(edge(b, a, c));
}
prim();
printf("Case %d: %.2f\n", ccase, ans);
}
return 0;
}