蓝桥杯python：扫地机器人

题目：小明公司的办公区有一条长长的走廊，由 N 个方格区域组成，走廊内部署了 K 台扫地机器人，其中第 i 台在第 Ai 个方格区域中。 已知扫地机器人每分钟可以移动到左右相邻的方格中，并将该区域清扫干净。
请你编写一个程序，计算每台机器人的清扫路线，使得：

它们最终都返回出发方格，
每个方格区域都至少被清扫一遍，
从机器人开始行动到最后一台机器人归位花费的时间最少。
注意多台机器人可以同时清扫同一方块区域，它们不会互相影响。
输出最少花费的时间。 在上图所示的例子中，最少花费时间是 6。第一台路线：2-1-2-3-4-3-2，清 扫了 1、2、3、4 号区域。第二台路线 5-6-7-6-5，清扫了 5、6、7。第三台路线 10-9-8-9-10，清扫了 8、9 和 10。

输入格式

第一行包含两个整数 N 和 K。 接下来 K 行，每行一个整数 Ai。

输出格式

输出一个整数表示答案。

全部代码：

def judge(m,q):
   for i in range(k):
       if (a[i]-q) <= m:
           if a[i]<= m:
               m = a[i] + q -1
           else:
               m = a[i] + q -(a[i] - m)
       else:
           return 0
   if m < n:
       return 0
   return 1
   
n,k = map(int,input().split())
a=[0]*k
for i in range(k):
   a[i]=int(input())
a.sort()
if (n/k) >=a[0]:
   q = int(n/k)
else:
   q = a[0]
for i in range(q,n+1):
   m = 0
   if judge(m,i)==1:
       print(2 * (i - 1))
       break