Mio visits ACGN Exhibition（dp，思维)

最新推荐文章于 2022-04-30 22:00:42 发布

Looy_cai

最新推荐文章于 2022-04-30 22:00:42 发布

阅读量272

点赞数 1

分类专栏：思维 dp 文章标签： dp 思维

本文链接：https://blog.csdn.net/lalala625/article/details/120946213

版权

思维同时被 2 个专栏收录

68 篇文章 0 订阅

订阅专栏

19 篇文章 0 订阅

订阅专栏

本文介绍了一种解决从地图起点到终点的路径问题，要求路径中0和1的数量满足特定条件的高效算法。通过DP方法，只记录0的数量，巧妙地避免了大规模状态空间。核心思路是利用dp[i][j][k]来表示当前位置及0的数量，有效减少了内存消耗。

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

传送门
题目：
在这里插入图片描述

示例1
输入

2 2 1 1
0 0
1 1
输出

2
示例2
输入

3 3 2 0
0 0 1
0 0 1
1 0 0
输出
6

题目大意：
$n\times m 的地图，其中每个格子上存1/0，求从（1,1）到（n,m)的路径中所有满足0的个数至少有p个、1的个数至少有q个的路径总数$

思路：
$首先：$
$1 . 如果用 d f s 去寻找很有可能超时；$
$2 . 如果 d p 用四维数组分别存放 x 轴坐标、 y 轴坐标和 0 、 1 的个数那么会超内存$
$所以：$
$需要我们换个角度去解题；$
$由分析我们可以发现，无论走哪条路都满足：$
$0 的个数 + 1 的个数 = n + m - 1;$
$由此可得： 1 的个数 = (n + m - 1) - 0 的个数$
$所以：可以通过 d p 标记不同 0 的数目的路径个数，最终统计 0 数目满足题目要求的所有路径之和即可；$

$优化过后 d p [i] [j] [k] 代表含义：$
$i = 1 : 当前行; i = 0 : 上一行; j : 当前列; k : 0 的个数$
$d p [0] [j] [k] : 上一行第 j 列 0 有 k 个时的路径数$
$d p [1] [j] [k] : 当前行第 j 列 0 有 k 个时的路径数$

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
//#define int long long
const int N=1e3+10;
const int mod=998244353;

int a[510][510],dp[2][510][N];

signed main(){
	ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
	int n,m,p,q;
	cin>>n>>m>>p>>q;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=m;j++)cin>>a[i][j];
	if(p+q>n+m-1){
		cout<<0<<endl;return 0;
	}
	if(a[1][1]){//起点为1时 
		dp[1][1][0]=1;
	}else dp[1][1][1]=1;//起点为0时 
	for(int i=1;i<=n;i++){
		for(int j=1;j<=m;j++){
			if(i==1&&j==1)continue;
			if(a[i][j]){
				for(int k=0;k<n+m;k++)//更新当列不同0个数时的路径数 
					dp[1][j][k]=(dp[1][j-1][k]+dp[0][j][k])%mod;
			}else{
				dp[1][j][0]=0;//当前第j列有0个0的路径数目为0 
				for(int k=1;k<n+m;k++){
					dp[1][j][k]=(dp[1][j-1][k-1]+dp[0][j][k-1])%mod;
// 					cout<<dp[1][j][k]<<' ';
				}
			}
		}
		for(int j=1;j<=m;j++)//更新，将当前行置为下一次循环的上一行 
			for(int k=0;k<n+m;k++)dp[0][j][k]=dp[1][j][k];
	}
	ll ans=0;
	for(int i=p;i<=m+n-1-q;i++)ans=(ans+dp[0][m][i])%mod;
	cout<<ans<<endl;
	return 0;
}