LightOJ - 1037 Agent 47(状压DP)

题目大意：有N个敌人，刚开始你有一把枪，每开一枪，可以对敌人造成1点伤害，你杀掉相应的敌人后，可以得到相应敌人的武器。

给你N个敌人的血量，和一个武器伤害矩阵，第i行第j列表示的是第i个敌人的武器，可以对第j个敌人造成1枪val[i][j]的伤害

问至少要开多少枪才能干掉所有的敌人

解题思路：状态压缩，用dp[i]表示干掉i状态下的敌人需要开多少枪

#include <cstdio>
#include <cstring>
#include <queue>
#include <algorithm>
using namespace std;
const int N = 20;
const int S = (1 << 15) + 10;

int n, cas = 1;
int life
, mat

;
int dp[S];
char str
;

void init() {
scanf("%d", &n);
for (int i = 0; i < n; i++)
scanf("%d", &life[i]);

for (int i = 0; i < n; i++) {
scanf("%s", str);
for (int j = 0; j < n; j++) {
mat[i][j] = str[j] - '0';
}
}
}

void solve() {
memset(dp, 0x3f, sizeof(dp));
dp[0] = 0;
queue<int> Q;
Q.push(0);

while (!Q.empty()) {
int s = Q.front();
Q.pop();
if (s == (1 << n) - 1) continue;
for (int i = 0; i < n; i++) {
if (s & (1 << i)) continue;
bool flag = false;
if (dp[s] + life[i] < dp[s | (1 << i)]) {
dp[s | (1 << i)] = dp[s] + life[i];
flag = true;
}
for (int j = 0; j < n; j++) {
if ((s & (1 << j) ) && mat[j][i]) {
int t = life[i] / mat[j][i];
t += (life[i] % mat[j][i] ? 1 : 0);
if (dp[s | (1 << i)] > dp[s] + t) {
dp[s | (1 << i)] = dp[s] + t;
flag = true;
}
}
}
if (flag) Q.push(s | (1 << i));
}
}
printf("Case %d: %d\n", cas++, dp[(1 << n) - 1]);
}

int main() {
int test;
scanf("%d", &test);
while (test--) {
init();
solve();
}
return 0;
}