题意:
- 给了一个无向图的邻接矩阵,求这个无向图分为两区域,不同区域间节点距离最大值。
方法:
- 用dfs枚举每个点在哪个区域即可。
- 这里开始我用的是每次枚举到树的叶子节点后,在计算距离,这样每次计算都是
n2
,改为每变化一次就改变距离之后时间缩小了5倍,因为这样每达到一个叶子节点均摊的计算量都不足n。
- 这里第二种代码,中dfs函数的参数列表多了一个sum来维护距离
枚举到叶子节点再计算的代码: (1766MS)
#include <set>
#include <stack>
#include <queue>
#include <vector>
#include <cstdio>
#include <map>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#define LL long long int
using namespace std;
const int M=29,INF=0x3fffffff;
int diagram[M][M],n,str[M],ans=-INF;
int cal(void){
int sum=0;
for(int i=1;i <= n;i++){
if(str[i])
for(int j=1;j <= n;j++){
if(!str[j]){
sum+=diagram[i][j];
}
}
}
return sum;
}
void dfs(int x){
if(x > n){
int x=cal();
if(x > ans) ans=x;
return;
}
for(int i=0;i < 2;i++){
str[x]=i;
dfs(x+1);
str[x]=!i;
}
return;
}
int main(void){
while(~scanf("%d",&n)){
ans=-INF;
for(int i=1;i <= n;i++){
for(int j=1;j <= n;j++){
scanf("%d",&diagram[i][j]);
}
}
dfs(1);
printf("%d\n",ans);
}
return 0;
}
每做一次改变就改变距离的代码:(297MS)
#include <set>
#include <stack>
#include <queue>
#include <vector>
#include <cstdio>
#include <map>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#define LL long long int
using namespace std;
const int M=29,INF=0x3fffffff;
int diagram[M][M],n,str[M],ans=-INF;
void dfs(int x, int sum){
if(x > n){
if(sum > ans) ans = sum;
return;
}
for(int i=0;i < 2;i++){
int temp = 0, key = str[x];
if(str[x] != i) {
str[x] = i;
for(int j = 1; j <= n; j++) {
if(j != x && str[j] == i) temp -= diagram[j][x];
else if(str[j] != i) temp += diagram[j][x];
}
}
dfs(x + 1, sum + temp);
str[x] = key;
}
return;
}
int main(void){
while(~scanf("%d",&n)){
ans=-INF;
for(int i=1;i <= n;i++){
for(int j=1;j <= n;j++){
scanf("%d",&diagram[i][j]);
}
}
dfs(1, 0);
printf("%d\n",ans);
}
return 0;
}