UVA 11795 - Mega Man's Mission（状态压缩DP）

题目链接：点击打开链接

思路：

很容易想到用d[i][s]表示杀死了i个机器人了， 杀死的机器人集合是s的方法数。

转移也很容易我就不说了， 我的转移复杂度是n^2的， 当然你也可以预处理一下某个机器人能被哪些机器人集合杀死， 直接用位运算与来判断， 使得转移变成O(n)

细节参见代码：

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <string>
#include <vector>
#include <stack>
#include <ctime>
#include <bitset>
#include <cstdlib>
#include <cmath>
#include <set>
#include <list>
#include <deque>
#include <map>
#include <queue>
#define Max(a,b) ((a)>(b)?(a):(b))
#define Min(a,b) ((a)<(b)?(a):(b))
using namespace std;
typedef long long ll;
typedef long double ld;
const double eps = 1e-6;
const double PI = acos(-1);
const int mod = 1000000000 + 7;
const int INF = 0x3f3f3f3f;
const ll INF64 = ll(1e18);
const int maxn = 17;
int T,n,m, vis[maxn][1<<maxn], kase = 0, init[maxn], able[maxn][maxn];
ll d[maxn][1<<maxn];
ll dp(int i, int s) { // 杀死前i个人，杀死的人的集合为s的方法数
    ll& ans = d[i][s];
    if(i == n) return 1LL;
    if(vis[i][s] == kase) return ans;
    vis[i][s] = kase;
    ans = 0;
    for(int j = 1; j <= n; j++) { // 枚举这次杀死哪个人
        if(s & (1<<j)) continue;
        if(init[j]) { // 如果初始武器就能杀死
            ans += dp(i+1, s | (1<<j));
        }
        else {
            bool ok = false;
            for(int k = 1; k <= n; k++) {
                if(s & (1<<k)) {
                    if(able[k][j]) { ok = true; break; }
                }
            }
            if(ok) ans += dp(i+1, s | (1<<j));
        }
    }

    return ans;
}
char s[maxn];
int main() {
    scanf("%d",&T);
    while(T--) {
        scanf("%d", &n);
        scanf("%s", s+1);
        for(int j = 1; j <= n; j++) {
            int cur = s[j]-'0';
            init[j] = cur;
        }
        for(int i = 1; i <= n; i++) {
            scanf("%s", s+1);
            for(int j = 1; j <= n; j++) {
                int cur = s[j] - '0';
                able[i][j] = cur;
            }
        }
        ++kase;
        ll ans = dp(0, 0);
        printf("Case %d: %lld\n", kase, ans);
    }
    return 0;
}