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 vote157 views1 answer
Mypy Should never get here in normal mode, got NoneType: instead of TypeInfo
Python
1 vote207 views1 answer
Model.predict([x_test]) error
Python
3 votes160 views3 answers
ModuleNotFoundError: No module named 'MySQLdb' in windows
Python
0 vote144 views1 answer
Assign exec function to variable python
Python
12 votes178 views5 answers
Python get the intersection of two lists
Python
1 vote147 views1 answer
Python how to change back to the later directory
Python
1 vote138 views1 answer
Ffmpeg python video from images
Python
22 votes160 views4 answers
Stop program python
Python
0 vote165 views1 answer
Pandas recognize type from strings
Python
2 votes161 views3 answers
How to set indian timezone in django
Python