编程之美--游戏之乐--1.8小飞的电梯调度问题

#coding=utf-8
#author='HL'
class Solution():
def solve(self,n_Person):
'''
:param n_Person:type--list,每层电梯停留人数
:return:
'''
Y = 0
for i in range(3,len(n_Person)):
Y+=(i-2)*n_Person[i]
n_Person[i]+=n_Person[i-1]
print '到第i层为止，目前的人数为：'
print n_Person
print '第2层停止，需要走的步数：'
print Y
res = Y
number = 2
'''
N1 i层以下所有人数
N2 第i层的人数
N3 i层以上的人数
'''
for i in range(3,len(n_Person)):
N1 = n_Person[i-1]
N2 = n_Person[i]-n_Person[i-1]
N3 = n_Person[-1]-n_Person[i]
Y = Y+N1-N2-N3
if Y<res:
res = Y
number = i
else:
break
return res,number

zz = Solution()
print zz.solve([0,0,1,2,3,4,5])