小a和uim之大逃离(luogu P1373 dp)
给你一个n*m的矩阵,其中元素的值在1~k内。限制只能往下和往右走,问从任意点出发,到任意点结束,且经过了偶数个元素的合法路径有多少个。在此题中,定义在一条路径中,第奇数个元素之和为X,第偶数个元素之和为Y。X+Y同模k+1的路径是合法路径。答案模1e9+7。n,m<=800,1<=k<=15。
状态是\(f[i][j][h][0/1]\),表示第i行第j列的元素,奇数元素和X-偶数元素和Y差为h,当前在路径中是第奇数个点或第偶数个点的方案数。
设\(p1=(p-a[j]+k+1)\% (k+1)\),\(p2=(p+a[j])\%(k+1)\)。那么状态转移方程就是:\[f[i][j][p][0]=f[i][j-1][p1][1]+f[i-1][j][p1][1]+(p1?0:1)\]\[f[i][j][p][1]=f[i][j-1][p2][0]+f[i-1][j][p2][0]\]
状压一下即可。
#include <cctype>
#include <cstdio>
#include <cstring>
using namespace std;
const int mod=1e9+7;
const int maxn=805, maxk=20;
typedef int ia3[maxn][maxk][2];
int n, m, k, ans, a[maxn];
ia3 fnow, fpre;
void getint(int &x){
char ch; int flag=1; x=0;
for (ch=getchar(); !isdigit(ch); ch=getchar())
if (ch=='-') flag=-1;
for (x=ch-48; ch=getchar(), isdigit(ch);)
x=x*10+ch-48;
x*=flag;
}
int main(){
scanf("%d%d%d", &n, &m, &k);
int p1, p2;
for (int i=1; i<=n; ++i)
for (int j=1; j<=m; ++j){
getint(a[j]);
for (int p=0; p<=k; ++p){
p1=(p-a[j]+k+1)%(k+1); //这一步若小A走
p2=(p+a[j])%(k+1); //这一步若uim走
fnow[j][p][0]=(fnow[j-1][p1][1]
+fpre[j][p1][1]+(p1?0:1))%mod;
fnow[j][p][1]=(fnow[j-1][p2][0]
+fpre[j][p2][0])%mod;
if (p==0) ans=(ans+fnow[j][p][1])%mod;
}
memcpy(fpre[j], fnow[j], sizeof(fnow[j]));
}
printf("%d\n", ans);
return 0;
}