gistfile1.py

"""Solve 'Towers of Hanoi'"""
import pylab as p;
import mpl_toolkits.mplot3d.axes3d as p3;

def solve(g,n):
    X = [sum(g[0])]
    Y = [sum(g[1])]
    Z = [sum(g[2])]

    moved = 0
    for i in range(2**n - 1):
        tops = [a[0] for a in g]
        movable = False
        j = 0
        while not movable:
            max_legal = max([t for t in tops if g[j][0] % 2 != t % 2])
            if g[j][0] != moved and g[j][0] > 0:
                num_zeros = len([z for z in tops if z == 0])
                if g[j][0] < max_legal or num_zeros > 0:
                    movable = True
            if not movable:
                j += 1
        
        moved = g[j][0]
        
        legal = False
        k = 2
        while not legal:
            if (tops[k] % 2 != g[j][0] % 2 and g[j][0] < tops[k]) or tops[k] == 0:
                legal = True
            if not legal:
                k -= 1

        g[k] = [g[j][0]] + g[k]
        g[j] = g[j][1::]

        print g

        S = [sum(s) for s in g]
        X += [S[0]]
        Y += [S[1]]
        Z += [S[2]]
    
    return (X,Y,Z)

n = 10
game = [range(1,n+1)+[0], [0], [0]]

X, Y, Z = solve(game,n)

fig = p.figure()
ax = p3.Axes3D(fig)
ax.plot3D(X,Y,Z)
fig.add_axes(ax)
p.show()