Light OJ 1037 - Agent 47（预处理状态压缩DP）

 1 #include<cstdio>
 2 #include<cstring>
 3 #include<iostream>
 4 #include<algorithm>
 5 #include<cmath>
 6 #include<queue>
 7 #include<vector>
 8 #include<map>
 9 using namespace std;
10 typedef long long LL;
11 const int INF = 1e9+7;
12 const int MAXN = 60055;
13 int dp[MAXN], Att[MAXN][20];
14 int HP[20], M[20][20], n, Lim;
15 int DFS(int sta)
16 {
17     if(dp[sta] != -1)
18         return dp[sta];
19     if(sta == 0)
20         return dp[0] = 0;
21     dp[sta] = INF;
22     for(int i=0; i<n; i++)
23     {
24         if( sta&(1<<i) )
25         {
26             int newSta = sta - (1<<i);
27             int k = (HP[i] + Att[newSta][i] - 1)/Att[newSta][i];
28             dp[sta] = min(dp[sta], k + DFS(newSta));
29         }
30     }
31     return dp[sta];
32 }
33 
34 int main()
35 {
36     int T, cas = 1;
37     char str[50];
38     scanf("%d", &T);
39     while(T --)
40     {
41         scanf("%d", &n);
42         memset(dp, -1, sizeof(dp));
43         memset(Att, 0, sizeof(Att));
44         for(int i=0; i<n; i++)
45         {
46             scanf("%d", &HP[i]);
47             Att[0][i] = 1;
48         }
49         for(int i=0; i<n; i++)
50         {
51             scanf("%s", str);
52             for(int j=0; j<n; j++)
53                 M[i][j] = str[j] - ‘0‘;
54         }
55         Lim = (1<<n) - 1;
56         /**把所有的攻击力状态预处理出来*/
57         for(int i=0; i<=Lim; i++)///30000
58         for(int j=0; j<n; j++)///15
59         {
60             if( (i&(1<<j)) )
61             {
62                 int newSta = i - (1<<j);
63                 for(int k=0; k<n; k++)
64                     Att[i][k] = max(Att[newSta][k], M[j][k]);
65             }
66         }
67         printf("Case %d: %d\n",cas ++, DFS(Lim) );
68     }
69     return 0;
70 }