洛谷P1373 小a和uim之大逃离

DP

题目传送门

定义$f[i][j][k][0/1]$表示当前在$(i,j)$，两个人魔瓶中魔液的差为k（小a的减去uim的），轮到小a/uim取时的方案数。

然后转移方程就很好推啦！

f[i][j][(p+a[i][j]+k)%k][0]+=f[i-1][j][p][1]+f[i][j-1][p][1];
f[i][j][(p-a[i][j]+k)%k][1]+=f[i-1][j][p][0]+f[i][j-1][p][0];

注意可以从任意一个格子开始，也可以从任意一个格子结束。

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#define MAXN 800
using namespace std;
const int MOD=1e9+7;
int n,m,k;
int f[MAXN+5][MAXN+5][16][2];
int a[MAXN+5][MAXN+5];
int main(){
    scanf("%d%d%d",&n,&m,&k); k++;
    for (int i=1;i<=n;i++)
        for (int j=1;j<=m;j++){
            scanf("%d",&a[i][j]);
            f[i][j][a[i][j]][0]=1;
        }
    for (int i=1;i<=n;i++)
        for (int j=1;j<=m;j++)
            for (int p=0;p<k;p++){
                (f[i][j][(p+a[i][j]+k)%k][0]+=(f[i][j-1][p][1]+f[i-1][j][p][1])%MOD)%=MOD;
                (f[i][j][(p-a[i][j]+k)%k][1]+=(f[i][j-1][p][0]+f[i-1][j][p][0])%MOD)%=MOD;
            }
    int ans=0;
    for (int i=1;i<=n;i++)
        for (int j=1;j<=m;j++)
            (ans+=f[i][j][0][1])%=MOD;
    printf("%d\n",ans);
    return 0;
}