这是一道求方案数的题。我很菜,不太会写方案数的题,所以看了下题解,并写下这篇博客。
首先,我们是在一个n*m的格子里走,有两种走法,一个是向右,一个是向下。起点任意,终点任意。在你走的路径上,第奇数个点是小a取走其中的液体,第偶数个点是uim取走其中的液体,必须走偶数个点。我们所要求的是,走到终点后,小a和uim的液体数一样的路径的方案数。
觉得题不好写,主要是觉得需要限制的条件太多了。首先起点是什么不知道,小a与uim的液体数量也得考虑进去。最开始不知道脑子哪抽了在想动态转移方程,,这明摆着是个枚举所有方案的题,不过是我们用dp的思想优化递推而已。至于起点,在最初给dp数组赋值的时候考虑进去,每个点都当成起点。k(小a的液体),l(uim的液体)不好递推下去,本来就不用递推下去,直接把他们的值当做dp数组的下标。那么思路就清晰了,枚举i,j和k,l,然后把当前方案数加到可以移动的下一个位置,每次当k==l的时候,ans加上当前的值,递推下去就好了。
不过这道题n,m的范围是800,k与l的范围又是16,怎么说复杂度都不过关。我们发现,当我们加答案给ans的时候,只考虑了k与l的差值。所以我们没有必要枚举k与l的值,只需要枚举差值就可以,当然最后再加一个维度存储当前这个点由谁来取。递推答案的时候,先枚举点i,j,然后枚举差值,向下扩展就可以了。复杂度为O(n*m*k)。
#include<bits/stdc++.h>
using namespace std;
const int mod=1000000007;
int ans,n,m,k,f[810][810][16][2],a[810][810];
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]);a[i][j]%=k;
f[i][j][a[i][j]][0]=1;
}
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
for(int h=0;h<k;h++){
if(i+1<=n){
f[i+1][j][(h+a[i+1][j])%k][0]=(f[i+1][j][(h+a[i+1][j])%k][0]+f[i][j][h][1])%mod;
f[i+1][j][(h-a[i+1][j]+k)%k][1]=(f[i+1][j][(h-a[i+1][j]+k)%k][1]+f[i][j][h][0])%mod;
}
if(j+1<=m){
f[i][j+1][(h+a[i][j+1])%k][0]=(f[i][j+1][(h+a[i][j+1])%k][0]+f[i][j][h][1])%mod;
f[i][j+1][(h-a[i][j+1]+k)%k][1]=(f[i][j+1][(h-a[i][j+1]+k)%k][1]+f[i][j][h][0])%mod;
}
if(h==0)
ans=(ans+f[i][j][h][1])%mod;
}
printf("%d\n",ans);
return 0;
}