问题描述
娜娜是一个特别可爱的女孩子,作为学神的她最近在情感方面出现了一点点小问题。
她暗恋的琦琦是一名学霸,他只喜欢长得漂亮和学习很好的女生。
娜娜学习确实很神,但是她在琦琦面前却总是表现不出平时的神力。
琦琦感受到了娜娜对他的爱,但是他还是觉得娜娜的学习并不是特别好,于是他出了一道题给娜娜。
“娜娜,我们之间的关系需要在不断深入的同时保持一定的平衡,不可以你总是强势或者我总是弱势。”
琦琦给了娜娜一些两两不等的数,希望娜娜能把这些数分成两组A和B,满足以下条件:
1:每一次只能操作一个数,即只取出一个数分入A中或B中;
2:每一次操作完成后,A中数之和与B中数之和的差不能超过r。
新时代的丘比特们啊,帮帮娜娜吧!
输入格式
输入共两行。
第一行包括两个正整数n和r,n表示琦琦一共给了n个数,r的意义见题目描述。
第二行包括n个正整数,分别表示琦琦给的n个数。
输出格式
输出共两行,分别把A与B两组数按从小到大输出。
注意输入中n个数的第一个必须分入A组。
琦琦保证这样的输出唯一。
样例输入
4 10
9 6 4 20
样例输出
4 6 9
20
样例说明
先把4和6先后分入A组,再把20分入B组,最后把9分入A组。
数据规模和约定
很小,真的很小。
解题思路:
利用dfs深度搜索,把数组中的每个数依次先往A放,如果不符合条件再回溯往B放,但是注意第一个数必须放在A,但是没说第一次就必须把第一个数就放在A中,所以如果刚开始第一个数放在A中不符合条件的时候,那就从后面开始看,都不行,就放B,但是B数组是从第二个元素开始的。
代码:
#include<bits/stdc++.h>
using namespace std;
int a[1010];
int b[1010];
int x[1010];
int lena,lenb;
int visit[1010];
int n,r,f;
int sum(int *c,int len)
{
int ans=0;
for(int i=0;i<len;i++)
{
ans+=c[i];
}
return ans;
}
void dfs(int k)
{
if(abs(sum(a,lena)-sum(b,lenb))>r) //剪枝,此时两数组的元素不符合条件
{
return ;
}
if(k>=n) //元素分配完毕
{
sort(a,a+lena);
sort(b,b+lenb);
for(int i=0;i<lena;i++)
{
cout<<a[i]<<" ";
}
cout<<endl;
for(int i=0;i<lenb;i++)
{
cout<<b[i]<<" ";
}
cout<<endl;
f=1; //标记已找到结果,且只有一个结果,所以后续无须再找
return ;
}
for(int i=0;i<n;i++) //先把每个元素放A数组,不符合条件的话再放B数组
{
if(f==1)
return ;
if(lena==0&&abs(x[0]-sum(b,lenb))>r) //第一个数虽然必须放在A中,但是倘若第一个数放入后不符合条件,就先不放第一个数,放后面的数试试,都不行就往给B放
{
continue;
}
if(visit[i]==0)
{
visit[i]=1;
a[lena++]=x[i];
dfs(k+1);
visit[i]=0;
a[--lena]=0;
}
}
for(int i=1;i<n;i++)
{
if(f==1)
return ;
if(visit[i]==0)
{
visit[i]=1;
b[lenb++]=x[i];
dfs(k+1);
visit[i]=0;
b[--lenb]=0;
}
}
}
int main()
{
cin>>n>>r;
for(int i=0;i<n;i++)
cin>>x[i];
dfs(0);
return 0;
}