ZOJ3613 Wormhole Transport（斯坦纳树）

题目链接

题目描述：
若干星球，有些星球上建立了若干个工厂，有些星球是资源星球
一个资源星球只能供应一座工厂，求最大可被供应的工厂数以及在此条件下的最小花费

分析：
有可能一个星球即是资源星球又是工厂建设地吗？应该是有可能的
遇到这种情况，我只要让其中一座工厂在自己的星球上就地取材即可（剩下的工厂还是需要与其他资源星球连通）

还是朴素的斯坦纳树转移

注意一点，所有的状态都要满足一个条件：连通的星球上工厂的个数要大于等于资源的个数，这样才是一个合法的状态

最后再找到所含资源最多，花费最小的合法方案就是答案

tip

注意如果没有合法方案，花费是0：printf("%d %d\n",tt+ans,cost==INF? 0:cost);

#include<cstdio>
#include<cstring>
#include<iostream>
#include<queue>

using namespace std;

const int INF=0x3333333;
const int N=205;
int n,m,st[N],tot=0,ans=0;
int er[N],fac[N],rec[N],fac_cnt,rec_cnt;
struct node{
    int y,v,nxt;
};
node way[10010];
int f[N][1100],g[1100];
bool in[N];
queue<int> q;

void add(int u,int w,int z) {
    tot++;way[tot].y=w;way[tot].v=z;way[tot].nxt=st[u];st[u]=tot;
    tot++;way[tot].y=u;way[tot].v=z;way[tot].nxt=st[w];st[w]=tot;
}

void spfa(int S) {
    while (!q.empty()) {
        int now=q.front(); q.pop();
        in[now]=0;
        for (int i=st[now];i;i=way[i].nxt) {
            int y=way[i].y;
            if (f[y][S]>f[now][S]+way[i].v) {
                f[y][S]=f[now][S]+way[i].v;
                if (!in[y]) q.push(y),in[y]=1;
            }
        }
    }
}

int check(int a) {
    int t=0;
    for (int i=1;i<=n;i++) 
        if (a&er[i]) t+=fac[i],t-=rec[i];    
    return t>=0;                                  //工厂>资源 
}

int getnum(int a) {
    int t=0;
    for (int i=1;i<=n;i++)
        if (a&er[i]) t+=rec[i];
    return t;
}

void dp() {
    int cnt=1<<(rec_cnt+4);
    for (int S=1;S<cnt;S++) {
        for (int s=(S-1)&S;s;s=(s-1)&S)
            for (int i=1;i<=n;i++)
                f[i][S]=min(f[i][S],f[i][s]+f[i][S^s]);
        for (int i=1;i<=n;i++)
            if (f[i][S]<INF&&!in[i])
                q.push(i),in[i]=1;
        spfa(S);
    } 

    for (int i=1;i<cnt;i++) 
        for (int j=1;j<=n;j++)
            g[i]=min(g[i],f[j][i]);

    int cost=INF,tt=0;      
    for (int i=1;i<=cnt;i++) 
        if (check(i)) {                             //合法状态 
            for (int j=(i-1)&i;j;j=(j-1)&i) 
                if (check(j)&&check(i^j))
                    g[i]=min(g[i],g[j]+g[i^j]);     //转移合法状态
            int tmp=getnum(i);                      //此状态有多少资源
            if (tmp>tt) tt=tmp,cost=g[i];
            else if (tmp==tt) cost=min(cost,g[i]); 
        }
    printf("%d %d\n",tt+ans,cost==INF? 0:cost);
}

void init() {
    memset(st,0,sizeof(st)); tot=0;
    memset(in,0,sizeof(in));
    memset(er,0,sizeof(er));
    memset(f,0x33,sizeof(f));
    memset(g,0x33,sizeof(g));
    while (!q.empty()) q.pop();
    fac_cnt=0,rec_cnt=0,ans=0;
}

int main()
{
    while (scanf("%d",&n)!=EOF) {
        init();

        for (int i=1;i<=n;i++) {
            scanf("%d%d",&fac[i],&rec[i]);
            if (fac[i]&&rec[i]) fac[i]--,rec[i]--,ans++;      //既有工厂又有资源 
            if (fac[i]) {
                er[i]=1<<fac_cnt; fac_cnt++;
                f[i][er[i]]=0;
            }
            if (rec[i]) {
                er[i]=1<<(rec_cnt+4); rec_cnt++;
                f[i][er[i]]=0;
            }
        }

        scanf("%d",&m);
        for (int i=1;i<=m;i++) {
            int u,w,z;
            scanf("%d%d%d",&u,&w,&z);
            add(u,w,z);
        }
        dp();
    }
    return 0;
}