UVALive 4999 Electricity Connection

来源:互联网 发布:淘宝店招设计效果图 编辑:程序博客网 时间:2024/05/21 05:43

最小斯坦纳树:求图G的一个真子集里面的所有点连通,且总边权值和最小的生成树称为MinimalSteinerTree(最小斯坦纳树)。

模板:
dp[u][state]表示以u为根,指定集合中的点的连通状态为state的生成树的最小总权值。
1.通过连通状态的子集进行转移。
dp[u][state]=min{ dp[u][sub1]+dp[u][sub2] | sub1^sub2 == state }

2.在当前枚举的连通状态下,对该连通状态进行松弛操作。
dp[u][state]=min{ dp[v][state]+e[u][v] }

代码:

#include<bits/stdc++.h>using namespace std;#define CLR(A, X) memset(A, X, sizeof(A))const int INF = 0x3f3f3f3f;bool inq[70];char mp[10][10];int w[70], dir[4][2] = {1,0, -1,0, 0,-1, 0,1};int hou[1<<8], dp[70][1<<8];queue<int> Q;void spfa(int S) {    for(int i = 0; i < 64; i++) {        inq[i] = 0;        if(dp[i][S] != INF) {            Q.push(i);            inq[i] = 1;        }    }    while(!Q.empty()) {        int u = Q.front(); Q.pop();        inq[u] = 0;        for(int i = 0; i < 4; i++) {            int x = u/8+dir[i][0], y = u%8+dir[i][1];            if(x<0||y<0||x>=8||y>=8) continue;            int v = x*8+y;            int t = dp[u][S]+1+w[v];            if(dp[v][S] > t) {                dp[v][S] = t;                if(!inq[v]) {                    inq[v] = 1;                    Q.push(v);                }            }        }    }}int main() {    int T;    scanf("%d", &T);    while(T--) {        CLR(dp, INF); CLR(hou, -1);        int pl, pw, r, k = 0;        scanf("%d%d", &pl, &pw);        for(int i = 0; i < 8; i++) {            scanf("%s", mp[i]);            for(int j = 0; j < 8; j++) {                int u = i*8+j;                if(mp[i][j] == 'G') r = u, w[u] = 0;                else if(mp[i][j] == '.') w[u] = pl;                else if(mp[i][j] == 'W') w[u] = pw;                else { w[u] = 0; hou[1<<k] = u; k++; }            }        }        int all = (1<<k)-1;        for(int S = 0; S <= all; S++) {            if(~hou[S]) dp[hou[S]][S] = 0;            for(int i = 0; i < 64; i++) {                for(int S0 = (S-1)&S; S0; S0 = (S0-1)&S) {                    dp[i][S] = min(dp[i][S], dp[i][S0]+dp[i][S^S0]-w[i]);                }            }            spfa(S);        }        static int cas = 0;        printf("Case %d: %d\n", ++cas, dp[r][all]);    }    return 0;}