Number of People in the Bus Codewars Python

There is a bus moving in the city, and it takes and drop some people in each bus stop.

You are provided with a list (or array) of integer pairs. Elements of each pair represent number of people get into bus (The first item) and number of people get off the bus (The second item) in a bus stop.

Your task is to return number of people who are still in the bus after the last bus station (after the last array). Even though it is the last bus stop, the bus is not empty and some people are still in the bus, and they are probably sleeping there :D

Take a look on the test cases.

Please keep in mind that the test cases ensure that the number of people in the bus is always >= 0. So the return integer can't be negative.

The second value in the first integer array is 0, since the bus is empty in the first bus stop.

def number(bus_stops):
    people_in = 0
    people_out = 0
    for i in bus_stops:
        people_in += i[0]
        print(people_in)
        people_out += i[1]
        print(people_out)
    return people_in - people_out

def number(bus_stops):
    return sum([stop[0] - stop[1] for stop in bus_stops])



# Sum the first and last in the list and subtract

print(number([[10,0],[3,5],[5,8]]))

Training on Number of People in the Bus | Codewars

Related Posts