状压DP UVA 11795 Mega Man's Mission

 

题目传送门

/*
    题意：洛克人有武器可以消灭机器人，还可以从被摧毁的机器人手里得到武器，问消灭全部机器人的顺序总数
    状态压缩DP：看到数据只有16，就应该想到状压(并没有)。因为是照解题报告写的，代码里加点注释，省的以后忘记了
*/
/************************************************
* Author        :Running_Time
* Created Time  :2015-8-8 10:41:28
* File Name     :UVA_11759.cpp
 ************************************************/

#include <cstdio>
#include <algorithm>
#include <iostream>
#include <sstream>
#include <cstring>
#include <cmath>
#include <string>
#include <vector>
#include <queue>
#include <deque>
#include <stack>
#include <list>
#include <map>
#include <set>
#include <bitset>
#include <cstdlib>
#include <ctime>
using namespace std;

#define lson l, mid, rt << 1
#define rson mid + 1, r, rt << 1 | 1
typedef long long ll;
const int MAXN = 20;
const int INF = 0x3f3f3f3f;
const int MOD = 1e9 + 7;
ll dp[(1<<16)+10];
int w[MAXN];
int s[(1<<16)+10];
char mega[MAXN];
char robot[MAXN];
int n;

int main(void)    {     //UVA 11795 Mega Man's Mission
    int T, cas = 0;  scanf ("%d", &T);
    while (T--) {
        scanf ("%d", &n);
        scanf ("%s", mega);

        memset (w, 0, sizeof (w));
        for (int i=0; i<n; ++i)    {
            scanf ("%s", robot);
            for (int j=0; j<n; ++j) {
                if (robot[j] == '1')   {
                    w[i] |= (1 << j);           //求出每个机器人拥有的武器，用二进制累加
                }
            }
        }

        memset (s, 0, sizeof (s));
        for (int i=0; i<n; ++i) {
            if (mega[i] == '1') {
                s[0] |= (1 << i);               //求出洛克人手里的武器
            }
        }
        for (int i=1; i<(1<<n); ++i)    {       //i表示机器人死亡的情况，例如00000表示一个都没消灭，11111表示全部消灭
            s[i] = s[0];
            for (int j=0; j<n; ++j) {
                if (i & (1 << j))   {           //意思是当前第j个机器人被消灭
                    s[i] |= w[j];               //那么能得到它的武器
                }
            }
        }

        memset (dp, 0, sizeof (dp));    dp[0] = 1;      //一个都没被消灭的方案数为1
        for (int i=0; i<(1<<n); ++i)    {
            if (!dp[i]) continue;
            for (int j=0; j<n; ++j) {
                if ((s[i] & (1 << j)) && (i & (1 << j)) == 0)   {       //意思是当前有武器能消灭第j个机器人并且要消灭它
                    dp[i|(1<<j)] += dp[i];                              //累加到消灭之后的状态里
                }
            }
        }
        printf ("Case %d: %lld\n", ++cas, dp[(1<<n)-1]);        //111111全部都被消灭的方案数
    }

    return 0;
}