数据结构习题回顾与总结 (2) List Leaves-CSDN博客

本文链接：https://blog.csdn.net/whtqsq/article/details/78109473

还是pat上的一道习题，具体描述如下:
这里写图片描述

题意大致是这样的:输入的第一行代表树中总节点个数,然后后面的行里,第一位代表左节点的位置，第二位代表右节点的位置,如果是”-“则代表左节点或右节点不存在。

举个例子:第0个节点也就是8下面的那行是1 -,就代表节点0的左节点为第1个节点(- -)右节点不存在。

然后根据给出的树，按照从上到下从左到右的顺序将所有的叶节点输出,输出形式如图。

看到这个题之后感觉很容易就会想到使用层序遍历。按照层序遍历将节点们按层序遍历的顺序放到一个list里,然后判断谁的左右都是“-”,然后按顺序输出即可。

首先大致回顾一下层序遍历,第一点是从上到下从左到右,第二点就是用队列或者堆栈实现。

下面是python的源码：

data = 0
left = 1
right = 2
n = int(input())
tr = []
leaves = []

def gen(tree,n):
    biao = [0]*n
    for i in range(n):
        if tree[i][left]!=-1:
            biao[tree[i][left]] = 1
        if tree[i][right]!=-1:
            biao[tree[i][right]] = 1
        else:
            continue
    for i in range(n):
        if biao[i]==0:
            return i
        else:
            continue

def sortt(tree,root,n):
    res = []
    que = []
    if que == []:
        que.insert(0,tree[root])  
    while que!=[]:                                 
        k = que.pop()
        res.append(k)
        if k[left] != -1:
            que.insert(0,tree[k[left]])
        if k[right] != -1:
            que.insert(0,tree[k[right]])
        else:
            continue        
    return res

for i in range(n):
    k=[]
    q = input().split()
    k.append(i)
    if q[0] != '-':
        k.append(int(q[0]))
    else:
        k.append(-1)
    if q[1] != '-':
        k.append(int(q[1]))
    else:
        k.append(-1)
    tr.append(k)

root = gen(tr,n)
mid = sortt(tr,root,n)

for i in range(len(mid)):
    if mid[i][left]==-1 and mid[i][right]==-1:
        leaves.append(mid[i][0])

print(" ".join([str(x) for x in leaves]))