携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第26天,点击查看活动详情
一、题目
给定一个表 tree,id 是树节点的编号, p_id 是它父节点的 id 。
+----+------+
| id | p_id |
+----+------+
| 1 | null |
| 2 | 1 |
| 3 | 1 |
| 4 | 2 |
| 5 | 2 |
+----+------+
树中每个节点属于以下三种类型之一:
- 叶子:如果这个节点没有任何孩子节点。
- 根:如果这个节点是整棵树的根,即没有父节点。
- 内部节点:如果这个节点既不是叶子节点也不是根节点。
写一个查询语句,输出所有节点的编号和节点的类型,并将结果按照节点编号排序。上面样例的结果为:
+----+------+
| id | Type |
+----+------+
| 1 | Root |
| 2 | Inner|
| 3 | Leaf |
| 4 | Leaf |
| 5 | Leaf |
+----+------+
解释
- 节点 '1' 是根节点,因为它的父节点是 NULL ,同时它有孩子节点 '2' 和 '3' 。
- 节点 '2' 是内部节点,因为它有父节点 '1' ,也有孩子节点 '4' 和 '5' 。
- 节点 '3', '4' 和 '5' 都是叶子节点,因为它们都有父节点同时没有孩子节点。
样例中树的形态如下:
1
/ \
2 3
/ \
4 5
注意
如果树中只有一个节点,你只需要输出它的根属性。
来源:力扣(LeetCode)
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
二、解题思路
创建数据表
根据题意创建Tree数据表
CREATE TABLE tree(
id INT,
p_id INT
);
INSERT INTO tree VALUES
(1,NULL),
(2,1),
(3,1),
(4,2),
(5,2);
解题方式
UNION
这里根据题意得知,如果p_id节点没有父节点,那么该节点就是根节点
SELECT id, 'Root' AS TYPE FROM tree WHERE p_id IS NULL
由此得出id为1的p_id是父节点
判断如果节点既不是叶子结点也不是根节点那么就是内节点
SELECT id, 'Inner' AS TYPE FROM tree
WHERE id IN (
SELECT DISTINCT p_id FROM tree WHERE p_id IS NOT NULL
)
AND p_id IS NOT NULL
这里使用DISTINCT对数据集去重,这里得到的数据集除了为NULL的,其他的就是内节点,因此id在这里的就都是内节点,再外层也要同样的去除这个空节点。
判断叶子节点,就是没有孩子节点,这里和内节点条件相反,就是id不在内节点,也不在根节点的节点,都是叶子节点
SELECT id, 'Leaf' AS TYPE FROM tree WHERE id NOT IN(
SELECT DISTINCT p_id FROM tree WHERE p_id IS NOT NULL
)
AND p_id IS NOT NULL;
把三种节点通过UNION连接起来
CASE WHEN THEN
解题思路同上,但是语句换一种。使用case做循环,when判断当id在不在集合中,是否为NULL,判断根节点和内节点后,最后的就是叶子节点,直接使用ELSE赋值
SELECT
id,
CASE
WHEN id=(SELECT id FROM tree WHERE tree.p_id IS NULL) THEN 'Root'
WHEN id IN (SELECT p_id FROM tree) THEN 'Inner'
ELSE 'Leaf'
END TYPE
FROM
tree
ORDER BY id;
IF
这里使用IF做判断,IF语句类似二元一次表达式,如果为NULL,那么就是Root,否则就是子节点,子节点里再区分,id是否在p_id中,在的就是内节点,不在的就是叶子节点。
SELECT
id,
IF(ISNULL(p_id), 'Root', IF(id IN (SELECT p_id FROM tree), 'Inner', 'Leaf')) TYPE
FROM tree
ORDER BY id;
三、执行语句
UNION
CASE WHEN THEN
IF
四、总结
这里主要是搞清楚什么是根节点,什么是内节点,什么是叶子节点
- p_id为null,就是根节点
- id存在p_id中,且不为null,就是内节点
- id不存在p_id中,且不为null,就是叶子节点
IF(expr1,expr2,expr3)
该语句是三个参数,第一个是条件,第二和第三个是结果,当然也可以嵌套,如上边的第三种执行语句一样。