【题解】BJOI-2017机动训练

Problem

bzoj & 洛谷

题目大意（题面又臭又长）：

给定一个地形图$n$行$m$列，移动类似于国际象棋中的王，但只能往终点移动，求从任意点到任意点的路径中的权值和，路径的权值即为所有路径中相同路径的数量（即求不同路径的数量的平方和）

Solution

一开始想了一个期望50的暴力加20的找规律，思想始终停在直接统计……

正解应该是将平方和转换为两个人走的路径一样的方案数，剩下就是就是记忆化搜索了，自己的记忆化搜索还不是很灵活……

剩下的在代码中有解释

Code

#include<bits/stdc++.h>
using namespace std;
#define rg register
#define cl1(x) memset(x,-1,sizeof(x))

template <typename _Tp> inline _Tp read(_Tp&x){
    char c11=getchar(),ob=0;x=0;
    while(c11^'-'&&!isdigit(c11))c11=getchar();if(c11=='-')c11=getchar(),ob=1;
    while(isdigit(c11))x=x*10+c11-'0',c11=getchar();if(ob)x=-x;return x;
}

const int N=31,p=1000000009;
int f[N][N][N][N],g[3][3][3][3];
int dx[4],dy[4],da[4],db[4];
char s[N][N];
int n,m,c1,c2;

int dp(int x,int y,int a,int b){
    if(s[x][y]!=s[a][b])return 0;
    if(x<1||x>n||y<1||y>m||a<1||a>n||b<1||b>m)return 0;
    if(~f[x][y][a][b])return f[x][y][a][b];
    int res(1);
    for(int i=0;i<c1;++i)for(int j=0;j<c2;++j)
        res=(res+dp(x+dx[i],y+dy[i],a+da[j],b+db[j]))%p;
    return f[x][y][a][b]=res;
}

int calc(int x,int y,int a,int b){//枚举完两人的方向，开始计算
    if(~g[x+1][y+1][a+1][b+1])return g[x+1][y+1][a+1][b+1];//g表示方向上的记忆化
    //下面八行算出移动向量
    c1=c2=0;
    for(int i=-1;i<2;++i)if(i==0||i==x)
    for(int j=-1;j<2;++j)if(j==0||j==y)if(i|j)
        dx[c1]=i,dy[c1]=j,++c1;
    for(int i=-1;i<2;++i)if(i==0||i==a)
    for(int j=-1;j<2;++j)if(j==0||j==b)if(i|j)
        da[c2]=i,db[c2]=j,++c2;
    //枚举两人的出发点
    int res(0);cl1(f);
    for(int i=1;i<=n;++i)for(int j=1;j<=m;++j)
    for(int I=1;I<=n;++I)for(int J=1;J<=m;++J)
        res=(res+dp(i,j,I,J))%p;

    //记录记忆化
    g[x+1][y+1][a+1][b+1]=g[-x+1][-y+1][-a+1][-b+1]=res;
    g[a+1][b+1][x+1][y+1]=g[-a+1][-b+1][-x+1][-y+1]=res;
    return res;
}

int dfs(int x,int y){//枚举第二个人的方向，容斥照常
    int res(0);
    res=(res+calc(x,y,1,1))%p;res=(res+calc(x,y,-1,-1))%p;
    res=(res+calc(x,y,1,-1))%p;res=(res+calc(x,y,-1,1))%p;
    res=(res-calc(x,y,1,0)+p)%p;res=(res-calc(x,y,0,-1)+p)%p;
    res=(res-calc(x,y,0,1)+p)%p;res=(res-calc(x,y,-1,0)+p)%p;
    return res;
}

int main(){
    read(n),read(m);cl1(g);
    for(rg int i=1;i<=n;++i)scanf("%s",s[i]+1);
    int ans=0;
    //对于方向上容斥一下，上下左右计算了两次，减去
    //下面两行是枚举第一个人的方向（起点-->终点）
    ans=(ans+dfs(1,1))%p;ans=(ans+dfs(-1,-1))%p;ans=(ans-dfs(1,0)+p)%p;ans=(ans-dfs(0,-1)+p)%p;
    ans=(ans+dfs(1,-1))%p;ans=(ans+dfs(-1,1))%p;ans=(ans-dfs(0,1)+p)%p;ans=(ans-dfs(-1,0)+p)%p;
    printf("%d\n",ans);
    return 0;
}