题目
给定一个表 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)
链接:https://leetcode.cn/problems/tree-node
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解答
树节点
方法 1:使用 UNION [Accepted]
想法
我们可以按照下面的定义,求出每一条记录的节点类型。
Root: 没有父节点
Inner: 它是某些节点的父节点,且有非空的父节点
Leaf: 除了上述两种情况以外的节点
算法
将上述定义转化,我们可以得到下面的代码。
根节点是没有父节点的节点。
sql
SELECT
id, ‘Root’ AS Type
FROM
tree
WHERE
p_id IS NULL
叶子节点是没有孩子节点的节点,且它有父亲节点。
sql
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
内部节点是有孩子节点和父节点的节点。
sql
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
所以本题的一种解法是将这些情况用 UNION 合并起来。
sql
SELECT
id, ‘Root’ AS Type
FROM
tree
WHERE
p_id IS NULL
UNION
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
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
ORDER BY id;
方法 II:使用流控制语句 CASE [Accepted]
算法
与上面解法类似,本解法使用流控制语句,流控制语句对基于不同输入产生不同输出非常有效。本方法中,我们使用 CASE 语句。
sql
SELECT
id AS Id
,
CASE
WHEN tree.id = (SELECT atree.id FROM tree atree WHERE atree.p_id IS NULL)
THEN ‘Root’
WHEN tree.id IN (SELECT atree.p_id FROM tree atree)
THEN ‘Inner’
ELSE ‘Leaf’
END AS Type
FROM
tree
ORDER BY Id
;
MySQL 除了 CASE 语句以外还提供了不同的流控制语句。你可以尝试将上面的方法用 IF 重写。
方法 III;使用 IF 函数 [Accepted]
算法
我们还可以使用 IF 函数来避免复杂的流控制语句。
sql
SELECT
atree.id,
IF(ISNULL(atree.p_id),
‘Root’,
IF(atree.id IN (SELECT p_id FROM tree), ‘Inner’,‘Leaf’)) Type
FROM
tree atree
ORDER BY atree.id
作者:LeetCode
链接:https://leetcode.cn/problems/tree-node/solution/shu-jie-dian-by-leetcode/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。