UVALive 7009 Secret Binary Tree（二分查找、dfs）

题目链接：
UVALive 7009 Secret Binary Tree
分析：
先用dfs把所有可能的构造方案都处理出来，然后用二分查找处理出s[i]，排序。
要想使的secret number最小，一定是将比较大的数放在完全二叉树的最下层，将小的数放在完全二叉树的倒数第二层。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <algorithm>
#include <climits>
#include <cmath>
#include <ctime>
#include <cassert>
#define IOS ios_base::sync_with_stdio(0); cin.tie(0);
using namespace std;
typedef long long ll;
const int MAX_N=20;
const int MAX_M=50000;
const double eps=1e-8;

int T,n,total;
int num[MAX_N],vis[MAX_N]; //num[i]是选i个数的所有方案数
double r[MAX_N],s[MAX_M*6],choose[MAX_N][MAX_M][MAX_N]; 
//choose[i][j][k]是一共选择i个数的j个方案的k个数的数值

inline void dfs(int id,int now,int sum)
{
    if(now==sum){
        int j=0;
        for(int i=0;i<sum;i++){
            while(vis[j]==0) { j++ ;}
            choose[sum][num[sum]][i]=r[j];
            j++;
        }
        num[sum]++;
        return ;
    }
    for(int i=id;i<n;i++){
        vis[i]=1;
        dfs(i+1,now+1,sum);
        vis[i]=0;
    }
}

inline bool check(double d,double data[],int len)
{
    double res=0.0;
    for(int i=0;i<len;i++){
        res+=pow(data[i],d);
    }
    if(res < 1.0) return true;
    else return false;
}

inline double solve(double data[],int len)
{
    double low=0.0,high=15,mid;
    for(int i=0;i<100;i++){
        mid=(high+low)/2.0;
        if(check(mid,data,len)){
            high=mid;
        }else {
            low=mid;
        }
    }
    return mid;
}

inline void GetS()
{
    total=0;
    for(int i=2;i<=n;i++){
        //printf("num[%d]=%d\n",i,num[i]);
        for(int j=0;j<num[i];j++){
            s[total++]=solve(choose[i][j],i);
        }
    }
}

int main()
{
    IOS;
    cin>>T;
    while(T--){
        cin>>n;
        for(int i=0;i<n;i++){
            cin>>r[i];
        }
        memset(s,0,sizeof(s));
        memset(num,0,sizeof(num));
        for(int i=2;i<=n;i++){
            memset(vis,0,sizeof(vis));
            dfs(0,0,i);
        }
        GetS();
        sort(s,s+total);

        int tmp=1;
        for(int i=1;i<=n;i++){
            tmp*=2;
        }
        int small=n,big=tmp-n-2;
        double ans;
        //printf("small=%d big=%d\n",small,big);
        if(n==2) ans=s[0];
        else if(n==3) ans=s[3]*pow(0.5,(n-1)*1.0);
        else ans=max(s[small]*pow(0.5,(n-2)*1.0),s[big]*pow(0.5,(n-1)*1.0));

        printf("%.6lf\n",ans);
    }
    return 0;
}