Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- inputLine = input()
- trains = dict()
- while inputLine != "It's Training Men!":
- if ':' in inputLine:
- inputData = inputLine.split(' : ')
- wagonPower = int(inputData[1])
- inputData = inputData[0].split(' -> ')
- trainName, wagonName = inputData
- if trainName in trains:
- trains[trainName][wagonName] = wagonPower
- else:
- trains[trainName] = {wagonName: wagonPower}
- elif '=' in inputLine:
- inputData = inputLine.split(' = ')
- trainName, otherTrainName = inputData
- trains[trainName] = {}
- for wagon in trains[otherTrainName]:
- trains[trainName][wagon] = trains[otherTrainName][wagon]
- else:
- inputData = inputLine.split(' -> ')
- trainName, otherTrainName = inputData
- if not trainName in trains:
- trains[trainName] = {}
- for wagon in trains[otherTrainName]:
- trains[trainName][wagon] = trains[otherTrainName][wagon]
- del trains[otherTrainName]
- inputLine = input()
- totalWagonPower = dict()
- for train in trains:
- temp = trains[train].values()
- totalWagonPower[train] = (-sum(temp), len(temp))
- for train in sorted(totalWagonPower.keys(), key = lambda x: totalWagonPower[x]):
- print('Train: {0}'.format(train))
- for wagon in sorted(trains[train].items(), key = lambda x: (x[1]), reverse = True):
- print('###{0} - {1}'.format(wagon[0], wagon[1]))
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement