#!/usr/bin/python # -*- coding: utf-8 -*- import sys def getOriRoom(roomCount, endRoomIndex, afterRooms):     minRoomPersonNum = afterRooms[0]     startRoomIndex = 0     oriRoom=[0]*roomCount     for i, n in enumerate(afterRooms):         if n<minRoomPersonNum:             startRoomIndex=i             minRoomPersonNum=n                  if startRoomIndex < endRoomIndex:         #         for i, n in enumerate(afterRooms):             oriRoom[i]=afterRooms[i] - afterRooms[startRoomIndex]             if i<=endRoomIndex and i>startRoomIndex:               oriRoom[i]=oriRoom[i] - 1         oriRoom[startRoomIndex] = afterRooms[startRoomIndex]*roomCount+1     else:         #         for i, n in enumerate(afterRooms):             oriRoom[i]=afterRooms[i] - afterRooms[startRoomIndex]             if i<=endRoomIndex or i>startRoomIndex:               oriRoom[i]=oriRoom[i] - 1         oriRoom[startRoomIndex] = afterRooms[startRoomIndex]*roomCount+1     return oriRoom if __name__ == '__main__':     roomCount, endRoomIndex, afterRooms = (3, 0, [6,5,1])     ret = getOriRoom(roomCount, endRoomIndex, afterRooms)     print "roomCount=%d, endRoomIndex=%d, ret=%s, afterRooms=%s"%(roomCount, endRoomIndex, ret, afterRooms)     roomCount, endRoomIndex, afterRooms = (3, 2, [6,1,3])     ret = getOriRoom(roomCount, endRoomIndex, afterRooms)     print "roomCount=%d, endRoomIndex=%d, ret=%s, afterRooms=%s"%(roomCount, endRoomIndex, ret, afterRooms)