POJ 2531 Network Saboteur（DFS+剪枝）

题目链接：http://poj.org/problem?id=2531

有n个点，给出任意两点之间的权值，现在将点分别放入A集合或者B集合，求两个集合的权值之和的最大值。

直接暴力过去也没什么问题，110ms，加了一个剪枝到47ms。

初始认为所有集合都在A中，然后枚举将点加入B中。

具体解释见代码：

#include<iostream>
#include<cstdio>
#include<string.h>
#include<algorithm>
#include<cmath>
using namespace std;

const double eps=1e-6;
const int INF=0x3f3f3f3f;
const int maxn=25;
int n,c[maxn][maxn];
int set[maxn];
int ans;

void dfs(int id,int sum){
	set[id]=1;//把id放入B集合
	int tmp=sum;
	for(int i=1;i<=n;i++){
		if(set[i]==0) tmp+=c[i][id];//加上与id不在同一集合的权值
		else tmp-=c[i][id];//减去修改之前和id在不同集合的权值
	}
	if(tmp>ans) ans=tmp;
	for(int i=id+1;i<=n;i++){
		if(tmp>sum){//剪枝，如果修改之后权值变小，则不需要继续向下遍历
			dfs(i,tmp);
			set[i]=0;
		}
	}
}

int main(){
#ifndef ONLINE_JUDGE
    freopen("test.in","r",stdin);
    freopen("test.out","w",stdout);
#endif
    while(~scanf("%d",&n)){
    	for(int i=1;i<=n;i++)
    		for(int j=1;j<=n;j++)
    			scanf("%d",&c[i][j]);
    	memset(set,0,sizeof(set));//初始所有集合放入A集合
    	ans=0;
    	dfs(1,0);//从第一个依次向下枚举
    	printf("%d\n",ans);
    }
	return 0;
}