POJ2531 Network Saboteur dfs

传送门：POJ2531

题意：有几台电脑，每两台电脑之间有固定的流量，要求将这些电脑分成两部分，使这两部分之间的流量总和最大。

解法：dfs搜索其中一堆，如果将第i台电脑加进去能将两部分之间的流量变大的话，就将这一台标记出来加入这一堆，并更新流量最大值，否则就放入另一堆。

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
int map[22][22];
int book[22];
int n,ans=0;
void dfs(int sum,int cnt)
{
	int t=sum;
	book[cnt]=1;//先将此台电脑标记出来（加入这一堆）
	for(int i=0;i<n;i++)//判断总流量是否变大
	{
		if(!book[i])
		t+=map[i][cnt];
		else
		t-=map[i][cnt];
	}
	if(t>ans)
	ans=t;
    if(t<sum)//总和没变大的话就取消标记（同时代表将此台电脑放入了第二堆）
    book[cnt]=0;
	for(int i=cnt+1;i<n;i++) 
	{	 
		dfs(t,i);
		book[i]=0;//不能忘记回溯的时候取消标记	
	}
	
}
int main()
{
	while(~scanf("%d",&n))
	{
		ans=0;
		memset(book,0,sizeof(book));
		memset(map,0,sizeof(map));
		for(int i=0;i<n;i++)
		{
			for(int j=0;j<n;j++)
			scanf("%d",&map[i][j]);
		}
		dfs(0,0);
		/*for(int i=0;i<n;i++)
		printf("%d\n",book[i]);*/
		printf("%d\n",ans);
	}
return 0;
}

上面是我自己写的，我觉得dfs部分更好理解，但是没优化到最好。下面是百度的题解，时间缩短一些但是自己感觉理解有点困难。

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
int map[22][22];
int book[22];
int n,ans=0;
void dfs(int sum,int cnt)
{
	int t=sum;
	book[cnt]=1;
	for(int i=0;i<n;i++)
	{
		if(!book[i])
		t+=map[i][cnt];
		else
		t-=map[i][cnt];
	}
	if(t>ans)
	ans=t;
	if(t>sum)//重点就是此处不好理解，其实可以这样想，既然把这台电脑加入会使总和变小，那么一定不能在将这台电脑
	{<span style="white-space:pre">		</span>//标记出来的情况下继续搜索，这样是毫无意义的，可以返回到上一层将标记解除后再继续搜索。
		for(int i=cnt+1;i<n;i++) 
		{
		 
		dfs(t,i);
		book[i]=0;
		
		}
	}
}
int main()
{
	while(~scanf("%d",&n))
	{
		ans=0;
		memset(book,0,sizeof(book));
		memset(map,0,sizeof(map));
		for(int i=0;i<n;i++)
		{
			for(int j=0;j<n;j++)
			scanf("%d",&map[i][j]);
		}
		dfs(0,0);
		/*for(int i=0;i<n;i++)
		printf("%d\n",book[i]);*/
		printf("%d\n",ans);
	}
return 0;
}