13 lines
200 B
Python

from solution import solution
test_cases = [
[[2], 2],
[[3], 5],
[[4], 14],
]
for i, o in test_cases:
print('input', *i, 'output', o)
print('solution', solution(*i))
print()