HackerEarth - Josh and Hotel Solution

HackerEarth - Josh and Hotel Solution

Josh owns a hotel, there are X rooms in the hotel and every room can be allocated to maximum 1 person at a time.

If a customer arrives and there are no rooms available, the customer will leave the hotel without staying, thus causing loss to his hotel. Input consists of a sequence of uppercase letters ,followed by a positive integer X (0≤X≤26) in the next line, representing the number of rooms.

The first occurrence of a letter indicates the arrival of a customer, the second indicates the departure of that same customer. No letter will occur in more than one pair.

Customers who leave without staying in the hotel always depart before customers who are currently staying.

INPUT:
Input file consists of several test cases (<20), First line indicates a string followed by X (number of rooms). Input will be terminated by 0 .

OUTPUT:
For each test case print how many customers will leave the hotel without staying

Constraints:
2≤|stringlength|<=52

0≤X≤26

SAMPLE INPUT

ABCBCA
1
0

SAMPLE OUTPUT

2

Solution in Python

while True:
    s = input().strip()
    if s =="0": break
    n = int(input())
    room = set()
    wait = set()
    fill = 0
    left = 0
    for person in s:
        if person in wait:
            left += 1
        elif person in room:
            room.discard(person)
            fill -= 1
        elif fill<n:
            room.add(person)
            fill += 1
        else:
            wait.add(person)
    print(left)

Additional info

fill - Room currently occupied

left - Visitors left without staying

room - Person currently staying in room

wait - Person currently waiting to leave