free-the-bunny-workers/solution.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14

from itertools import combinations


def solution(num_buns, num_required):
    c = list(combinations(range(num_buns), num_buns - (num_required - 1)))
    res = []

    for bunny in range(num_buns):
        res.append([k for k, b in enumerate(c) if bunny in b])

    return res


print(solution(5, 3))