洛谷 P1373 小a和uim之大逃離
阿新 • • 發佈:2018-12-24
思路:
f[i][j][k][0/1]表示在座標(i,j),該0(小a)或1(uim)取的方案數。
轉移方程:
f[i][j][k][0]=(f[i][j][k][0]+f[i][j-1][(k-a[i][j]+K)%K][1]+f[i-1][j][(k-a[i][j]+K)%K][1])%md;
f[i][j][k][1]=(f[i][j][k][1]+f[i][j-1][(k+a[i][j])%K][0]+f[i-1][j][(k+a[i][j])%K][0])%md;
程式碼:
#include<bits/stdc++.h>
using namespace std;
#define maxn 800
#define maxk 15
#define read(x) scanf("%d",&x)
#define ll long long
#define md ((int)1e9+7)
int n,m,K;
int a[maxn+5][maxn+5];
int f[maxn+5][maxn+5][maxk+5][2];
int main() {
read(n),read(m),read(K);
K++;
for(int i=1;i<=n;i++) {
for(int j=1;j<=m;j++) {
read(a[i][j]);
}
}
int ans=0;
for(int i=1;i<=n;i++) {
for(int j=1;j<=m;j++) {
f[i][j][a[i][j]%K][0]=1;
for(int k=0;k<=K;k++) {
f[i][j][k][0]=((ll)f[i][j][k][0]+f[i][j-1][(k-a[i][j]+K)%K][1]+f[i-1][j][(k-a[i][j]+K)%K][1])%md;
f[i][j][k][1]=((ll)f[i][j][k][1]+f[i][j-1][(k+a[i][j])%K][0]+f[i-1][j][(k+a[i] [j])%K][0])%md;
}
ans=((ll)ans+f[i][j][0][1])%md;
}
}
printf("%d",ans);
return 0;
}