首页 > 代码库 > CodeForces 372B 脑洞大开的DP递推

CodeForces 372B 脑洞大开的DP递推

题目

做了两个多小时,脑洞大开,给了一个01矩阵,求以a,b,为左上角,c,d为右下角的矩阵内有多少包含部分全为0的子矩阵

对于这道题目,一开始就想到了DP递推,感觉而已,虽然准,可是做不出啊,想好了递推式子可是细节部分没办法去处理。看了CF上的题解,顿时脑洞大开,这个做法真的是太厉害了,这方法代码简洁明了,同时也提醒到了我,在方程假设出来后,对于转移的细节处理,

其实一开始我想到过这个递推式子

dp[i][j][k][l] += dp[i][j][k - 1][l] + dp[i][j][k][l - 1] - dp[i][j][k - 1][l - 1];对于这个式子,就还差一个,那就是包含(c,d)也就是右下角的矩阵数目,这个当时无法去求,现在觉得这方法真的太巧妙了,可以先开一个二维数组last[i][j],代表当前(i,j)这个点距离它同一行左边距离它最近的 一个1的距离,如果它左边没有1,那么就是距离左边第一列 也就是矩阵左边界的距离,

一开始设定一个值 l - j + 1,也就是左上角的点与右下角的点的 列数之间的距离,然后开始枚举从右下角到左上角的 行数 中的last[kk][l]的值,(i<=kk<=k),取两个之间晓得的那个加上,为什么呢

比如说:

000

000,这个子矩阵一开始l - j + 1 = 3,然后枚举行数,发现最终答案是6,数一数包含右下角的点的 矩阵个数确实是六,因为要包含右下角所以多一行其实就是个数多了一倍而已

再举个例子:

010

000,一开始l - j + 1 = 3,开始枚举行数,最终答案是4,为什么呢,因为要包含右下角的点,所以当你一旦遇到1的时候,那么其实从这个1所在位置开始的 左边上边部分都可以去掉了,只有这个1以及1上边的右边部分可以用了, 这方法真神!

int n,m,q;

int mp[40 + 5][40 + 5];

int dp[40 + 5][40 + 5][40 + 5][40 + 5];

int last[40 + 5][40 + 5];//与左边最近的一个1的距离,

void init() {
	memset(dp,0,sizeof(dp));
	memset(last,0,sizeof(last));
}

bool input() {
	while(cin>>n>>m>>q) {
		for(int i=1;i<=n;i++) {
			string s;
			cin>>s;
			for(int j=1;j<=m;j++) {
				mp[i][j] = s[j - 1] - '0';
			}
		}
		return false;
	}
	return true;
}

void slove() {
	for(int i=1;i<=n;i++) {
		for(int j=1;j<=m;j++) {
			if(mp[i][j]){last[i][j] = 0;continue;}
			last[i][j] = last[i][j - 1] + (mp[i][j] == 0);
		}
	}
	for(int i=1;i<=n;i++) {
		for(int j=1;j<=m;j++) {
			for(int k=i;k<=n;k++) {
				for(int l=j;l<=m;l++) {
					int tmp = l - j + 1;
					//int tt = 0;
					for(int kk = k;kk >= i;kk--) {
						tmp = min(tmp,last[kk][l]);
						dp[i][j][k][l] += tmp;
						//tt += tmp;
						//cout<<tt<<"****"<<endl;
					}
					dp[i][j][k][l] += dp[i][j][k - 1][l] + dp[i][j][k][l - 1] - dp[i][j][k - 1][l - 1];
				}
			}
		}
	}
}

void cal() {
	slove();
	while(q--) {
		int aa,bb,cc,dd;
		cin>>aa>>bb>>cc>>dd;
		cout<<dp[aa][bb][cc][dd]<<endl;
	}
}

void output() {

}

int main() {
	while(true) {
		init();
		if(input())return 0;
		cal();
		output();
	}
	return 0;
}


CodeForces 372B 脑洞大开的DP递推