poj 2454 Jersey Politics dfs

这个题目第一步还是比较明显的，先把最小的n个值去掉，剩下的问题就是能不能把数据分成两半，使得每一半和都大于n*500，这个刚开始考虑了下dp的做法，但是复杂度不满足要求。

那么能想到的就是搜索了，实际上这个搜索的上限还是比较高的，就只能依靠良好的剪枝试试了，两个情况，剩下的值全部选最大还小于要求，剩下的值全部选最小另一半还小于要求。0ms ac

 

#include <iostream>

#include <cstdio>

#include <cstring>

#include <algorithm>

using namespace std;

const int maxn=200;

struct D

{

    int key,id;

    bool operator <(const D &xx)const

    {

        return key>xx.key;

    }

}data[maxn];



bool cmp(const D &p,const D &q)

{

    return p.key<q.key;

}



int sum[maxn],n;

bool use[maxn],flag;

void dfs(int t,int ret,int ans)

{

//    printf("%d %d\n",ret,ans);

    if(flag) return ;

    if(ret==n)

    {

        if(ans>n*500&&sum[n+n]-ans>n*500)

        {

            for(int i=1;i<=n+n;i++)

            if(use[i])

            printf("%d\n",data[i].id);

            for(int i=1;i<=n+n;i++)

            if(!use[i])

            printf("%d\n",data[i].id);

            flag=true;

        }

        return ;

    }

    if(ans+sum[t]-sum[t-(n-ret)]<=n*500) return ;

    if(sum[n+n]-(ans+sum[n-ret])<=n*500) return ;

    for(int i=t;i>=n-ret;i--)

    {

        use[i]=1;

        dfs(i-1,ret+1,ans+data[i].key);

        use[i]=0;

    }

}



int main()

{

    scanf("%d",&n);

    for(int i=1;i<=n*3;i++)

    {

        scanf("%d",&data[i].key);

        data[i].id=i;

    }

    sort(data+1,data+1+n*3);

    sort(data+1,data+1+n+n,cmp);

    sum[0]=0;

    for(int i=1;i<=n+n;i++)

    sum[i]=sum[i-1]+data[i].key;

    flag=0;

    memset(use,0,sizeof(use));

    dfs(n+n,0,0);

    for(int i=n+n+1;i<=n*3;i++)

    printf("%d\n",data[i].id);

    return 0;

}