BZOJ 3699 GAL的数组

题目大意：给出3*N的非负整数，分配到3个长度为N的数组A,B,C中。定义三个数组的GAL值，GAL(A,B,C)=Σ(A[i]-B[i])*C[i],1 ≤ i ≤ N。求出GAL(A,B,C)的最大值。

#include <cstdio>
#include <algorithm>
#include <functional>
using namespace std;
typedef long long LL;
int T,n,N,topa,topc,s[100],a[30],b[30],c[30];
LL ans,now;
bool used[100];
bool check(int x) {
    for(int i=1;i<x;i++)
        if((c[i]>a[x] && a[i]-b[i]<c[x]) || (c[i]<a[x] && a[i]-b[i]>c[x]))
            return false;
    return true;
}
void dfs(int x,int pos) {
    if(x==n+1) {
        ans=max(ans,now);
        return ;
    }
    for(int i=x;i<=2*n;++i) {
        if(used[i]) continue;
        used[i]=true;
        a[x]=s[i];
        for(int j=max(i,pos)+1;j<=2*n;++j) {
            if(used[j]) continue;
            used[j]=true;
            c[x]=s[j];
            now+=(a[x]-s[3*n-x+1])*c[x];
            if(now*n>ans*x && check(x))dfs(x+1,j);
            now-=(a[x]-s[3*n-x+1])*c[x];
            used[j]=false;
        }
        used[i]=false;
        break;
    }
    return ;
}
int main() {
    for(scanf("%d%d",&T,&n),N=3*n;T;T--) {
        ans=0;
        for(int i=1;i<=N;++i) scanf("%d",s+i);
        sort(s+1,s+N+1,greater<int>());
        for(int i=1;i<=n;++i) b[i]=s[N-i+1];
        dfs(1,0);
        printf("%lld\n",ans);
    }
    return 0;
}