P1373 小a和uim之大逃离 四维dp,维护差值

题目链接

对于这道题,dp是显然的,不过状态设计就比较奇葩,dp[i][j][h][which]表示在第i行,第j列,差值为h,0表示小a,1表示uim。哪一个人which取到了宝物,不用去记录他们两个人的值,维护差值转移即可。如果超过k+1的话,就直接膜就可以了。

初始状态:dp[i][j][a[i][j]][0]=1。(因为是小a先取)枚举i,j,h转移,h的差值不超过k

转移方程:

dp[i][j][h][0]+=dp[i-1][j][h-a[i][j]][1]%mod;小a的转移从uim来,取了a[i][j]差值就从h增大了,从上方左边转移而来
dp[i][j][h][0]+=dp[i][j-1][h-a[i][j]][1]%mod;
dp[i][j][h][1]+=dp[i-1][j][h+a[i][j]][0]%mod;uim同理,只不过他取了过后差值减小,转移同理
dp[i][j][h][1]+=dp[i][j-1][h+a[i][j]][0]%mod;

最后累加dp[i][j][0][1]的答案即可,因为最后他俩的数量一样,差值为0,最后是uim取。

代码:

#include<bits/stdc++.h>
using namespace std;
const int N=950;
const int mod=1000000007;
int ditu[N][N];
int dp[N][N][18][2];
long long ans;
int n,m,k;
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",&ditu[i][j]);
            dp[i][j][ditu[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++){
                dp[i][j][h][0]+=dp[i-1][j][(h-ditu[i][j]+k)%k][1]%mod;
                dp[i][j][h][0]+=dp[i][j-1][(h-ditu[i][j]+k)%k][1]%mod;
                dp[i][j][h][1]+=dp[i-1][j][(h+ditu[i][j]+k)%k][0]%mod;
                dp[i][j][h][1]+=dp[i][j-1][(h+ditu[i][j]+k)%k][0]%mod;
            }
        }
    }
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            ans=(ans+dp[i][j][0][1])%mod;
        }
    }
    printf("%lld\n",ans);
    return 0;
} 
View Code

 

posted @ 2019-08-16 23:32  JBLee  阅读(141)  评论(0编辑  收藏  举报