题目
给定一颗二叉搜索树,请找出其中的第k大的结点。
思路
如果中序遍历一棵二叉搜索树,遍历序列的数值则是递增排序,因此只需中序遍历一个二叉搜索树即可。
#include <iostream> using namespace std; struct tree { double data; struct tree *left,*right; tree(int d=0):data(d) { left=right=nullptr; } }; class Solution { public: void create(tree *&root); tree *kth_node(tree * root,unsigned int k); tree *kth_node_core(tree *root,unsigned int &k); tree *root; }; void Solution::create(tree *&root) { double x; cin>>x; if(x==0) root=nullptr; else { root=new tree(); root->data=x; create(root->left); create(root->right); } } tree *Solution::kth_node(tree *node,unsigned int k) { if(!node||k<=0) return nullptr; return kth_node_core(node,k); } tree *Solution::kth_node_core(tree *node,unsigned int &k) { tree *target=nullptr; if(node->left)//遍历左子树 target=kth_node_core(node->left,k); if(!target) { if(k==1) target=node; --k; } if(!target&&node->right)//右子树 target=kth_node_core(node->right,k); return target; } int main() { Solution s; s.create(s.root); tree *node=s.kth_node(s.root,4); if(node) cout<<node->data<<endl; return 0; }
原文地址:https://www.cnblogs.com/tianzeng/p/10269202.html
时间: 2024-10-13 04:46:41