def check(expected, k, start=0): actual = solve(k=k, start=start) expected[0].sort() actual[0].sort() expected[1].sort() actual[1].sort() if expected != actual and expected[0] != actual[1]: print('no\nexpected: ' + str(expected) + '\nactual: ' + str(actual)) print('') else: print('yay! k=' + str(k) + ', ' + str(actual))
from powerpuzzle import solve def check(expected, k, start=0): actual = solve(k=k, start=start) expected[0].sort() actual[0].sort() expected[1].sort() actual[1].sort() if expected != actual and expected[0] != actual[1]: print('no\nexpected: ' + str(expected) + '\nactual: ' + str(actual)) print('') else: print('yay! k=' + str(k) + ', ' + str(actual)) check([[1,2],[0,3]], k=1, start=0) check([[1, 2, 4, 7],[0, 3, 5, 6]], k=2, start=4) check([[1, 4, 2, 11, 7, 8, 13, 14], [12, 10, 6, 15, 0, 9, 3, 5]], k=3, start=16) print(solve(k=4, start=16))