其他分享
首页 > 其他分享> > 洛谷 P2258 子矩阵

洛谷 P2258 子矩阵

作者:互联网

前言

那个 Atcoder Beginner 263 的 E 还真是恶心……
呃,我什么也没说,我什么也没说……

正文

题意

有个 $ n \times m $ 的矩阵,从里面选 $ r $ 行 $ c $ 列出来。
问这 $ r $ 行 $ c $ 列的交叉点“相邻元素的差”的和最少为多少。

$ 60 pts $

思路

直接暴力枚举。
先枚举 $ r $,再枚举 $ c $。

时间复杂度:$ O(C _{n} ^ {r} C _{m} ^{c} rc) $

空间复杂度:$ O(r + c) $

代码

代码就不放了,谁都会写。

$ 100 pts $

思路

是不是有很多人觉得满分根本不用枚举,$ O(n^k) $ 的暴力就搞定了?
我告诉你们吧,正解还是要枚举的……
先选出 $ r $ 行来,然后再来 $ DP $。
定义 $ f_{i, j} $ 为前 $ i $ 列选 $ j $ 列,第 $ i $ 列必须选,和最小为多少。
容易得到 $ f_{i, j} = f_{k, j - 1} + sg_{i} + tc_{k, i} (j - 1 \le k \lt i)$,其中 $ sg_{i} $ 为单单第 $ i $ 列的差之和(single),而 $ tc_{i, j} $ 为单单第 $ i $ 列与第 $ j $ 列横向的差之和(two columns)。


好了,天真的我以为这样就结束了。
然而……

总之,我想到的用山峰的高度表示是:
1 2 3 5 7 8 10 (一路顺风)
实际上的是:
1 100 1 100 1 100 1 (坎坷不平)
AZ……


时间复杂度:$ O(C _{n} ^{r} m^{2}) $

空间复杂度:$ O(m^{2}) $

代码

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

int a[20][20];
int choose[20];
int f[20][20];

int ans = 0x3f3f3f3f;
int n, m, r, c;
int yyy_2013_OI[20], yyy_2013_Math[20][20];

void brute_force() {
	for (int j = 1; j <= m; j++) {
		yyy_2013_OI[j] = 0;
		for (int i = 1; i <= r - 1; i++) {
			yyy_2013_OI[j] += abs(a[choose[i]][j] - a[choose[i + 1]][j]);
		}
	}
	for (int i = 1; i <= m; i++) {
		for (int j = i + 1; j <= m; j++) {
			yyy_2013_Math[i][j] = 0;
			for (int k = 1; k <= r; k++) {
				yyy_2013_Math[i][j] += abs(a[choose[k]][i] - a[choose[k]][j]);
			}
		}
	}
}

void dp() {
	for (int i = 1; i <= m; i++) {
		for (int j = 1; j <= min(i, c); j++) {
			if (j == 1) {
				f[i][j] = yyy_2013_OI[i];
			} else if (j == i) {
				f[i][j] = f[i - 1][j - 1] + yyy_2013_OI[i] + yyy_2013_Math[i - 1][i];
			} else {
				f[i][j] = 0x3f3f3f3f;
				for (int k = j - 1; k <= i - 1; k++) {
					f[i][j] = min(f[i][j], f[k][j - 1] + yyy_2013_OI[i] + yyy_2013_Math[k][i]);
				}
			}
			if (j == c) {
				ans = min(ans, f[i][c]);
			}
		}
	}
}

void dfs(int lvl) {
	if (lvl == r + 1) {
		brute_force();
		dp();
		return;
	}
	for (int i = choose[lvl - 1] + 1; i <= n; i++) {
		choose[lvl] = i;
		dfs(lvl + 1);
		choose[lvl] = 0;
	}
}

int main() {
	scanf("%d %d %d %d", &n, &m, &r, &c);
	for (int i = 1; i <= n; i++) {
		for (int j = 1; j <= m; j++) {
			scanf("%d", &a[i][j]);
		}
	}
	dfs(1);
	printf("%d", ans);
	return 0;
}

感谢你能看到这儿,来一个推荐吧!!!

标签:洛谷,P2258,int,复杂度,矩阵,枚举,20,sg,tc
来源: https://www.cnblogs.com/AProblemSolver/p/16563385.html