Metro Card

You just bought a public transit card that allows you to ride the Metro for a certain number of days.

Here is how it works: upon first receiving the card, the system allocates you a 31-day pass, which equals the number of days in January. The second time you pay for the card, your pass is extended by 28 days, i.e. the number of days in February (note that leap years are not considered), and so on. The 13th time you extend the pass, you get 31 days again.

You just ran out of days on the card, and unfortunately you've forgotten how many times your pass has been extended so far. However, you do remember the number of days you were able to ride the Metro during this most recent month. Figure out the number of days by which your pass will now be extended, and return all the options as an array sorted in increasing order.

Example

For lastNumberOfDays = 30, the output should be

metro_card(lastNumberOfDays) = [31]

Input/Output

• [input] integer lastNumberOfDays

  A positive integer, the number of days for which the card was extended the last time.

  Guaranteed constraints:lastNumberOfDays = 28 or lastNumberOfDays = 30 or lastNumberOfDays = 31.

There are 30 days in April, June, September and November, so the next months to consider are May, July, October or December. All of them have exactly 31 days, which means that you will definitely get a 31-days pass the next time you extend your card.

Solution

PYTHON

def metro_card(last_number_of_days):
    if last_number_of_days < 31:
        return [31]

    return [28, 30, 31]


print(metro_card(30))
print(metro_card(31))
print(metro_card(28))