资源限制
内存限制:256.0MB C/C++时间限制:1.0s Java时间限制:3.0s Python时间限制:5.0s
问题描述
娜娜是一个特别可爱的女孩子,作为学神的她最近在情感方面出现了一点点小问题。
她暗恋的琦琦是一名学霸,他只喜欢长得漂亮和学习很好的女生。
娜娜学习确实很神,但是她在琦琦面前却总是表现不出平时的神力。
琦琦感受到了娜娜对他的爱,但是他还是觉得娜娜的学习并不是特别好,于是他出了一道题给娜娜。
“娜娜,我们之间的关系需要在不断深入的同时保持一定的平衡,不可以你总是强势或者我总是弱势。”
琦琦给了娜娜一些两两不等的数,希望娜娜能把这些数分成两组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组。
数据规模和约定
很小,真的很小。
n, r = map(int, input().split())
nums = list(map(int, input().split()))
# num_1存第一个数字,方便最后判断哪个是a,哪个是b
# a,b分别是存两组数的数组,queue是存所有数字的队列
# flag用来存每一个数字存入哪个组,1表示a数组,2表示b数组
num_1 = nums[0]
a = []
b = []
nums = sorted(nums)
queue = nums[:]
flag = []
# 循环部分是对每一个数进行处理,先放到a数组里,如果不满足条件就取出放到b数组,如果都不行就取出,放到队列尾,将上一个已经成功的数也取出
# (这里的将上一个成功的数也取出来就是回溯的过程,如果不能满足条件的话就会一直向前回溯知道满足条件)
while len(queue) > 0:
last = queue.pop(0)
a.append(last)
flag.append(1)
sum_a = sum(a)
sum_b = sum(b)
# 放到a中不满足条件,取出放b数组里
if abs(sum_a - sum_b) > r:
sum_a -= last
a.pop()
flag.pop()
sum_b += last
b.append(last)
flag.append(2)
# b数组也不行,取出放到队尾,将上一个数也取出放队尾
if abs(sum_a - sum_b) > r:
sum_b -= last
b.pop()
flag.pop()
queue.append(last)
if flag[-1] == 1:
queue.append(a.pop())
else:
queue.append(b.pop())
flag.pop()
# 这一步是为了解决一直回溯导致第一个数位置的不确定性
if num_1 in b:
a, b = b, a
# 将a,b分别转换成字符数组再进行拼接,然后一次性输出
print(' '.join(map(str, a)))
print(' '.join(map(str, b)))