华为OD机试 - 流浪地球
题目描述
流浪地球计划在赤道上均匀部署了N个转向发动机,按位置顺序编号为
0~N。
- 初始状态下所有的发动机都是未启动状态;
- 发动机启动的方式分为”手动启动"和”关联启动"两种方式;
- 如果在时刻1一个发动机被启动,下一个时刻2与之相邻的两个发动机就会被”关联启动”;
- 如果准备启动某个发动机时,它已经被启动了,则什么都不用做;
5)发动机0与发动机N-1是相邻的;
地球联合政府准备挑选某些发动机在某些时刻进行“手动启动”。当然最终所有的发动机都会被启动。
哪些发动机最晚被启动呢?
输入描述
第一行两个数字N和E,中间有空格
N代表部署发动机的总个数,E代表计划手动启动的发动机总个数
1<N<=1000,1<=E<=1000,E<=N
接下来共E行,每行都是两个数字T和P,中间有空格
T代表发动机的手动启动时刻,P代表此发动机的位置编号。
0<=T<=N.0<=P<N
输出描述
第一行一个数字N,以回车结束
N代表最后被启动的发动机个数
第二行N个数字,中间有空格,以回车结束
每个数字代表发动机的位置编号,从小到大排序
示例1
输入
8 2
0 2
0 6
输出
2
0 4
说明
8个发动机;
时刻0启动(2,6);
时刻1启动(1,3.5,7)(其中1,3被2关联启动,5,7被6关联启动);
时刻2启动(0,4)(其中0被1,7关联启动,4被3,5关联启动);
至此所有发动机都被启动,最后被启动的有2个,分别是0和4。
import sys
import copy
def link(maxlen, n):
# 环形数组的定位
if n == 0:
return [maxlen-1, 1]
if n == maxlen-1:
return [n-1, 0]
return [n-1, n+1]
for line in sys.stdin:
# 获取输入的内容从而确定发动机数量,启动时间与被启动引擎
enginset=line.strip().split()
engin_num=int(enginset[0])
engin_handstart=int(enginset[1])
engin_hslist={}
engin_state={}
for i in range(engin_handstart):
a,b=map(int,sys.stdin.readline().strip().split())
if a not in engin_hslist:
engin_hslist[a]=[]
engin_hslist[a].append(b)
for j in range(engin_num):
engin_state[j]=0
# print (engin_state)
# print (engin_hslist)
time=0
maxlen=len(engin_state)
start_list=[]
last_state={}
while any(value==0 for value in engin_state.values()):
# 当存在0的情况下继续循环直到全被启动
if time not in engin_hslist:
#如果手动启动是从1开始的,那么可以直接跳过0时间点,不然会出错
time+=1
continue
start_list=engin_hslist[time]
for engin in start_list:
engin_state[engin]=1
a,b=link(maxlen,engin)[0],link(maxlen,engin)[1]
if time+1 in engin_hslist:
# 将相邻引擎加入下一个时间点的启动队列
if a not in engin_hslist[time+1]:
engin_hslist[time+1].append(a)
if b not in engin_hslist[time+1]:
engin_hslist[time+1].append(b)
else:
engin_hslist[time+1]=[]
engin_hslist[time+1].append(a)
engin_hslist[time+1].append(b)
time+=1
# print(engin_state)
# 记录当前引擎状态,方便后续追溯最后被启动的引擎
last_state[time]=copy.deepcopy(engin_state)
#要用深拷贝,不然last_state都会指向最新的engin_state
# print(last_state)
unstart=0
unstart_list=copy.deepcopy(last_state[time-1])
output=[]
for key,value in unstart_list.items():
if value==0:
unstart+=1
output.append(key)
print(unstart)
print(*output,sep=" ")
总结:
暴力破解了,直觉引导我这么做
重点:
环形数组
深浅拷贝
状态跟踪
优化方向:
后面出现的启动引擎会和前面已经启动过的重复,不影响结果,可以优化以提升速度
4611

被折叠的 条评论
为什么被折叠?



