1. 程式人生 > >洛谷 P1373 小a和uim之大逃離

洛谷 P1373 小a和uim之大逃離

思路:
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; }