POJ2531

将所有节点分为两个子集,A,B。

求A的各个元素与B的各个元素之间的和。

深搜就行,看POJ分类说要剪枝,想了半天不知道怎么去剪,随便写了下,竟然被我一次水过了。

实训后第一次写代码果然效率高啊。并且代码风格果断改善了很多。。

呵呵。。

#include<iostream>
const int MAX_NODES = 20;
int node_num = 0;
int max_traffic, choose_num, un_choose_num;
int choose[MAX_NODES], un_choose[MAX_NODES];
int net[MAX_NODES][MAX_NODES];
void dfs(int cur) {
	int temp_max;
	if (cur == node_num) {
		temp_max = 0;
		for (int i = 0;i < choose_num;i++) {
			for (int j = 0;j < un_choose_num;j++) {
				temp_max += net[choose[i]][un_choose[j]];
			}
		}
		if (temp_max > max_traffic) {
			max_traffic = temp_max;
			return ;
		}
	} else {
		choose[choose_num++] = cur;
		dfs(cur+1);
		choose_num--;
		un_choose[un_choose_num++] = cur;
		dfs(cur+1);
		un_choose_num--;
	}
}
int main()
{
	std::cin >> node_num;
	for (int i = 0;i < node_num;i++) {
		for (int j = 0;j < node_num;j++) {
			std::cin >> net[i][j];
		}
	}
	max_traffic = 0;
	dfs(0);
	std::cout << max_traffic << std::endl;
	return 0;
}


你可能感兴趣的:(POJ2531)