Mio visits ACGN Exhibition

题目链接

滚动数组+dp

dp[i][j][k] 表示第i行j列有k个1 

#include<iostream>
using namespace std;
const int mod=998244353;
long long a[505][505];
long long n,m,p0,q1;
long long dp[2][505][505*2];
int main(){
	scanf("%lld%lld%lld%lld",&n,&m,&p0,&q1);
	for(int i=1;i<=n;i++){
		for(int j=1;j<=m;j++){
			scanf("%lld",&a[i][j]);
		}
	} 
	for(int i=1,j=0;i<=m;i++){
		dp[1][i][j+=(a[1][i]==1)]=1;//初始化  
 	}
	for(int i=2;i<=n;i++){
		for(int j=1;j<=m;j++){
			for(int k=0;k<=i+j-1;k++){//统计所有当前走过路径的可能性方案 
			//从1,1到i,j共走了i+j-1个格子 
				if(a[i][j]==0){
					dp[i&1][j][k]=dp[(i-1)&1][j][k]+dp[i&1][j-1][k];//从左方以及上方继承状态 
					dp[i&1][j][k]%=mod;
				}
				else{
					if(k==0) dp[i&1][j][k]=0;//滚动数组的更新 
					dp[i&1][j][k]=dp[(i-1)&1][j][k-1]+dp[i&1][j-1][k-1];
					dp[i&1][j][k]%=mod;
				}
			}
		}
	}
	long long ans=0;
	for(int i=q1;n+m-1-i>=p0;i++) ans=(ans%mod+dp[n&1][m][i]%mod)%mod;//统计答案 
	printf("%lld",ans);
	return 0;
}