jzoj3657 [NOI2014模拟]随机游走

Description

---

在一个二维平面上每次随意往四周一个方向上行走的模型叫随机游走。现在对这个模型进行一些修改：有一个n*m的矩阵，你初始站在（1，1），想要走到（n，m），每次随机往四周不超出矩形方向走一步，问走到（n，m）的期望步数。

Solution

---

第一眼裸题，学乖了看看数据范围发现反而不太会了
可以设第一行或第一列全部为未知数，表示点到（n，m）的期望距离，那么就能推出下一行/列，高斯消元即可。注意只有一行或只有一列的情况，略坑
这题显然按列做比较优秀

Code

---

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <math.h>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)
#define drp(i,st,ed) for (int i=st;i>=ed;--i)
#define fill(x,t) memset(x,t,sizeof(x))
#define copy(x,t) memcpy(x,t,sizeof(x))

const int N=205;

double last[N][N],a[N][N];

void gauss(int n) {
    rep(i,1,n-1) {
        rep(j,i+1,n) {
            if (fabs(a[j][i])>fabs(a[i][i])) std:: swap(a[i],a[j]);
        }
        rep(j,i+1,n) {
            double tmp=a[j][i]/a[i][i];
            rep(k,i,n*2) a[j][k]-=a[i][k]*tmp;
            a[j][0]-=a[i][0]*tmp;
        }
    }
    drp(i,n-1,1) {
        rep(j,i+1,n) {
            double tmp=a[i][j]/a[j][j];
            a[i][j]=0; a[i][0]-=a[j][0]*tmp;
            rep(k,n+1,n*2) a[i][k]-=a[j][k]*tmp;
        }
    }
}

int main(void) {
    int T; scanf("%d",&T);
    while (T--) {
        int n,m; scanf("%d%d",&n,&m);
        fill(a,0);
        rep(i,1,n) {
            a[i][i]=1; a[i][0]=1;
            if (i!=n) {
                double tmp=1;
                if (m!=1) tmp++;
                if (i!=1) tmp++;
                a[i][i+1]=-1.0/tmp;
                if (m!=1) a[i][i+n]=-1.0/tmp;
                if (i!=1) a[i][i-1]=-1.0/tmp;
            }
        }
        gauss(n);
        rep(i,1,m-1) {
            copy(last,a); fill(a,0);
            rep(j,1,n) {
                a[j][j]=1; a[j][0]=1;
                double tmp=1;
                if (j!=n) tmp++;
                if (j!=1) tmp++;
                if (i!=m-1) tmp++;
                if (j!=n) a[j][j+1]=-1.0/tmp;
                if (j!=1) a[j][j-1]=-1.0/tmp;
                if (i!=m-1) a[j][j+n]=-1.0/tmp;
                tmp=-1.0/tmp/last[j][j];
                rep(k,1,n) a[j][k]-=tmp*last[j][k+n];
                a[j][0]-=tmp*last[j][0];
            }
            gauss(n);
        }
        printf("%.0lf\n", (a[1][0]/a[1][1]-1));
    }
    return 0;
}