思路:首先给小uim改个名,叫他小b好了,那么我们定义dp[i][j][k][0/1]为,在i,j点,小a小b的魔瓶差值为k,0代表小a吸收了这一格的魔液,1则是小b吸收了这一格的魔液,有几种方法到达这种状态。这个k坑了我好久。。由于小a与小b两者魔瓶的差值可以为k,可是当k模k的时候得数却是0,所以我们得让k加1,开始数组开小了也WA了好几次。
通过定义的dp数组我们可以得到递推式
1.dp[i][j][l][0] = dp[i][j][l][0] + dp[i-1][j][(l-maps[i][j]+k)%k][1] + dp[i][j-1][(l-maps[i][j]+k)%k][1]
2.dp[i][j][l][1] = dp[i][j][l][1] + dp[i-1][j][(l+maps[i][j])%k][0] + dp[i][j-1][(l+maps[i][j])%k][0]
主要是第三维不好写,我们设第三维应该填x,那么1中的l%k=((a+maps)-b)%k,等价于 (l-maps)%k=(a-b)%k,所以1中的第三维填l-maps[i][j]+k)%k,+k主要是为了防止括号内为负数,2中l%k=(a-(b+maps))%k,等价于(l+maps)%k=(a-b)%k,所以2中的第三维填(l+maps[i][j])%k。那么就可以编写代码了
#include <iostream>
#include <cstring>
#include <cstdio>
#include <cstdlib>
using namespace std;
const int maxn = 800+2, MOD = 1e9+7;
int n, m, k, dp[maxn][maxn][17][2], maps[maxn][maxn];
long long ans;
int main() {
ios::sync_with_stdio(false);
/**************input***************/
cin >> n >> m >> k; k++;
for(int i = 1; i <= n; i++) {
for(int j = 1; j <= m; j++) {
cin >> maps[i][j];
maps[i][j] %= k;
dp[i][j][maps[i][j]][0]++;
}
}
/**********************************/
for(int i = 1; i <= n; i++) {
for(int j = 1; j <= m; j++) {
if(i == 1 && j == 1) continue;
for(int l = 0; l <= k; l++) {
dp[i][j][l][0] += dp[i-1][j][(l-maps[i][j]+k)%k][1] % MOD + dp[i][j-1][(l-maps[i][j]+k)%k][1] % MOD;
dp[i][j][l][0] %= MOD;
dp[i][j][l][1] += dp[i-1][j][(l+maps[i][j])%k][0] % MOD + dp[i][j-1][(l+maps[i][j])%k][0] % MOD;
dp[i][j][l][1] %= MOD;
}
}
}
for(int i = 1; i <= n; i++) {
for(int j = 1; j <= m; j++) {
ans += dp[i][j][0][1];
ans %= MOD;
}
}
cout << ans << endl;
return 0;
}