A - Mio visits ACGN Exhibition(dp)

A - Mio visits ACGN Exhibition

借鉴了:https://blog.csdn.net/qq_63010655/article/details/123699263
并做了些许优化

dp[i][j][k]来表示 到第i行第j列,经过了k个0的所有路径数。
又可以用0和1两个状态来优化i
妙啊,orz

AC代码:

#include<bits/stdc++.h>
using namespace std;
#define ll long long 
#define r0 return 0
#define endl "\n"
#define pii pair<int,int>
#define inf 0x3f3f3f3f
const int N = 2e5 + 5;
int g[505][505];
ll dp[2][505][10005];
const int mod = 998244353;
int main()
{
	ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
	int n, m, sum0, sum1;
	cin >> n >> m >> sum0 >> sum1;
	for (int i = 1; i <= n; i++)
		for (int j = 1; j <= m; j++)
			cin >> g[i][j];
	if (g[1][1] == 1) dp[1][1][0] = 1; // 初始化,默认第一行是1
	else if (g[1][1] == 0) dp[1][1][1] = 1; // 第三维表示0的个数 ,整体表示路径的个数
	int s = 1; //用s来优化行,每一行只需要不一样就可以了所以用0和1来表示.
	// 默认第一行是1
	for (int i = 1; i <= n; i++)
	{
		for (int j = 1; j <= m; j++)
		{
			if (i == 1 && j == 1) continue;
			for (int k = 0; k <= i + j - 1; k++)
			{
				if (g[i][j] == 0)
				{ 
					if (k == 0) dp[s][j][k] = 0; //这里记得赋值,因为必须覆盖掉上一行的
					else
					dp[s][j][k] = (dp[s ^ 1][j][k - 1] + dp[s][j - 1][k - 1])%mod;
				}
				else
				{
					dp[s][j][k] = (dp[s ^ 1][j][k] + dp[s][j - 1][k])%mod;
				}

			}
		}
		s ^= 1; // 换行
	}
	s ^= 1; // 多换了一次,换回来
	int ans = 0;
	for (int k = sum0; n + m - 1 - k >= sum1; k++)
	{
		ans = (ans + dp[s][m][k]) % mod;

	}
	cout << ans << endl;
	r0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值