试题 算法训练 娜神平衡

---

前言

这道题有一个数据超时了，暂时没找到问题出在哪。

---

一、题目详情

二、解题代码

#include<iostream>
#include<algorithm>
#include<vector>
#include<cmath>
using namespace std;

const int N = 1e6;
int n,h;
int s[N],j[N];
vector<int> l,r;
int a[N],b[N];
bool st[N],g;

int main()
{
    scanf("%d %d",&n,&h);//h表示能相差最大的数
    
    for(int i=0;i<n;i++) scanf("%d",&s[i]);
    for(int i=0;i<n;i++) j[i]=i;
    
    for(int i=0;i<1<<n;i++) //二进制枚举 1表示该数被选中放入L中
    {
    	sort(j,j+n);//排序为全排列函数做准备 全排列对应的数值对应选取的顺序
    	do
    	{
    	    a[i]=0,b[i]=0,st[i]=false;//a，b分别储存每一个二进制枚举对应的值
            for(int k=0;k<n;k++) //遍历二进制里的每一位一
                if(i&(1<<j[k]) || j[k]==0) //如果这一位是1或者这一位是第一位放入L中
                {
                    a[i] += s[j[k]];
                    if(abs(a[i]-b[i])>h) st[i]=true;//st判断这组数据能否使用 true表示不能
                     l.push_back(s[j[k]]); 

                }
                else 
                {
                    b[i] += s[j[k]];
                    if(abs(a[i]-b[i])>h) st[i]=true;
                     r.push_back(s[j[k]]);
                }
                if(abs(a[i]-b[i])<=h && !st[i]) //当满足条件的时候输出结果
                {
                    sort(l.begin(),l.end()),sort(r.begin(),r.end());
                    for(vector<int>::iterator it=l.begin();it != l.end();it++) cout << *it <<" ";
                    puts("");
                    for(vector<int>::iterator it=r.begin();it != r.end();it++) cout << *it<<" ";
                    puts("");
                    g=true;
                    break;
                }
                
                if(g) break;
                l.clear(),r.clear(); //每一次把动态数组清空
    	}while(next_permutation(j,j+n)); //进入下一个排列
    	if(g) break; // 满足条件不再进入下一个循环 直接退出
    }
    return 0;
}