Python蓝桥杯ALGO-998 娜神平衡

每日一题

虽然还是磕磕绊绊写了很久，但是觉得比以前稍微熟练一些了！

题目描述

问题描述
　　有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执行相应的回溯。
• 8. 最后的答案容易是反过来的，需要看输入数组第一个元素在不在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=" ")