UVa1496/LA5717 Peach Blossom Spring

 题目链接

        本题是2011年 ACM ICPC 亚洲区域赛 北京 赛区 的 E 题。

题意

        有n个房子(4 ≤ n ≤ 50)，前k个房子每个里面住着一户人家k (1 ≤ k ≤ 5, 2k ≤ n)，末k的房子是隐蔽点。这些房子间总共有m条道路(0 ≤ m ≤ 1000)，但所有道路都是损坏的，每条道路的维修费用是w(1 ≤ w ≤ 1000)。请计算最小的维修费用，使得前k户人家和最后k个隐蔽点一一对应。

分析

        经典问题：图论中的斯坦纳树。

        斯坦纳树动态规划的状态转移中第二部分本质是最短路问题，可以直接用Dijkstra或SPFA来处理。也可以用floyd预处理更新u、v之间的边权w[u][v]，使其为修复u、v连通的最小费用，则状态转移中第二部分可以直接推进，效率更高。

        另外，本题有坑（导致我WA了很多次）：两个房子间的道路可能有多条，读取边权时要保存最小值。

AC代码

#include <iostream>
using namespace std;

#define K 5
#define N 50
#define INF 1100000
int f[N][1<<(K<<1)], d[1<<K][1<<K], w[N][N], c[1<<K]={0}, n, m, k;

void solve() {
    cin >> n >> m >> k;
    int x = (1<<(k<<1))-1, y = (1<<k)-1;
    for(int i=0; i<n; ++i) {
        for (int j=0; j<n; ++j) w[i][j] = INF;
        for (int s=x; s>0; --s) f[i][s] = INF;
    }
    for (int i=0; i<k; ++i) f[i][1<<i] = 0, f[n-k+i][1<<(i+k)] = 0;
    for (int i=1; i<=y; ++i) {
        c[i] = c[(i-1)&i] + 1;
        for (int j=y; j>0; --j) d[i][j] = INF;
    }
    while (m--) {
        int u, v, c; cin >> u >> v >> c;
        --u; --v; w[v][u] = w[u][v] = min(w[u][v], c);
    }
    for (int k=0; k<n; ++k) for (int i=0; i<n; ++i) for (int j=0; j<n; ++j) w[i][j] = min(w[i][j], w[i][k]+w[k][j]);
    for (int s=1; s<=x; ++s) {
        for (int i=0; i<n; ++i) for (int t=(s-1)&s; t>0; t=(t-1)&s) f[i][s] = min(f[i][s], f[i][t]+f[i][s^t]);
        for (int i=0; i<n; ++i) for (int j=0; j<n; ++j) f[i][s] = min(f[i][s], f[j][s]+w[i][j]);
        for (int i=0, l, r; i<n; ++i) if (c[l = s&y] == c[r = s>>k]) d[l][r] = min(d[l][r], f[i][s]);
    }
    for (int i=1; i<=y; ++i) for (int j=1; j<=y; ++j) if (c[i] == c[j]) {
        for (int s=(i-1)&i; (s<<1)>i; s=(s-1)&i) for (int t=(j-1)&j; t>0; t=(t-1)&j) if (c[s] == c[t])
            d[i][j] = min(d[i][j], d[s][t]+d[i^s][j^t]);
    }
    d[y][y] < INF ? cout << d[y][y] << endl : cout << "No solution" << endl;
}

int main() {
    ios::sync_with_stdio(false);
    short t; cin >> t;
    while (t--) solve();
    return 0;
}