3 pegs

pegs.py

import sys


def solve(pegs, pfrom, pto, n):
    if n == 1:
        pegs[pfrom] -= 1
        pegs[pto] += 1
        print('{:d}->{:d}'.format(pfrom + 1, pto + 1))
    elif n > 1:
        solve(pegs, pfrom, third_peg(pfrom, pto), n - 1)
        solve(pegs, pfrom, pto, 1)
        solve(pegs, third_peg(pfrom, pto), pto, n - 1)


def third_peg(peg_one, peg_two):
    return 3 - peg_one - peg_two


if __name__ == '__main__':
    print("3 pegs - " + sys.argv[1] + " cones")
    arg_n = int(sys.argv[1])
    solve([arg_n, 0, 0], 0, 2, arg_n)