UVA 11795 Mega Man's Mission

洛克人最初有一把武器,可以殺死某些機器人,每殺死一個機器人,就可以獲得他的武器,每個武器可以殺死某些特定的機器人,計算出殺死這些所有的機器人順序總數? 

簡單狀態壓縮DP。dp[ S ] 表示殺死機器人狀態爲S(1表示死,0活着)的總方案數,先預處理出對於每個S,所獲得的武器能夠殺死哪些機器人。1表示可殺,0不可以。狀態轉移方程dp[S] += dp[S0]; 其中     S0爲S的子集 即S0=S^(1<<j) 且S0 擁有的武器能夠殺死j號機器人 即s[S0]&(1<<j) == 1




#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <iostream>
#include <vector>
#include <algorithm>
#include <queue>
using namespace std;
#define clr(a, x) memset(a, x, sizeof(a))
#define rep(i, n) for (int i = 0; i < (int)(n); i++)
#define REP(i,a,b) for(int i=a;i<=(b);i++)
typedef long long lld;

lld dp[1<<16];
int s[1<<16];
int a[20];
int trans(string str)
{
    int res=0;
    int len=str.length();
    for(int i=len-1;i>=0;i--)
    {
        res=(res<<1)+str[i]-'0';
    }
    return res;
}
int main()
{
    int T,cas=1;
    scanf("%d",&T);
    while(T--)
    {
        int n;
        scanf("%d",&n);
        string str;
        REP(i,0,n) cin>>str,a[i]=trans(str);
        clr(s,0);
        s[0]=a[0];
        REP(S,1,(1<<n)-1)
        {
            s[S]=a[0];
            rep(j,n)
            if(S&(1<<j)) s[S]|=a[j+1];
        }
        clr(dp,0);
        dp[0]=1;
        int all= (1<<n)-1;
        for(int S=1;S<=all;S++)
        {
            for(int j=0;j<n;j++)
                if(S&(1<<j))
            {
                int S0=S^(1<<j);
                if(s[S0]&(1<<j)) dp[S]+=dp[S0];
            }
        }
        printf("Case %d: %lld\n",cas++,dp[all]);
    }
    return 0;
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章