Vasya - Clerk

Description

The new "Avengers" movie has just been released! There are a lot of people at the cinema box office standing in a huge line. Each of them has a single 100, 50 or 25 dollars bill. A "Avengers" ticket costs 25 dollars.

Vasya is currently working as a clerk. He wants to sell a ticket to every single person in this line.

Can Vasya sell a ticket to each person and give the change if he initially has no money and sells the tickets strictly in the order people follow in the line?

Return YES, if Vasya can sell a ticket to each person and give the change. Otherwise return NO.

Solution

from operator import add
def tickets(people):
    deposit = [0,0,0]
    for p in people:
        if p == 25:
            deposit = map(add, deposit, [1,0,0])
        elif p == 50:
            deposit = map(add, deposit, [-1,1,0])
        else:
            if deposit[0] >=1 and deposit[1] >=1:
               deposit = map(add, deposit, [-1,-1,0])
            elif deposit[1] == 0 and deposit[0] >=3:
               deposit = map(add, deposit, [-3, 0, 0])
            else:
               deposit = map(add, deposit, [-3, 0, 0])
        if deposit[0] < 0 or deposit[1] < 0 or deposit[2] < 0: return 'NO'
    return 'YES'