Leetcode (+): Kth smallest Element in a BST

Source: Internet
Author: User

Given A binary search tree, write a function to kthSmallest find the k-th smallest element in it.

Note:
You may assume k are always valid, 1≤k≤bst's total elements.

Intuitively, looking for a small number of k, is not traversing to the number k? So the middle sequence traversal is easy to think of, as follows:

/** * Definition for a binary tree node. * struct TreeNode {*     int val; *     TreeNode *left; *     TreeNode *right; *     TreeNode (int x): Val (x), left (NULL) , right (NULL) {}}; */class Solution {public:    int kthsmallest (treenode* root, int k) {    stack<treenode*> nodestack;    treenode* Tmp=root;    int kthmin = 0;    int kthvalue;    while ((!nodestack.empty () | | tmp!=null) && kthmin!=k)    {while    (tmp! = NULL)    {    Nodestack.push (TMP);    TMP = tmp->left;    }    TMP = Nodestack.top ();    Nodestack.pop ();    kthmin++;//minor modifications to the middle order traversal    Kthvalue = tmp->val;    TMP = tmp->right;    }    return kthvalue;}    ;

In addition, looked at the answer of the netizen, very ingenious. He is the first to count the number of nodes on the left subtree, if the number of nodes is less than k, in the right sub-tree to find the k-n-1 small number, if just K is the current node, if greater than K, then continue to find the number K small on the left subtree. However, each recursive to count the number of nodes, will it lead to increased complexity?


int kthsmallest (treenode* root, int k) {    if (!root) return 0;    if (k==0) return root->val;    int n=count_size (root->left);    if (k==n+1) return root->val;    if (n>=k) {        return kthsmallest (Root->left, k);    }    if (n<k) {        return kthsmallest (Root->right, k-n-1);}    } int count_size (treenode* root) {    if (!root) return 0;    Return 1+count_size (Root->left) +count_size (root->right);}


Copyright NOTICE: This article for Bo Master original article, without Bo Master permission not reproduced.

Leetcode (+): Kth smallest Element in a BST

Contact Us

The content source of this page is from Internet, which doesn't represent Alibaba Cloud's opinion; products and services mentioned on that page don't have any relationship with Alibaba Cloud. If the content of the page makes you feel confusing, please write us an email, we will handle the problem within 5 days after receiving your email.

If you find any instances of plagiarism from the community, please send an email to: info-contact@alibabacloud.com and provide relevant evidence. A staff member will contact you within 5 working days.

A Free Trial That Lets You Build Big!

Start building with 50+ products and up to 12 months usage for Elastic Compute Service

  • Sales Support

    1 on 1 presale consultation

  • After-Sales Support

    24/7 Technical Support 6 Free Tickets per Quarter Faster Response

  • Alibaba Cloud offers highly flexible support services tailored to meet your exact needs.