POJ-2531 Network Saboteur (DFS小水题)

题目链接:https://vjudge.net/problem/POJ-2531
题意:给一个NxN矩阵,第i行第j个元素表示第i个点到第j个点的距离,将点集划分为两个子集A,B,求(所有(A中元素到B中元素距离的和)的和)的最大值。N<=20。

万幸的是这题数据量小,真的让我直接暴力的水过了,稍微大一点对我来说就麻烦了,调用函数浪费的时间真的特别多,如果要优化就尽量减少递归的层数。

代码如下:

#include<cstdio>
using namespace std;
int c[25][25], flag[25], ans, n, cnt;
inline void dfs(int row, int tip) {
	flag[row] = tip;
	if (row == n - 1) {
		int sum = 0;
		for (int i = 0; i < n; i++) {
			if (flag[i])continue;
			for (int j = 0; j < n; j++)
				if (flag[j])sum += c[i][j];
		}
		if (sum > ans)ans = sum;
		return;
	}
	dfs(row + 1, 0);
	dfs(row + 1, 1);
}
int main(void) {
	scanf("%d", &n);
	for (int i = 0; i < n; i++)
		for (int j = 0; j < n; j++)
			scanf("%d", &c[i][j]);
	ans = 0;
	dfs(0, 0);
	dfs(0, 1);
	printf("%d\n", ans);
	return 0;
}
发布了104 篇原创文章 · 获赞 97 · 访问量 4515

猜你喜欢

转载自blog.csdn.net/TK_wang_/article/details/105102659
今日推荐