leetcode:剑指 Offer 54. 二叉搜索树的第k大节点

方法:

中序遍历,迭代,深度递归(优先队列)

解题思路

一 中序遍历
中序遍历是:左子树-父节点-右子树,题目是二叉搜索树,故采用中序遍历的话,值是从小到大依次遍历输出的,套用二叉树的解题模板c++二叉树:层序,前,中,后 序遍历的解题模板,将中序遍历的值存入vector,则倒数第k大的值在vec.size()-k的位置。

代码

class Solution {
public:    
    int kthLargest(TreeNode* root, int k) {     
        //中序:左子树-父节点-右子树
		TreeNode* p = root;
        vector<int> data;
		stack<TreeNode*> st;
		while (p||!st.empty()) {
			if (p) {
				st.push(p);
				p = p->left;
			}
			else {
				p = st.top();
				st.pop();
				data.push_back(p->val);//
                cout << p->val << " ";
				p = p->right;
			}
		}
        return data[data.size()-k];
    }   
};

二 迭代
右-根-左的遍历方式,因为右>根>左,所以从最右端的根依次出栈,知道k --==0,就返回此时的栈顶元素的值。

class Solution {
public:    
    int kthLargest(TreeNode* root, int k) {
        stack<TreeNode*>sta;
        TreeNode* p=root;
        while(!sta.empty()||k){
            while(p){
                sta.push(p);
                p=p->right;
            }    
            cout<<sta.top()->val<<" ";    
            TreeNode* temp=sta.top();
            --k;
            if(!k)return sta.top()->val;            
            sta.pop();
            if(temp->left)
                p=temp->left;
        }
        return -1;
    }   
};

三 深度递归
采用priority_queue ,用了最大堆,先将所有的元素一次推入priority_queue,然后再每次弹出一个,知道弹出的第k个

class Solution {
public:
    priority_queue<int> pri_que;
    int kthLargest(TreeNode* root, int k) {
        dfs(root);
        int n;
        while( k){
            n=pri_que.top();
            k--;
            cout<<n<<"--";
            pri_que.pop();
        }
        return n;
    }
    void dfs(TreeNode* root){
        pri_que.push(root->val);
        if(root->left)dfs(root->left);
        if(root->right)dfs(root->right);
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值