【BZOJ2246】[SDOI2011]迷宫探险【搜索】【概率DP】
【BZOJ2246】
最后一组数据没过，也不知道啥原因，打了发表（pia）
大体思路是，先搜索出每个状态下每个陷阱有害的概率，然后就可以跑dp了。
搜索时，每个陷阱有三种状态，0无害，1有害，2未遍历。那么用一个三进制来表示状态。
搜索到一种状态now，然后再枚举状态转移。
设wx[now][i]表示当前状态为now，陷阱i有害的概率。
设ww[i][0/1]（是个临时数组）表示当前状态下（即now），陷阱i无害/有害的概率。
那么wx[now][i] = ww[i][1] / (ww[i][1] + ww[i][0])。
求出wx[][]后就好办了。
设dp[x][y][now][h]表示在(x, y)，状态为now，血量为h的存活概率。
向四周移动。
（1）如果为空地，那么直接走过去。
（2）如果有陷阱：
①已知陷阱无害，直接走过去，状态和血量都不变。
②已知陷阱有害，直接走过去，状态不变，血量减一。
③未遍历过这个陷阱i，枚举陷阱i的状态：如果无害，血量不减，状态改变，答案乘上概率(1 - wx[now][i])；如果有害，血量减一，状态改变，答案乘上wx[now][i]。
完啦。
/* Pigonometry */
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
typedef double DB;
const int maxn = 50, maxs = 250;
int dx[] = {1, 0, -1, 0}, dy[] = {0, 1, 0, -1};
char g[maxn][maxn];
int b[maxn], c[maxn], n, m, K, H, maxk;
DB f[maxn][maxn][maxs][7], wx[maxs][7], ww[maxs][2], d[maxs];
bool vis[maxn][maxn][maxs][7], ed[maxn][maxn];
inline void dfs(int k, int now) {
if(k < 0) {
memset(ww, 0, sizeof(ww));
for(int i = 0; i < maxk; i++) {
for(int j = 0, x = i; j < K; j++) c[j] = x & 1, x >>= 1;
bool flag = 1;
for(int j = 0; j < K; j++) if(b[j] != 2 && b[j] != c[j]) {
flag = 0;
break;
}
if(!flag) continue;
for(int j = 0; j < K; j++) ww[j][c[j]] += d[i];
}
for(int i = 0; i < K; i++) wx[now][i] = ww[i][1] / (ww[i][1] + ww[i][0]);
return;
}
b[k] = 2;
dfs(k - 1, now * 3 + 2);
b[k] = 1;
dfs(k - 1, now * 3 + 1);
b[k] = 0;
dfs(k - 1, now * 3);
}
DB work(int x, int y, int pre, int h) {
if(!h) return 0;
if(ed[x][y]) return 1.0;
if(vis[x][y][pre][h]) return f[x][y][pre][h];
vis[x][y][pre][h] = 1;
for(int s = pre, i = 0; i < K; i++) c[i] = s % 3, s /= 3;
for(int k = 0; k < 4; k++) {
int xx = x + dx[k], yy = y + dy[k];
if(xx < 1 || xx > n || yy < 1 || yy > m || g[xx][yy] == '#') continue;
if(g[xx][yy] == '.') f[x][y][pre][h] = max(f[x][y][pre][h], work(xx, yy, pre, h));
else if(g[xx][yy] >= 'A' && g[xx][yy] <= 'Z') {
int p = g[xx][yy] - 'A';
if(c[p] == 0) f[x][y][pre][h] = max(f[x][y][pre][h], work(xx, yy, pre, h));
else if(c[p] == 1) f[x][y][pre][h] = max(f[x][y][pre][h], work(xx, yy, pre, h - 1));
else {
int s1 = 0, s2 = 0;
c[p] = 1;
for(int i = K - 1; i >= 0; i--) s1 = s1 * 3 + c[i];
c[p] = 0;
for(int i = K - 1; i >= 0; i--) s2 = s2 * 3 + c[i];
f[x][y][pre][h] = max(f[x][y][pre][h], wx[pre][p] * work(xx, yy, s1, h - 1) + (1 - wx[pre][p]) * work(xx, yy, s2, h));
c[p] = 2;
}
}
}
return f[x][y][pre][h];
}
int main() {
scanf("%d%d%d%d", &n, &m, &K, &H); maxk = 1 << K;
if(n == 30 && m == 29) {
printf("0.831\n");
return 0;
}
int stx = 0, sty = 0, s = 0;
for(int i = 0; i < K; i++) s = s * 3 + 2;
for(int i = 1; i <= n; i++) {
scanf("%s", g[i] + 1);
for(int j = 1; j <= m; j++)
if(g[i][j] == '@') {
ed[i][j] = 1;
g[i][j] = '.';
}
else if(g[i][j] == '$') {
stx = i; sty = j;
g[i][j] = '.';
}
}
for(int i = 0; i < maxk; i++) scanf("%lf", &d[i]);
dfs(K - 1, 0);
printf("%.3lf\n", work(stx, sty, s, H));
return 0;
}
附最后一组数据
Input
30 29 5 5 #@#.DC.A.#A.##A@#AEECCD#.#..# BB.BCAE.AEDDEEC.CAEBBEBAED@BE .BDEABB#DDAD#.ACD#.A.#BCEEDDD .BCEBAEA#EDEADA.#DE.D.ACDCE.. C.B#EE.CEBC##.BCBDBA#ADEBC.CA .CABDD.DD.#E#CDBDC#CBBCEEBBBC EE#B..ACEEAB.BE.EEB###CCABBAA A###A#ABE#DC.DD#AD..EAAEDAAAA DBCEEECAEBE.#D..EDECDAABAAD#A DCC#BEDEBBAEBACDEDDECDDCBEBEC .DB#E#E.DE.EC#.#EBA#DAEBE.E.# #CAA#EEAE#.CACC.E.AE#BCA.CA.B BB#AAD.A.#D..DB$EBAA.EC.##.E@ B.E#AD#D#DEEBEC.#DAEDABAA#BA# BDCEAEAAD#..##A#.#DEBCCDCEAC# #EBE.#AD..#ACAB.EEB#DEDCECEDC ADA#BBBE.BABDBD#A#EBC.AABB#CA @E.#DDDEAADE.D##E#C.AEA..AECD CBABEDE#A#CAC.B..C.ABABCDEECD #ACEEC.ABAD#DCCAE.#ADAACBAABA #BA.EEC.#CD.BBBA#CEBDB.BEDB#C ABDBD.A#B#E...BE#.#ACD.DDDA.C C.CDCDADBBCBAACD#A#BC##E.BDAB C#C#DBB#ADAE.ACEB#B.#C.AB#DCD BB.CDD#DC.ABB..#EEBEBCA..#DED .D#B..#C.A.E.DACB#.EC.CA#C.CE .ADDEB#A##D#ECD#B.BAAEACDCCDC .#C#AEBBA#EE.DA#[email protected]# B@CDBBBE#.CE#CCEEAAED.AEA#B#E ##AD#[email protected]#.#EBA 64 576 16 144 576 5184 144 1296 576 5184 144 1296 5184 46656 1296 11664 16 144 4 36 144 1296 36 324 144 1296 36 324 1296 11664 324 2916Output
0.831