Codeforces Round #745 (Div. 2) C - Portal

题目链接:https://codeforces.com/contest/1581/problem/C

单纯的二维前缀和只能O(n2m2),看了答案发现需要取后缀最小值,因为可以做到前后两部分互不干扰,时间复杂度为O(n2m)。

#include <bits/stdc++.h>
using namespace std;

const int N = 400 + 5; 

int t, n, m;

char g[N][N];
int sum[N][N], f[N];

int getSum(int x1, int y1, int x2, int y2) {
	return sum[x2][y2] - sum[x1 - 1][y2] - sum[x2][y1 - 1] + sum[x1 - 1][y1 - 1];
}

/*
从答案里面可以看出,本题计算所有贡献值都是通过前缀和计算的,
上下边,中心空白都要通过前缀和来计算,取后缀最小值来搭配前缀,
后缀部分和前缀部分是相互独立的关系,互不干扰,时间复杂度O(mn^2) 
*/

int main(void) {
//	freopen("in.txt", "r", stdin);
	scanf("%d", &t);
	while (t--) {
		scanf("%d%d", &n, &m);
		for (int i = 1; i <= n; i++) {
			scanf("%s", g[i] + 1);
			for (int j = 1; j <= m; j++) g[i][j] -= '0';
		}
		for (int i = 1; i <= n; i++)
			for (int j = 1; j <= m; j++)
				sum[i][j] = sum[i][j - 1] + g[i][j];
		for (int j = 1; j <= m; j++)
			for (int i = 1; i <= n; i++)
				sum[i][j] += sum[i - 1][j];
		int ans = n * m;
		for (int i = 1; i <= n; i++)
			for (int j = i + 4; j <= n; j++) {
				for (int r = 4; r<= m; r++) {
					f[r] = getSum(i + 1, 1, j - 1, r - 1); // empty相关部分 
					f[r] -= getSum(i, 1, i, r - 1) + getSum(j, 1, j, r - 1); // 上下边相关部分 
					f[r] += 2 * (r - 1); // 上下边相关部分 
					f[r] += (j - i - 1) - getSum(i + 1, r, j - 1, r); // 右边贡献 
				}
				for (int r = m - 1; r >= 4; r--)
					f[r] = min(f[r], f[r + 1]);
				for (int l = 1; l + 3 <= m; l++) {
					int cur = f[l + 3];
					cur -= getSum(i + 1, 1, j - 1, l); // empty相关部分
					cur += getSum(i, 1, i, l) + getSum(j, 1, j, l); // 上下边相关部分 
					cur -= 2 * l; // 上下边相关部分 
					cur += (j - i - 1) - getSum(i + 1, l, j - 1, l); // 左边贡献 
					ans = min(ans, cur);
				}
			}
		printf("%d\n", ans);
	}
	return 0;
}
上一篇:Codeforces #745 div2 C. Portal


下一篇:Spring Security使用原理解读