zoukankan      html  css  js  c++  java
  • 二叉搜索树的第K大节点

    题目描述

    给定一颗二叉搜索树,请找出其中的第k大的结点。

    分析

    对二叉搜索树进行逆向中序遍历(先右再左),则遍历序列是降序排序的,因此中序遍历一颗二叉搜索树,可以很容易的得到它的第k大的节点。使用一个计数器变量,每遍历一个节点,计数器加1,当计数器的值等于k时,root节点即为所求节点。

    解法一:保存遍历过的节点,计算其数量。

    public class Solution {  
        vector<TreeNode> vec;  
         TreeNode KthNode(TreeNode pRoot, int k)  
        {  
            if(k == 0 ) return null;  
                   zhongxu(pRoot);  
            if(k > vec.size()) return null;      
            return vec.get(k-1);  
        }  
         void zhongxu(TreeNode pRoot){  
                    if(pRoot == null) return ;  
                    zhongxu(pRoot.right);  
                    vec.push_back(pRoot);  
                    zhongxu(pRoot.left);  
        }  
      
    } 

    解法二:利用递归的返回值,判断是否结束。

    class Solution {
        int count = 0;
    public:
        TreeNode* KthNode(TreeNode* pRoot, unsigned int k)
        {
            if(pRoot){ 
                    TreeNode *ret = KthNode(pRoot->right, k);   //如果是叶子节点,到这里返回。然后记数
                    if(ret) return ret;
                    count++;   //this ++ because root is not null
                    if(count == k)
                        return pRoot;
                    ret = KthNode(pRoot->left,k);  //路径是先右后左
                    if(ret) return ret;
            }
            return nullptr;
        }
    };
  • 相关阅读:
    HDU-5514 Frogs 容斥
    2019ICPC EC-FINAL H-King 随机
    2019ICPC EC-FINAL E-Flow 贪心
    洛谷P4200 千山鸟飞绝 Splay
    CodeForces 1249F Maximum Weight Subset 树形dp
    HDU-5534 Partial Tree 完全背包优化
    【数论】Lucas定理
    [APIO2009]抢掠计划 解题报告
    tarjan(缩点)
    树状数组总结
  • 原文地址:https://www.cnblogs.com/LUO77/p/8634015.html
Copyright © 2011-2022 走看看