UVA 11795 Mega Man's Missions

状态压缩DP

题意： 洛克人现在手上有一个武器，能够杀死指定的敌人，而杀死敌人后能够获得新的武器，新的武器也只能杀指定的人。问杀死所有敌人共有多少种不同的顺序。数据范围n<=16。

解法： 状态压缩DP，注意要维护每个状态下能杀死的敌人。我写的记忆化搜索。

设dp[s]，s为杀死的敌人的集合dp[s]为要求的方法数，则dp[s] = sigma(dp[s的子集])。具体见代码。

/* **********************************************
Author      : Nero
Created Time: 2013-8-25 19:30:31
Problem id  : UVA 11795
Problem Name: Mega Man's Missions
*********************************************** */

/*

状态压缩DP。
WA四次的原因，是没有注意起始状态与单点状态之间的转移，导致单点实际上是不能使用最初始的武器。

*/

#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
#define REP(i,a,b) for(int i=(a); i<(int)(b); i++)
#define clr(a,b) memset(a,b,sizeof(a))
typedef long long lld;

int n;
lld dp[(1<<16) + 10];
int cc[(1<<16) + 10];
char s[20];

lld gao(int s) {
    if(dp[s] >= 0) return dp[s];
    dp[s] = 0;
    for(int i = 0; (1<<i) <= s; i ++) {
        if(s & (1<<i)) {
            if((cc[s^(1<<i)] & (1<<i)) == 0) continue;
            dp[s] += gao(s^(1<<i));
        }
    }
    return dp[s];
}

int main() {
    int cas, ca = 0;
    scanf("%d", &cas);
    while(cas--) {
        scanf("%d", &n);
        scanf("%s", s);
        clr(cc,0);
        REP(i,0,n) if(s[i] == '1') cc[0] |= (1<<i);
        REP(i,0,n) {
            scanf("%s", s);
            REP(j,0,n) if(s[j] == '1') cc[1<<i] |= (1<<j);
            cc[1<<i] |= cc[0];
        }
        int all = (1<<n) - 1;
        for(int s = 2; s <= all; s ++) {
            int cnt = 0, cur;
            for(int i = 0; (1<<i) <= s; i ++) {
                if(s&(1<<i)) cnt ++, cur = i;
            }
            if(cnt == 1) continue;
            cc[s] = cc[1<<cur] | cc[s ^ (1<<cur)];
        }
        clr(dp,-1);
        dp[0] = 1;
        printf("Case %d: ", ++ca);
        printf("%lld\n", gao(all));
    }
    return 0;
}