精品推荐:
《征服数据结构》专栏:50多种数据结构彻底征服
《经典图论算法》专栏:50多种经典图论算法全部掌握
一网友最近在求职过程中发现华为校招似乎特别缺人,各大网站上都有华为的员工在招人,目前为止已经有三个部门的HR联系他了。而网友回复他说这是HR的kpi任务,还有说这是脚本自动发的。不过我觉得这应该是真的招人,如果浙大的简历都不过,还想要什么样的?
--------------下面是今天的算法题--------------
来看下今天的算法题,这题是LeetCode的第114题:二叉树展开为链表。
问题描述
来源:LeetCode第114题
难度:中等
给你二叉树的根结点 root ,请你将它展开为一个单链表:
1,展开后的单链表应该同样使用 TreeNode ,其中 right 子指针指向链表中下一个结点,而左子指针始终为 null 。
2,展开后的单链表应该与二叉树先序遍历顺序相同。
示例1:
输入:root = [1,2,5,3,4,null,6]
输出:[1,null,2,null,3,null,4,null,5,null,6]
示例2:
输入:root = []
输出:[]
树中结点数在范围 [0, 2000] 内
-100 <= Node.val <= 100
问题分析
这题是让把二叉树展开为链表的形状,实际上它不是一个链表,还是一棵二叉树,只不过这棵二叉树没有左子树只有右子树,所以形状像一个链表。
如果我们直接展开,也就是把二叉树的左子树放到右子树上,那么右子树上的节点就会被覆盖,所以这种方式是行不通的。
这题要求的是展开之后的顺序和二叉树的前序遍历顺序相同,前序遍历是:根节点->左子树->右子树。我们来逆向思考一下,如果对二叉树的遍历方式改成:右子树->左子树->根节点,这种方式相当于链表的逆序遍历,我们只需要在遍历的时候把这些节点逆序串起来即可。
JAVA:
// 相当于链表的头节点,始终是变动的。
TreeNode head = null;
// 参照二叉树的后序遍历
public void flatten(TreeNode root) {
if (root == null)
return;
flatten(root.right);// 先展开右子树
flatten(root.left);// 在展开左子树
// 操作当前节点
root.right = head;
root.left = null;// 左子树为空
head = root;// 更新head节点
}
C++:
public:
// 相当于链表的头节点,始终是变动的。
TreeNode *head = nullptr;
void flatten(TreeNode *root) {
if (root == nullptr)
return;
flatten(root->right);// 先展开右子树
flatten(root->left);// 在展开左子树
// 操作当前节点
root->right = head;
root->left = nullptr;// 左子树为空
head = root;// 更新head节点
}
Python:
head = None # 相当于链表的头节点,始终是变动的。
# 参照二叉树的后序遍历
def flatten(self, root: Optional[TreeNode]) -> None:
if root is None:
return
self.flatten(root.right) # 先展开右子树
self.flatten(root.left) # 在展开左子树
# 操作当前节点
root.right = self.head
root.left = None # 左子树为空
self.head = root # 更新head节点
笔者简介
博哥,真名:王一博,毕业十多年,《算法秘籍》作者,专注于数据结构和算法的讲解,在全球30多个算法网站中累计做题2000多道,在公众号中写算法题解800多题,对算法题有自己独特的解题思路和解题技巧,喜欢的可以给个关注,也可以下载我整理的1000多页的PDF算法文档。
数组,稀疏表(Sparse Table),单向链表,双向链表,块状链表,跳表,队列和循环队列,双端队列,单调队列,栈,单调栈,双端栈,散列表,堆,字典树(Trie树),ArrayMap,SparseArray,二叉树,二叉搜索树(BST),笛卡尔树,AVL树,树堆(Treap),FHQ-Treap,哈夫曼树,滚动数组,差分数组,LRU缓存,LFU缓存
……
图的介绍,图的表示方式,邻接矩阵转换,广度优先搜索(BFS),深度优先搜索(DFS),A*搜索算法,迭代深化深度优先搜索(IDDFS),IDA*算法,双向广度优先搜索,迪杰斯特拉算法(Dijkstra),贝尔曼-福特算法(Bellman-Ford),SPFA算法,弗洛伊德算法(Floyd),卡恩(Kahn)算法,基于DFS的拓扑排序,约翰逊算法(Johnson)
……