【华为OD机试Python】广度优先遍历之树状结构查询

广度优先问题是机考常考题型之一。这类题型一般有独特的解题结构。

---

---

题目表述

通常使用多行的结点、父节点表示一棵树，比如
西安 陕西
陕西 中国
江西 中国
中国 亚洲
泰国 亚洲
输入一个节点后、请打印出来树中他的所有下层结点

输入

第一行输入行数，下面是多行数据，每行以空格区分结点和父节点
最后是查询结点

输出

输出查询结点的所有下层结点。以字典序排序。

示例1

输入

5
b a
c a
d c
e c
f d
c

输出

d
e
f
c的子节点有d、e
e的子节点有f
所以最终输入d、e、f

示例2

输入

6
b a
c a
d b
e c
g e
x y
a

输出

b
c
d
e
g
a的下层结点是b和c
b的下层结点是d
c的下层结点是e
e的下层结点是g

题目说明

题意：由父节点输出所有子节点
思路：广度优先，层序遍历。

解题思路及代码

典型的广度优先遍历（bfs）。先用字典存储父子结点，运用列表存储待展开结点，并把子节点添加到列表最后。

代码如下（示例）：

n = int(input())
r = {}
for i in range(n):
    s = input().split()
    if s[1] not in r.keys():
        r[s[1]] = []
    r[s[1]].append(s[0])
t = input()
res = []
wait = []
wait.extend(r[t])
while len(wait) > 0:
    a = wait.pop(0)
    res.append(a)
    if a in r.keys():
        wait.extend(r[a])  # 一个结点一般会有多个子节点，这里用extend方便遍历
res.sort()
for j in res:
    print(j)

---

总结

广度优先遍历常用于解决包含层序关系的题目。