zoj 3613 Wormhole Transport (STNT + DP)

题目：http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=3613

题目大意：n个星球，有几个是有资源的，有几个有工厂，一个工厂需要一个资源星球来维持，有工厂的星球最多4个，有资源的星球最多4个，一个星球可能有几个工厂，但最多只有1个资源。给你m条边，问你在正常工作的工厂数最多的情况下，建的边的总长度最小。

思路：STNT + DP。先求STNT，因为答案有可能是森林，所以再做DP，d[ s ] = min( d[ s1 ] + d[ s - s1 ]) ，s1是s的子集 && check( s1 ) && check( s2 )，check 这里是用作判断，只有工厂数 >= 资源数的状态才是合法的（这一点很重要，需要自己想想！）最后答案找一遍存在状态的资源数，最大的就行。

自己做的时候，>= 这里先开始很想当然的写成 == 了，WA了几次，偷看了一下别人写的，才改过来的，。。。。 = =

代码如下：

#include<cstdio>
#include<cstring>
#include<queue>
#include<algorithm>
using namespace std;

const int INF = 0x0fffffff;

const int MAXN = 222;

int fac[MAXN],res[MAXN];

struct Edge
{
    int t,next,len;
} edge[5555<<1];

int tot,head[MAXN];

void add_edge(int s,int t,int len)
{
    edge[tot].t = t;
    edge[tot].len = len;
    edge[tot].next = head[s];
    head[s] = tot++;
}

struct Node
{
    int s,id;
    Node(){}
    Node(int a,int b) : s(a),id(b) {}
};

int d[1<<11][MAXN];

int val[MAXN];

queue <Node> q;

bool inq[1<<11][MAXN];

void spfa()
{
    while(!q.empty())
    {
        int s = q.front().s;
        int id = q.front().id;
        q.pop();
        inq[s][id] = 0;
        for(int e = head[id];e != -1; e = edge[e].next)
        {
            int v = edge[e].t;
            int len = edge[e].len;

            int to = s|val[v];
            int tmp = d[s][id] + len;
            if(d[to][v] > tmp)
            {
                d[to][v] = tmp;
                if(to == s && !inq[to][v])
                {
                    inq[to][v] = 1;
                    q.push(Node(to,v));
                }
            }
        }
    }
}

int contain(int a,int b)
{
    while(b)
    {
        if(b&1)
        {
            if(!(a&1)) return 0;
        }
        a >>= 1;
        b >>= 1;
    }
    return 1;
}

void stnt(int n,int k)
{
    int s_sum = 1<<k;
    for(int s = 1;s<s_sum;s ++)
    {
        for(int i = 0;i<n;i++)
        {
            if(val[i] && !contain(s,val[i])) continue;
            for(int s1 = (s - 1)&s ;s1 ;s1 = (s1 - 1)&s)
            {
                int s2 = s - s1;
                d[s][i] = min(d[s][i],d[s1|val[i]][i] + d[s2|val[i]][i]);
            }
            if(d[s][i] < INF) q.push(Node(s,i)),inq[s][i] = 1;
        }
        spfa();
        //for(int i = 0;i<n;i++)
            //printf("s = %d,i = %d,d = %d\n",s,i,d[s][i]);
    }
}

int check(int s,int cnt_p)
{
    int cnt1 = 0;
    for(int i = 0;i<cnt_p;i++)
    {
        if(s&1)
            cnt1++;
        s >>= 1;
    }
    int cnt2 = 0;
    while(s)
    {
        if(s&1)
            cnt2 ++;
        s >>= 1;
    }
    return cnt1 >= cnt2 ? 1 : 0;
}

int one(int s,int cnt_fac)
{
    s >>= cnt_fac;
    int cnt = 0;
    while(s)
    {
        if(s&1) cnt++;
        s >>= 1;
    }
    return cnt;
}

int d2[1<<11];

int main()
{
    int n;
    while(~scanf("%d",&n))
    {
        int sum = 0;
        for(int i = 0;i<n;i++)
        {
            scanf("%d%d",&fac[i],&res[i]);
            sum += fac[i] + res[i];
        }
        int s_sum = 1<<sum;
        for(int s = 0;s<s_sum;s++)
            for(int i = 0;i<n;i++)
                d[s][i] = INF;
        //printf("s_sum = %d\n",s_sum);
        int cnt_fac = 0;
        int cnt_res = 0;
        for(int i = 0;i<n;i++)
        {
            val[i] = 0;
            for(int j = 0;j<fac[i];j++)
            {
                val[i] += 1<<cnt_fac;
                cnt_fac ++ ;
            }
            for(int j = 0;j<res[i];j++)
            {
                val[i] += 1<<(sum - cnt_res - 1);
                cnt_res ++;
            }
            //printf("i = %d,val = %d\n",i,val[i]);
            d[val[i]][i] = 0;
        }

        int m;
        scanf("%d",&m);
        tot = 0;
        memset(head,-1,sizeof(head));
        while(m--)
        {
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            a--;
            b--;
            add_edge(a,b,c);
            add_edge(b,a,c);
        }

        stnt(n,sum);

        for(int s = 0;s < s_sum ;s++)
        {
            d2[s] = INF;
            for(int i = 0;i<n;i++)
                d2[s] = min(d2[s],d[s][i]);
        }
        for(int s = 1;s < s_sum; s++)
        {
            if(check(s,cnt_fac))
                for(int s1 = (s - 1)&s ; s1 ;s1 = (s1 - 1)&s)
                {
                    int s2 = s - s1;
                    if(check(s1,cnt_fac) && check(s2,cnt_fac))
                    {
                        d2[s] = min(d2[s],d2[s1] + d2[s2]);
                    }
                }
        }
        int ans1 = 0,ans2 = INF;
        for(int s = s_sum - 1; s >= 1;s--)
        {
            if(!check(s,cnt_fac)) continue;
            //printf("s = %d,d2 = %d\n",s,d2[s]);
            if(d2[s] < INF)
            {
                int tmp = one(s,cnt_fac);
                if(tmp > ans1)
                {
                    ans1 = tmp;
                    ans2 = d2[s];
                }
                else if(tmp == ans1)
                {
                    ans2 = min(ans2,d2[s]);
                }
            }
        }
        if(ans1 == 0) ans2 = 0;
        printf("%d %d\n",ans1,ans2);
    }
    return 0;
}