Hackerrank - Bus Station Solution
There are groups of friends, and each group is numbered from 1 to . The th group contains people.
They live near a bus stop, and only a single bus operates on this route. An empty bus arrives at the bus stop and all the groups want to travel by the bus.
However, group of friends do not want to get separated. So they enter the bus only if the bus can carry the entire group.
Moreover, the groups do not want to change their relative positioning while travelling. In other words, group 3 cannot travel by bus, unless group 1 and group 2 have either (a) already traveled by the bus in the previous trip or (b) they are also sitting inside the bus at present.
You are given that a bus of size can carry people simultaneously.
Find the size of the bus so that (1) the bus can transport all the groups and (2) every time when the bus starts from the bus station, there is no empty space in the bus (i.e. the total number of people present inside the bus is equal to )?
The first line contains an integer . The second line contains space-separated integers .
Print all possible sizes of the bus in an increasing order.
8 1 2 1 1 1 2 1 3
3 4 6 12
In the above example, = 1, = 2, = 1, = 1, = 1, = 2, = 1, = 3.
If x = 1 : In the first trip, go by the bus. There will be no second trip because the bus cannot accommodate group 2. Hence "x = 1" is not the required answer.
If x = 2 : No bus trip is possible. That's because cannot go alone, as one seat will be left vacant in the bus. And, & cannot go together, because the bus is cannot accommodate both the groups simultaneously.
If x = 3 : In the first trip, & go by the bus. In the second trip, , & go by the bus. In the third trip, & go by the bus. In the fourth trip, go by the bus.
If x = 4 : In the first trip, , & go by the bus. In the second trip, , & go by the bus. In the third trip, & go by the bus.
Similarly you can figure out the output for x= 5, 6 & 7.
Solution in Python
def solve(a): s = sum(a) sums =  for i in range(len(a)): t = (sums or )[-1] + a[i] if t<=s/2: sums.append(t) else: break sums.append(s) for i in sums: if not s%i: t = i for j in a: t = t or i t-= j if t<0: break else: yield i int(input()) a = list(map(int,input().split())) print(*solve(a))