我通过一道很简单的题来帮大家了解一下二叉树的前中后序遍历
题目描述:
给定一棵二叉搜索树,请找出其中第 k 大的节点的值。
示例:
这道题目很简单,但是我们要重点注意一下二叉搜索树,根据二叉搜索树的特性,我们可以提高代码的效率。
什么是二叉搜索树:
- 左子树上的各个节点均小于根节点的值
- 右子树上的各个节点均大于根节点的值
根据这个特性,我们再来看看题目,找到第k大的结点,那么就要求我们遍历的结点顺序是由大至小的
我们就以示例中的二叉树为例进行演示:
- 1.前序遍历:(先遍历根节点,在遍历左节点 ,在遍历右节点)
我们的深度优先遍历本质就是前序遍历
结果如下:3->1->2->4
void dfs(TreeNode root){
if(root==null) return;//终止条件
print(root.val);
dfs(root.left);
dfs(root.right);
}
- 2.中序遍历(先遍历左结点,在遍历根节点,最后遍历右节点)
结果如下:1->2->3->4
void dfs(TreeNode root){
if(root==null)return ;//循环终止条件
dfs(root.left);
print(root.val);
dfs(root.right);
}
- 3.后序遍历(先遍历左节点,在遍历右节点,最后遍历根节点)
结果如下:2->1->4->3
void dfs(TreeNode root){
if(root==null)return ;//循环终止条件
dfs(root.left);
dfs(root.right);
print(root.val);
}
有了以上知识,我们就可以以不变应万变
题目中找第k大的结点,然后我们的预期是按递减顺序排列,那么我们可以通过右序遍历(ps:名字我自己起的),什么意思呢:遍历的顺序为:右节点,中间节点,左节点。
结果如下:4->3->2->1
刚好符合我们的预期,最终代码如下:
void dfs(TreeNode root){
if(root==null)return ;//循环终止条件
dfs(root.right);
print(root.val);
dfs(root.left);
}
题目答案完整代码:
int res;
int index;
public int kthLargest(TreeNode root, int k) {
dfs(root, k);
return res;
}
public void dfs(TreeNode root, int k) {//表示统计第k个节点
{
if (root == null) return;
dfs(root.right, k); //右
index++;
if (k == index) res = root.val; //根
dfs(root.left, k); //左
}
}