HDU 4418 Time travel (高斯消元求期望) 好题

这题上来就没读懂题。。。。。。

题意：一个人在数轴上向左或向右走，走到头就改变方向，每次走，走ｉ步的概率为Pi。求从起点到终点的步数的期望。

思路：这个题的关键一步在于把向左和向右这两个起初状态变为一个状态。做法是把原先n个位置变成2*n-2个位置。

举例：01234   ------> 01234321 这样 如果开始是从2向左走，就等价于从第7个位置开始往右走。

这样需要注意终点可能会有两个。

设dp[i] 为从i开始到终点的步数的期望值。

那么dp[i] = sigma(p[j]*(dp[i+j]+j))。然后逐个列方程，移项，高斯消元求解。

注意几个细节：

1.起点和终点相同的话直接输出0.00，不然n = 1时会RE。

2.如果终点不可达，输出“Impossible !”。

3.在过程中有的点是不可达的，对应的方程也是不可列的，在列式的时候对应的方程列为E[I] = 0即可。

我的代码：

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<queue>

using namespace std;
const int MAX = 305;
const int maxn = 305;

int n,m,y,x,d,mod,s,t1,t2;
double p[MAX],dp[MAX][MAX];
bool vis[MAX];
queue<int> que;

void init(){
    mod = 2 * n - 2;
    s = x;
    if(d == 1) s = (mod - x) % mod;
    t1 = y ; t2 = (mod - y) % mod;
    memset(dp,0,sizeof(dp));
    memset(vis,0,sizeof(vis));
}

bool BFS(){
    vis[s] = true;
    que.push(s);

    while(!que.empty()){
        int u = que.front();que.pop();
        for(int i = 1 ; i <= m ; i++){
            int v = (u + i) % mod;
            if(p[i] < 1e-6) continue;
            if(!vis[v]){
                vis[v] = true;
                que.push(v);
            }
        }
    }
    return vis[t1] || vis[t2];
}


void build(){
    for(int i = 0 ; i < mod ; i++){
        dp[i][i] = 1;dp[i][mod] = 0;
        if(i == t1 || i == t2 || !vis[i]) continue;
        for(int j = 1 ; j <= m ; j++){
            int k = (i + j) % mod;
            dp[i][k] -= p[j];
            dp[i][mod] += j*p[j];
        }
    }
}

bool gauss(double arr[maxn][maxn], int n) {
	for(int i = 0; i < n; i ++) {
		int pivot = i;
		for(int j = i+1; j < n; j ++)
			if(fabs(arr[j][i]) > fabs(arr[pivot][i]))
				pivot = j;
		if(pivot != i)	swap_ranges(arr[i]+i,arr[i]+n+1,arr[pivot]+i);
		if(fabs(arr[i][i]) < 1E-8)		return false;
		for(int j = n; j >= i; j --)	arr[i][j] /= arr[i][i];
		for(int j = i+1; j < n; j ++)
			for(int k = n; k >= i; k --)
				arr[j][k] -= arr[i][k] * arr[j][i];
	}
	for(int i = n-1; i >= 0; i --)
		for(int j = i+1; j < n; j ++)
			arr[i][n] -= arr[j][n]*arr[i][j];
	return true;
}


int main(){
    int cas;
    scanf("%d",&cas);
    while(cas--){
        scanf("%d%d%d%d%d",&n,&m,&y,&x,&d);
        for(int i = 1; i <= m ; i++){
            scanf("%lf",&p[i]);
            p[i] /= 100.0;
        }
        if(x == y){
            printf("0.00\n");
            continue;
        }
        init();
        if(!BFS()){
            printf("Impossible !\n");
            continue;
        }
        build();
        if(!gauss(dp,mod)) printf("Impossible !\n");
        else{
            printf("%.2f\n",dp[s][mod]);
        }
    }
    return 0;
}