每日一题
虽然还是磕磕绊绊写了很久,但是觉得比以前稍微熟练一些了!
题目描述
问题描述
有N根木棍,需要将其粘贴成M个长木棍,使得最长的和最短的的差距最小。
输入格式
第一行两个整数N,M。
一行N个整数,表示木棍的长度。
输出格式
一行一个整数,表示最小的差距
样例输入
3 2
10 20 40
样例输出
10
数据规模和约定
N, M<=7
思路分析
这个题目的测试例子并不多。但是想一次写满分也不容易。。。。
这个题目的解题思路,总结:::有点回溯的思想
- 1.对输入数组进行排序
- 2.将数组存放到队列中
- 3.定义A和B两个数组,以队列不为空为循环条件,当队列为空说明所有元素已经分配到A,B两个数组中了
- 4.先从A开始向前试探一步,满足条件就队头出队,放入A中。
- 5.若不满足A的条件,撤回A,B向前试探一步
- 6.若A,B都不满足,这个数重新入队,放入队尾,等待下次遍历。
- 7.由于可能当前A,B分配到数字永远不满足条件,即进入死循环。因此需要回溯!!!!!!!!
定义一个flag,判断上一步是放入A,还是放入B。 根据flag对A或B执行相应的回溯。 -
- 最后的答案容易是反过来的,需要看输入数组第一个元素在不在A中,如果在B中,A,B交换。
代码如下
# import collections
n,r=map(int,input().split())
n_list=list(map(int,input().split()))
first=n_list[0]
A=[]
B=[]
n_list=sorted(n_list)
queue=n_list[:]
back_flag=[]
i=0
while len(queue)>0:
cur=queue.pop(0)
A.append(cur)
back_flag.append(1)
tmp_A=sum(A)
tmp_B=sum(B)
if abs(tmp_A-tmp_B)>r:
tmp_A-=cur
A.pop()
back_flag.pop()
tmp_B+=cur
B.append(cur)
back_flag.append(2)
if abs(tmp_A-tmp_B)>r:
tmp_B-=cur
B.pop()
back_flag.pop()
queue.append(cur)
if back_flag[-1]==1:
queue.append(A.pop())
back_flag.pop()
else:
queue.append(B.pop())
back_flag.pop()
if first in B:
A,B=B,A
for i in A:
print(i,end=" ")
print()
for i in B:
print(i,end=" ")