一根长度为1米的木棒上有若干只蚂蚁在爬动。它们的速度为每秒一厘米或静止不动,方向只有两种,向左或者向右。如果两只蚂蚁碰头,则它们立即交换速度并继续爬动。三只蚂蚁碰头,则两边的蚂蚁交换速度,中间的蚂蚁仍然静止。如果它们爬到了木棒的边缘(0或100厘米处)则会从木棒上坠落下去。在某一时刻蚂蚁的位置各不相同且均在整数厘米处(即1,2,3,…99厘米),有且只有一只蚂蚁A速度为0,其他蚂蚁均在向左或向右爬动。给出该时刻木棒上的所有蚂蚁位置和初始速度,找出蚂蚁A从此时刻到坠落所需要的时间。
输入描述:
第一行包含一个整数表示蚂蚁的个数N(2<=N<=99),之后共有N行,每一行描述一只蚂蚁的初始状态。每个初始状态由两个整数组成,中间用空格隔开,第一个数字表示初始位置厘米数P(1<=P<=99),第二个数字表示初始方向,-1表示向左,1表示向右,0表示静止。
输出描述:
蚂蚁A从开始到坠落的时间。若不会坠落,输出“Cannot fall!”
示例1
输入
4
10 1
90 0
95 -1
98 -1
输出
98
python3.5.2
class Ant:
def __init__(self,pos,speed):
self.pos=pos
self.speed=speed
#这道题要理解蚂蚁的相对位置不变,第i个蚂蚁最后还是在第i个,只是位置变了
n=int(input())
ants=[]
for i in range(n):
ant=input().split(' ')
pos=int(ant[0])
speed=int(ant[1])
ants.append(Ant(pos,speed))
ants.sort(key=lambda x:x.pos)
#先按照蚂蚁当前位置给蚂蚁排个序,下面找出静止的A蚂蚁
A=0
for i in range(n):
if ants[i].speed==0:
A=i
ants.sort(key=lambda x:x.pos+x.speed*100)
#按照pos+100*speed给蚂蚁排序,但存储的依旧是初始值pos和speed
#但是相对位置依然是不变的,初始在第3个现在还在第3个
if ants[A].speed==-1:
print(ants[A].pos)
#如果A最后向左,A掉下去的时间就是0~x.pos的长度
elif ants[A].speed==1:
print(100-ants[A].pos)
#如果A最后向右,A掉下去的时间就是x.pos~100的长度
else:
print("Cannot fall!")
#如果最后A静止,那么A永远掉不下去