题目链接: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;
}