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

0 vote355 views1 answer
ImportError: cannot import name 'FieldDoesNotExist' from 'django.db.models.fields' (C:\Users\Admin\AppData\Local\Programs\Python\Python39\lib\site-packages\django\db\models\fields\__init__.py)
Python Django
1 vote300 views1 answer
Installing wxpython on windows 10
Python
2 votes313 views2 answers
Django search
Python Django
0 vote282 views1 answer
Discord.py read custom status
Python
12 votes280 views2 answers
Python tkinter window fullscreen
Python
0 vote272 views2 answers
Django create view filter options
Python
0 vote276 views1 answer
How to call object of a list python
Python
4 votes273 views2 answers
Number length python
Python
0 vote300 views1 answer
Qtablewidget clear python
Python
0 vote283 views1 answer
Counting valleys hackerrank solution python
Python