2019牛客暑期多校训练营（第二场） F Partition problem（dfs+剪枝）

链接：https://ac.nowcoder.com/acm/contest/882/F

题意：给你n，有2n个人，每两个人(假设为i,j)之间都有一个竞争值v()，把这2n个人分成两个团体，使得两团体之间的总竞争值最大。

思路：直接爆搜，搜的同时计算当前总的竞争值，这里有一个重要的剪枝，详情看注释。

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 50;
ll v[N][N],ans,maxv;
int a[N],b[N],n,nn;
void dfs(int cur,int cnta,int cntb,ll res)
{
	//cur:当前分配的人 
	//cnta:当前a团体中的人数
	//cntb:当前b团体中的人数
	//res:当前总竞争值 
	//cout<<cur<<" "<<cnta<<" "<<cntb<<endl;
	
	//a团体还需要的人与b团体所有人的竞争值+b团体还需要的人与a团体中已有的人的竞争值+当前总竞争值 
	if((ll)(n-cnta)*n*maxv+(ll)(n-cntb)*cnta*maxv+res<ans) return ;
	//dfs终点 
	if(cur==nn+1)
	{
		if(res>ans)	ans=res;
		return ;
	} 
	ll temp;
	//a团体没满，则尝试把cur放入a团体 
	if(cnta<n)
	{
		temp=0;
		a[cnta+1]=cur;
		//计算cur当前的贡献 
		for(int i=1;i<=cntb;i++)
			temp+=v[cur][b[i]];	
		dfs(cur+1,cnta+1,cntb,res+temp);
	}
	//同理，把cur放入b团体 
	if(cntb<n)
	{
		temp=0;
		b[cntb+1]=cur;
		for(int i=1;i<=cnta;i++)
			temp+=v[cur][a[i]];
		dfs(cur+1,cnta,cntb+1,res+temp);
	}
	return ;
}
int main(void)
{
	ans=maxv=0;	
	scanf("%d",&n);
	nn=2*n;
	for(int i=1;i<=nn;i++)
		for(int j=1;j<=nn;j++)
		{
			scanf("%lld",&v[i][j]);
			//记下两人之间最大的竞争值 
			if(v[i][j]>maxv) maxv=v[i][j];
		} 
			
	dfs(1,0,0,0);
	printf("%lld\n",ans);
	return 0;
}