二叉树先序遍历动态演示——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()

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

    原文作者:Fi__Phoenix
    原文地址: https://blog.csdn.net/Fi__Phoenix/article/details/49785449
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞