python用turtle库绘画树形,python数据结构递归树

大家好，小编来为大家解答以下问题，python用turtle库绘画树形，python数据结构递归树，现在让我们一起来看看吧！

Source code download: 本文相关源码

二叉树形图，显示如下：

基本思路：
首先我们根据前几幅图可以知道迭代次数是2^n - 1，n为迭代层数，因此递归函数大致结构出来了，也就是：

def func():
    if xxx: return   
    else:
        do something
        func()
        func()

xxx是判断结束递归的条件，这里我们很容易知道条件就是level == 0

do something 就是我们要在图上画的动作

那么具体是怎么画呢？

turtle中有俩函数

turtle.right(angle)
turtle.forward(r)

第一个是以当前角度顺时针旋转一个angle角度

第二个是以当前位置为起点，按当前角度画线段，长度为r

记住两个函数都是以当前位置和当前角度的，因此在画线的时候必须调整好起点位置和角度

turtle中有这几个函数

turtle.setheading(0)    #调整角度到0度
turtle.penup()          #抬笔
turtle.goto(x,y)        #将画笔位置移到x，y
turtle.pendown()        #落笔

新的线段起点newx，newy位置可以根据原位置乘以角度的三角函数得到

newx = x+r*math.cos(angle/180 *math.pi)
newy = y-r*math.sin(angle/180 *math.pi)

综上所诉，我们的递归函数的就可以确定了。

代码如下：

import turtle
import math

R = 100

def TreeGraph(x,y,r,angle,level):
    if level == 0:return
    else:
        turtle.setheading(0)
        turtle.penup()
        turtle.goto(x,y)
        turtle.pendown()
        turtle.right(angle)
        turtle.forward(r)

        newx = x+r*math.cos(angle/180 *math.pi)
        newy = y-r*math.sin(angle/180 *math.pi)

        TreeGraph(newx,newy,r*0.5*(2**.5),angle+45,level-1)
        TreeGraph(newx,newy,r*0.5*(2**.5),angle-45,level-1)


if __name__ == "__main__":

    TreeGraph(0,-200,R,-90,9)

    turtle.mainloop()