概率dp-九度-1546-迷宫问题

题目链接：

http://ac.jobdu.com/problem.php?pid=1546

题目意思：

有一个起点S，多个出口E，#代表不能走，每次等概率的随机选择下一个可以行走的位置，求从S到出口的期望。

解题思路：

高斯消元求解期望。

先BFS预处理能够到达的出口的位置，然后如果从起点不能到达终点，直接输出-1.

然后对于无效的点，置该未知数的解为-1，否则依据dp[i][j]=1+dp[i-1][j]*1/4+dp[i][j+1]*1/4+dp[i+1][j]*1/4+dp[i][j-1]*1/4,构建n*m个方程，注意有些位置的可行位置数小于4，为cnt的话，此时的下一步概率为1/cnt.

然后解方程，求出唯一解。

PS:

解方程时，如果有的未知数有解，有的无解，可以将无解的情况置一个特殊值，然后按有唯一解的方式来解方程，避免无解未知数对有解未知数的影响。

方程系数要清零。wa了好几次。

代码：

#include<iostream>
#include<cmath>
#include<cstdio>
#include<sstream>
#include<cstdlib>
#include<string>
#include<cstring>
#include<algorithm>
#include<vector>
#include<map>
#include<set>
#include<stack>
#include<list>
#include<queue>
#include<ctime>
#include<bitset>
#define eps 1e-8
#define INF 0x3f3f3f3f
#define PI acos(-1.0)
#define ll __int64
#define LL long long
#define lson l,m,(rt<<1)
#define rson m+1,r,(rt<<1)|1
#define M 1000000007
#pragma comment(linker, "/STACK:1024000000,1024000000")
using namespace std;

#define Maxn 20

char sa[Maxn][Maxn];
int n,m,num,dir[4][2]={{-1,0},{0,1},{1,0},{0,-1}};
double dp[Maxn][Maxn],pp[Maxn][Maxn];
double g[Maxn*Maxn][Maxn*Maxn],ans[Maxn*Maxn];
bool vis[Maxn][Maxn];

void gaosi(int r,int c)
{
for(int i=0,j=0;i<r&&j<c;i++,j++)
{
int t=i;
for(int p=i+1;p<=r;p++)
if(fabs(g[p][j])>fabs(g[t][j]))
t=p;
if(fabs(g[t][j])<eps) //这是多解的情况
continue;
if(t-i) //不相同，则交换
{
for(int p=j;p<=c;p++)
swap(g[t][p],g[i][p]);
}
for(int p=i+1;p<=r;p++)
{
double tmp=g[p][j]/g[i][j];
if(fabs(tmp)<eps)
continue;
g[p][j]=0.0;
for(int q=j+1;q<=c;q++)
g[p][q]-=tmp*g[i][q];
}
}
for(int p=r;p>=0;p--)
{
if(fabs(g[p][p])<eps) //无解的情况
continue;
ans[p]=g[p][c];
for(int q=r;q>p;q--)
ans[p]-=g[p][q]*ans[q];
ans[p]/=g[p][p];
}
}
bool iscan(int x,int y)
{
if(x<0||x>=n||y<0||y>=m)
return false;
return true;
}
int sx,sy;

bool bfs() //从S出发找到所有可行的位置
{
bool flag=false;
queue<pair<int,int> >myq;
myq.push(make_pair(sx,sy));
memset(vis,false,sizeof(vis));
vis[sx][sy]=true;

while(!myq.empty())
{
int x=myq.front().first;
int y=myq.front().second;

myq.pop();
for(int i=0;i<4;i++)
{
int ans=x+dir[i][0],yy=y+dir[i][1];
if(!iscan(ans,yy)||vis[ans][yy]||sa[ans][yy]=='#')
continue;
vis[ans][yy]=true;
myq.push(make_pair(ans,yy));
if(sa[ans][yy]=='E') //能够到达终点
flag=true;
}
}
return flag;
}
double dl(double a)
{
if(fabs(a)<eps)
return 0;
return a;
}

int main()
{
while(~scanf("%d%d",&n,&m))
{
int ss;

for(int i=0;i<n;i++)
{
scanf("%s",sa[i]);
for(int j=0;j<m;j++)
{
if(sa[i][j]=='S')
{
sx=i,sy=j;
ss=i*m+j;
}
}
}
if(!bfs()) //不能够到达终点
{
printf("-1\n");
continue;
}
num=-1;
int last=n*m;

memset(g,0,sizeof(g));
for(int i=0;i<n;i++)
for(int j=0;j<m;j++)
{
num++;
int tmp=i*m+j;
if(!vis[i][j]) //无解的位置
{
g[num][tmp]=1;
g[num][last]=-1;
}
else
{
if(sa[i][j]=='E') //终点位置
{
g[num][tmp]=1;
g[num][last]=0;
continue;
}
int cnt=0,next[5];
for(int k=0;k<4;k++) //求出下一步可行的位置数
{
int x=i+dir[k][0],y=j+dir[k][1];
if(iscan(x,y)&&sa[x][y]!='#'&&vis[x][y])
{
cnt++;
next[cnt]=x*m+y;
}
}
g[num][tmp]=1;
g[num][last]=1;
for(int k=1;k<=cnt;k++) //构建当前位置的方程
g[num][next[k]]=-1.0/(cnt*1.0);
}
}
gaosi(n*m-1,n*m);
ans[ss]=dl(ans[ss]);
if(ans[ss]>0)
printf("%.2f\n",ans[ss]);
else
printf("-1\n");
}
return 0;
}