可以说,这道题是一道无需优化的DFS,由于N最大只有20,连O(2^n)的复杂度都能过,因此,只需要DFS枚举每一个节点分配到哪一个子集即可。
因此就能得出代码:
#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cmath>
#define max(x,y) (x>y?x:y)
#define min(x,y) (x<y?x:y)
#define LL long long
#define N 20
using namespace std;
int n,ans=0,len1=0,len2=0,s1[N+5],s2[N+5],a[N+5][N+5];
int read()
{
int x=0,f=1;char ch=getchar();
while((ch<'0'||ch>'9')&&ch!='-') ch=getchar();
if(ch=='-') f=-1,ch=getchar();
while(ch>='0'&&ch<='9') (x*=10)+=ch-'0',ch=getchar();
return x*=f;
}
void write(int x)
{
if(x<0) putchar('-'),x=-x;
if(x>9) write(x/10);
putchar(x%10+'0');
}
void search(int x)
{
if(x>n) {int t=0;for(int i=1;i<=len1;i++) for(int j=1;j<=len2;j++) t+=a[s1[i]][s2[j]];if(t>ans) ans=t;return;}//每个节点都分配完了,计算所需网络流量,并与ans比较
s1[++len1]=x,search(x+1),len1--,s2[++len2]=x,search(x+1),len2--;//枚举将该节点放入哪一个子集
}
int main()
{
n=read();
for(int i=1;i<=n;i++) for(int j=1;j<=n;j++) a[i][j]=read();
search(1);
write(ans);
return 0;
}