Free the Bunny Prisoners

from itertools import combinations


def solution(num_buns, num_required):
    keyrings = [[] for num in range(num_buns)]
    copies_per_key = num_buns - num_required + 1
    for key, bunnies in enumerate(combinations(range(num_buns), copies_per_key)):
        for bunny in bunnies:
            keyrings[bunny].append(key)

    return keyrings


if __name__ == "__main__":
    assert all([all([x == y for x, y in zip(a, b)])
                for a, b in zip([[0, 1, 2, 3, 4, 5], [0, 1, 2, 6, 7, 8], [0, 3, 4, 6, 7, 9], [1, 3, 5, 6, 8, 9], [2, 4, 5, 7, 8, 9]], solution(5, 3))])
    assert all([all([x == y for x, y in zip(a, b)])
                for a, b in zip([[0], [1], [2], [3]], solution(4, 4))])

Similar codes

1 vote221 views1 answer
Keep the user logged in even though user changes password django
Python
3 votes205 views1 answer
Visualize correlation matrix python
Python
0 vote245 views1 answer
Inspect last 5 rows of dataframe
Python
0 vote227 views1 answer
Python tokenize sentence italian spacy
Python
5 votes238 views1 answer
Python string match ignore case
Python
5 votes216 views1 answer
Check is lowercase
Python
3 votes256 views3 answers
Kadane's algorithm
Python
0 vote238 views1 answer
S=0 def sum(x,y): n=int(input("enter no. of terms") for i in range(n): l=int(input("enter no.")) s=s+l print(s) sum()
Python
0 vote236 views1 answer
Run for loop inside pdb
Python
0 vote217 views1 answer
Hostname python
Python