洛谷P1373 小a和uim的逃亡(动态规划，维护差值)

最新推荐文章于 2022-11-27 12:32:51 发布

Link_Ray

最新推荐文章于 2022-11-27 12:32:51 发布

阅读量158

点赞数

分类专栏：动态规划文章标签：差值

本文链接：https://blog.csdn.net/Link_Ray/article/details/88840633

版权

动态规划专栏收录该内容

34 篇文章 0 订阅

订阅专栏

博客详细介绍了如何利用动态规划解决洛谷P1373问题，通过维护小A和uim的魔法瓶差值优化状态设计，降低复杂度至O(n*n*k)，从而避免超时和内存溢出，实现100%数据的通过。

摘要由CSDN通过智能技术生成

https://www.luogu.org/problemnew/show/P1373

题解

这个状态很明显可以设计为dp[i][j][k1][k2][0/1]，代表走到(i,j)时，小A有k1，uim有k2，同时轮到谁走了。但这个转移方程要设计到维护4个变量，需要4重循环，只能过50%的数据。

对于100%的数据，这样设计不仅超时还要爆内存。所以需要优化一下状态设计。我们关注的点在于k1是否会等于k2，那么~~自然而然~~ 可以想到维护他们的差值即可。所以设计状态为dp[i][j][k][0/1]走到(i,j)时，他们的魔法瓶差值为k，k=(k1-k2)%k。复杂度就降到了 $O (n * n * k)$ 。
因为可以从任意点开始，所以初始化要对所有的dp[i][j][a[i][j]%k][0]都初始化为1。

代码

#include <bits/stdc++.h>
using namespace std;
#define FOR0(a,b) for(int i = a; i < b; ++i)
#define FORE(a,b) for(int i = a; i <= b; ++i)
typedef long long ll;
typedef pair<int,int> pii;
const int mod = 1e9+7;

int dp[805][805][16][2];
int a[805][805];

int main() {
	int n,m,k;
	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]);
			dp[i][j][a[i][j]%k][0] = 1;
		}
	int ans = 0;
	for(int i = 1; i <= n; ++i) {
		for(int j = 1; j <= m; ++j) {
			for(int kk = 0; kk < k; ++kk) {
				dp[i][j][kk][0] += dp[i-1][j][((kk-a[i][j])%k+k)%k][1];
				dp[i][j][kk][0] %= mod;
				dp[i][j][kk][0] += dp[i][j-1][((kk-a[i][j])%k+k)%k][1];
				dp[i][j][kk][0] %= mod;
				dp[i][j][kk][1] += dp[i][j-1][(kk+a[i][j])%k][0];
				dp[i][j][kk][1] %= mod;
				dp[i][j][kk][1] += dp[i-1][j][(kk+a[i][j])%k][0];
				dp[i][j][kk][1] %= 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;
}