[UVAlive4297]First Knight

Posted 2020-10-25 cjfdf

题面在这里

题意

给定一个\(n\times m\)的格网,从\((1,1)\)出发,每一格\((i,j)\)往上下左右移动的概率已经给出,询问到达\((n,m)\)的期望步数

数据范围

\[n,m\le40\]

sol

列方程高斯消元求解是显然的
\[E_{i,j}=1+p1_{i,j}\times E_{i+1,j}+p2_{i,j}\times E_{i,j+1}+p3_{i,j}\times E_{i-1,j}+p4_{i,j}\times E_{i,j-1}\]
但是直接套高斯消元复杂度为\(O(n^3m^3)\),会\(TLE\)
于是考虑优化

因为每项方程的系数只有\(4\)个,可以知道最后得到的矩阵满足一定的规律
比如当\(n=4,m=3\)时，系数矩阵如下
\[ \left[ \begin{matrix} 1&-p2_{1,1}&0&0&-p1_{1,1}&0&0&0&0&0&0&0\-p4_{1,2}&1&-p2_{1,2}&0&0&-p1_{1,2}&0&0&0&0&0&0\0&-p4_{1,3}&1&-p2_{1,3}&0&0&-p1_{1,3}&0&0&0&0&0\0&0&-p4_{1,4}&1&0&0&0&-p1_{1,4}&0&0&0&0\-p3_{2,1}&0&0&0&1&-p2_{2,1}&0&0&-p1_{2,1}&0&0&0\0&-p3_{2,2}&0&0&-p4_{2,2}&1&-p2_{2,2}&0&-p1_{2,2}&0&0&0\0&0&-p3_{2,3}&0&0&-p4_{2,3}&1&-p2_{2,3}&0&-p1_{2,3}&0\0&0&0&-p3_{2,4}&0&0&-p4_{2,4}&1&-p2_{2,4}&0&-p1_{2,4}\0&0&0&0&-p3_{3,1}&0&0&0&1&-p2_{3,1}&0&0\0&0&0&0&0&-p3_{3,2}&0&0&-p4_{3,2}&1&-p2_{3,2}&0\0&0&0&0&0&0&-p3_{3,3}&0&0&-p4_{3,3}&1&-p2_{3,3}\0&0&0&0&0&0&0&0&0&0&0&1\\end{matrix} \right]\times \left[ \begin{matrix} E_{1,1}\ E_{1,2}\ E_{1,3}\ E_{1,4}\ E_{2,1}\ E_{2,2}\ E_{2,3}\ E_{2,4}\ E_{3,1}\ E_{3,2}\ E_{3,3}\ E_{3,4}\\end{matrix} \right]= \left[ \begin{matrix} 1\1\1\1\1\1\1\1\1\1\1\0\\end{matrix} \right] \]
(建议缩小屏幕后食用)

在这个矩阵中,每行的系数都占据了\((2m+1)\)的长度,且以\(f_{i,i}\)为中心
因此我们在高斯消元的时候,只需要消除后\(m\)行中的\(m\)个系数
枚举每一行是\(O(nm)\)的,因此总复杂度即为\(O(nm^3)\)
于是就可以过了......

代码

#include<bits/stdc++.h>
#include<algorithm>
#include<iostream>
#include<cstdlib>
#include<iomanip>
#include<cstring>
#include<complex>
#include<vector>
#include<cstdio>
#include<string>
#include<bitset>
#include<cmath>
#include<queue>
#include<stack>
#include<map>
#include<set>
#define mp make_pair
#define pb push_back
#define RG register
#define il inline
using namespace std;
typedef unsigned long long ull;
typedef vector<int>VI;
typedef long long ll;
typedef double dd;
const dd eps=1e-10;
const int mod=1e9+7;
const int N=40;
const int M=50010*2;
il ll read(){
    RG ll data=0,w=1;RG char ch=getchar();
    while(ch!=‘-‘&&(ch<‘0‘||ch>‘9‘))ch=getchar();
    if(ch==‘-‘)w=-1,ch=getchar();
    while(ch<=‘9‘&&ch>=‘0‘)data=data*10+ch-48,ch=getchar();
    return data*w;
}

il void file(){
    freopen("a.in","r",stdin);
    freopen("a.out","w",stdout);
}

int n,m;
dd p[5][N+10][N+10],g[N*N+10][N*N+10];

il void gauss(int nm){
    for(RG int i=1;i<=nm;i++){
        for(RG int j=i;j<=min(nm,i+m);j++)
            if(g[j][i]>eps){swap(g[i],g[j]);break;}
        for(RG int j=i+1;j<=min(nm,i+m);j++){
            g[j][nm+1]-=g[i][nm+1]*g[j][i]/g[i][i];
            for(RG int k=min(nm,i+m);k>=i;k--)
                g[j][k]-=g[i][k]*g[j][i]/g[i][i];
        }
    }   
    for(RG int i=nm;i;i--){
        for(RG int j=i+1;j<=min(nm,i+m);j++)
            g[i][nm+1]-=g[j][nm+1]*g[i][j];
        g[i][nm+1]/=g[i][i];
    }
}

int main()
{
    while(1){
        n=read();m=read();if(!n)return 0;
        for(RG int k=1;k<=4;k++)
            for(RG int i=1;i<=n;i++)
                for(RG int j=1;j<=m;j++)
                    scanf("%lf",&p[k][i][j]);
        
        for(RG int i=1;i<=n*m;i++)
            for(RG int j=1;j<=n*m+1;j++)
                g[i][j]=0;
        
        for(RG int i=1;i<=n;i++)
            for(RG int j=1;j<=m;j++){
                RG int x=(i-1)*m+j;g[x][x]=g[x][n*m+1]=1;
                if(i<n)g[x][i*m+j]=-p[1][i][j];
                if(j<m)g[x][(i-1)*m+j+1]=-p[2][i][j];
                if(i>1)g[x][(i-2)*m+j]=-p[3][i][j];
                if(j>1)g[x][(i-1)*m+j-1]=-p[4][i][j];
                g[x][n*m+1]=1;
            }
        g[n*m][(n-1)*m]=g[n*m][n*m-1]=0;g[n*m][n*m+1]=0;
        
        gauss(n*m);
        printf("%.6lf\n",g[1][n*m+1]);
    }
    return 0;
}