hdu 2531 dfs 无向图分区

题意:

  • 给了一个无向图的邻接矩阵,求这个无向图分为两区域,不同区域间节点距离最大值。

方法:

  • 用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;
}

你可能感兴趣的:(DFS)