28 坠落的蚂蚁 北大复试

一根长度为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永远掉不下去