题目背景
小a和uim来到雨林中探险。突然一阵北风吹来,一片乌云从北部天边急涌过来,还伴着一道道闪电,一阵阵雷声。刹那间,狂风大作,乌云布满了天空,紧接着豆大的雨点从天空中打落下来,只见前方出现了一个披头散发、青面獠牙的怪物,低沉着声音说:“呵呵,既然你们来到这,只能活下来一个!”。小a和他的小伙伴都惊呆了!
题目描述
瞬间,地面上出现了一个n*m的巨幅矩阵,矩阵的每个格子上有一坨0~k不等量的魔液。怪物各给了小a和uim一个魔瓶,说道,你们可以从矩阵的任一个格子开始,每次向右或向下走一步,从任一个格子结束。开始时小a用魔瓶吸收地面上的魔液,下一步由uim吸收,如此交替下去,并且要求最后一步必须由uim吸收。魔瓶只有k的容量,也就是说,如果装了k+1那么魔瓶会被清空成零,如果装了k+2就只剩下1,依次类推。怪物还说道,最后谁的魔瓶装的魔液多,谁就能活下来。小a和uim感情深厚,情同手足,怎能忍心让小伙伴离自己而去呢?沉默片刻,小a灵机一动,如果他俩的魔瓶中魔液一样多,不就都能活下来了吗?小a和他的小伙伴都笑呆了!
现在他想知道他们都能活下来有多少种方法。
输入输出格式
输入格式:
第一行,三个空格隔开的整数n,m,k
接下来n行,m列,表示矩阵每一个的魔液量。同一行的数字用空格隔开。
输出格式:
一个整数,表示方法数。由于可能很大,输出对1000000007取余后的结果。
输入输出样例
输入样例#1: 复制
2 2 3
1 1
1 1
输出样例#1: 复制
4
题目大概就是这样,大家读完题以后可能马上会想到背包,这个题总结成一句话就是找出所有能保持两人魔瓶中魔液相等的情况,且使uim最后操作的方案。
题目思路出来了以后,我们马上想到的就是动态规划数组,但是有几个维度,每个维度表示什么,又需要重新考虑,前两个维度一定时表示当前位置i,j,这是毋庸置疑的。题目要求保证二人魔液数量相等,那么是不是代表着我们一定要开两个维度分别来表示两个人魔液的数量呢?
答案显然是不用的,我们如果开两个维度,在表示两个人魔液数量相等的时候,还需要一次判断操作,而且多开两个维度占用空间也是平方级的,显然十分不划算。正解是我们只开一个维度,表示二者魔液差,这样既节省了空间,又省去了一次判断操作。然后我们还需要一个维度,既然说题目要求最后一步需要uim操作,且第一步由小a开始,我们自然需要一个维度来表示最后一步是小a操作还是uim操作。
然后我们就要考虑状转了。
先把状转写出来:
dp[i][j][h][0]=(dp[i][j][h][0]+dp[i-1][j][(h-mp[i][j]+k)%k][1])%mod;
dp[i][j][h][0]=(dp[i][j][h][0]+dp[i][j-1][(h-mp[i][j]+k)%k][1])%mod;
dp[i][j][h][1]=(dp[i][j][h][1]+dp[i-1][j][(h+mp[i][j])%k][0])%mod;
dp[i][j][h][1]=(dp[i][j][h][1]+dp[i][j-1][(h+mp[i][j])%k][0])%mod;
嗯……
没毛病
就是这么长
我们来解释一下:f[i][j][p][q]表示他们走到(i,j),且两人魔瓶内魔液量的差为p时的方法数。q=0表示最后一步是小a走的,q=1表示最后一步是uim走的。题目中说魔瓶的容量为k,实际上就是动归时p需要对k+1取余数,即p只有0~k,k+1种可能。答案为所有f[i][j][0][1]的和。
这里还要强调一个问题,就是每个格子都有可能作为小a的起始格子,所以初始时对于所有i、j,f[i][j][mapp[i][j]][0]=1。
这样算法复杂度是O(n*m*k)。
嗯 血赚
这道题就这样结束了,最后附上代码:
int main(){
cin>>n>>m>>k;
k+=1;
for(int i=1;i<=n;i++) for(int j=1;j<=m;j++){cin>>mp[i][j];dp[i][j][mp[i][j]%k][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][j][h][0]+dp[i-1][j][(h-mp[i][j]+k)%k][1])%mod;
dp[i][j][h][0]=(dp[i][j][h][0]+dp[i][j-1][(h-mp[i][j]+k)%k][1])%mod;
dp[i][j][h][1]=(dp[i][j][h][1]+dp[i-1][j][(h+mp[i][j])%k][0])%mod;
dp[i][j][h][1]=(dp[i][j][h][1]+dp[i][j-1][(h+mp[i][j])%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;
cout<<ans<<endl;
return 0;
}
这其实是一道dp或者递推的经典题,非常适合拿来练手。