608. 树节点

115 阅读3分钟

携手创作,共同成长!这是我参与「掘金日新计划 · 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)

链接:leetcode.cn/problems/tr…

著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

二、解题思路

创建数据表

根据题意创建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

image.png

CASE WHEN THEN

image.png

IF

image.png

四、总结

这里主要是搞清楚什么是根节点,什么是内节点,什么是叶子节点

  • p_id为null,就是根节点
  • id存在p_id中,且不为null,就是内节点
  • id不存在p_id中,且不为null,就是叶子节点

IF(expr1,expr2,expr3)

该语句是三个参数,第一个是条件,第二和第三个是结果,当然也可以嵌套,如上边的第三种执行语句一样。