二叉树先序遍历动态演示——python版

前几天，学校算法实验周，有一个选做题是写一个二叉树先序遍历动态演示的程序，而且要写一个图形界面。我一开始想用Java来写，无奈好久没用过java，而且时间有限。后来想向其他同学一样在网上找一个代码，自己改改罢了。于是我找到一个使用C来写的。C语言我还是比较熟悉的，可是那个程序用了一个很“古老”的图形库——Graphics.h，这是TC自家的一个图形库，是Dos时代的，如果想使用的话还要安装一些插件。。。我没那个耐心了，要知道，读代码不比写代码简单。于是我买了瓶咖啡，用了一个晚上熟悉了一下python的Tkinter图形库——毕竟人生苦短，我用python，然后写了一个很low的程序，仅仅是为了实现老师的要求，不到一百行。

首先，以层次遍历输入一个二叉树，并用*补成满二叉树。然后回车，会输出该二叉树的先序遍历顺序（二叉树中的节点不能有重复的），并且弹出一个界面，用鼠标点击该界面，节点会按照先序遍历的顺序显示节点，等到最后一个节点显示后，再次点击界面，5秒后程序结束。

代码如下：

#coding=utf-8

#Write By FiPhoenix

#frome http://blog.csdn.net/fi__phoenix

#2015.11.11

import time,Tkinter

tree = []  #保存二叉树

l = [] #每次出现的节点的下标

#树的节点类

class tree_point:

    def __init__(self,name='',x=0,y=0):

        self.name = name

        self.x = x #节点的x坐标

        self.y = y #节点的y坐标

    def set_xy(self,x=0,y=0):

        self.x = x;

        self.y = y

        

#先序遍历

def preorder(btree,i,order):

    if i >= len(btree) or btree[i] == "*":

    return

    order.append(btree[i])

    preorder(btree,i*2+1,order)

    preorder(btree,i*2+2,order)

#获得父节点下标

def get_f():

    i = l[0]

    if i == 0:

        return -1

    if i/2.0-i/2 == 0.5:

        f = i/2

    else:

        f = i/2-1

    return f

#事件监听器，左击鼠标一次，显示一个节点

def draw_point(event):

    if l == []:

        print("End after 5's!...")

        time.sleep(5)

        exit(0)

    if tree[l[0]].name != "*":

        x = tree[l[0]].x

        y = tree[l[0]].y

        name = tree[l[0]].name

        f = get_f()

        

        p = c.create_oval(x-10,y-10,x+10,y+10)

        c.create_text(x,y,text=name)

        if f != -1:

            c.create_line(tree[f].x,tree[f].y,x,y)

    l.remove(l[0])

if __name__ == '__main__':

    order = list()  #先序遍历的顺序

    print("输入二叉树,空子树用*代替，回车结束:")

    s = raw_input()

    btree = list(s)

    #对二叉树进行先序遍历

    preorder(btree,0,order)

    print order

    for i in s:

        temp = tree_point(i)

        tree.append(temp)

    #每次出现的节点的下标

    for i in order:

        l.append(s.index(i))

    print l

    #设置每个节点的坐标

    tree[0].set_xy(400,10)

    for i in range(1,len(s)):

        if i/2.0 - i/2 == 0.5:

            f = i/2

            tree[i].set_xy(tree[f].x-20-100.0/(f+1),tree[f].y+30)

     #       print tree[i].x,tree[i].y,tree[i].name,f

        else:

            f = i/2-1

            tree[i].set_xy(tree[f].x+20+100.0/(f+1),tree[f].y+30)

     #       print tree[i].x,tree[i].y,tree[i].name,f

    #创建图形界面

    top = Tkinter.Tk()

    c = Tkinter.Canvas(top,height=700,width=800)

    c.bind("<Button-1>",draw_point)

    c.pack()

    top.mainloop()

    input()

最后装个*吧。转载请注明出处。