高一计算思维：用Python列表和turtle绘制完全二叉树

高一计算思维：用Python列表和turtle绘制完全二叉树

#打印完全二叉树
import turtle as t
import math as m
total=0
dis=[]#每个节点移动的距离
ds=100#初始移动的距离 ds=100
for i in range(5):
    temp=int(m.pow(2,i))
    total+=temp
    for i in range(temp):#相同层节点移动的距离一样
        dis.append(ds)
    ds=ds/2#下一层节点移动的距离减半
    
Number=[] #节点列表
v=[]      #是否访问过
for j in range(total):
    Number.append(j)
    v.append(0)

#根节点的访问
i=0
v[0]=1

#绘画设置
t.setheading(90)
t.penup()
t.goto(0,-100)
t.speed(0)

#深度遍历
while 1:
    #遍历左子树
    if 2*i+1<len(Number) and v[2*i+1]==0:
        t.pendown()
        t.setheading(90)
        t.fd(dis[i])
        t.setheading(110)
        t.fd(dis[i])
        v[2*i+1]=1
        i=2*i+1
    #遍历右子树    
    elif 2*i+2<len(Number) and v[2*i+2]==0:
        t.pendown()
        t.setheading(90)
        t.fd(dis[i])
        t.setheading(70)
        t.fd(dis[i])
        v[2*i+2]=1
        i=2*i+2
    else:
        #从右子树回退   
        if i%2==0:
            i=int((i-1)/2)
            t.penup()
            t.setheading(70)
            t.back(dis[i])
            t.setheading(90)
            t.back(dis[i])
        else:
            #从左子树回退 
            i=int(i/2)
            t.penup()
            t.setheading(110)
            t.back(dis[i])
            t.setheading(90)
            t.back(dis[i])
    #所有的节点都被访问了，退出程序
    if v.count(0)==0:
        print("End")
        break