二叉排序树中的第k小元素

【题目描述】

如题:查找二叉排序树中的第k小元素【若是结点的话,就直接输出TreeNode就行】。即leetcode的230题


【方法一】

1、计算左子树元素个数left。

2、 left+1 = K,则根节点即为第K个元素

      left >=k, 则第K个元素在左子树中,

     left +1 <k, 则转换为在右子树中,寻找第K-left-1元素。

计算树中的节点个数    
    int calcTreeSize(TreeNode* root){  
        if (root == NULL)  
            return 0;  
        return 1+calcTreeSize(root->left) + calcTreeSize(root->right);          
    }  
   
      int kthSmallest(TreeNode* root, int k) {  
        if (root == NULL)  
            return 0;  
        int leftSize = calcTreeSize(root->left);  
        if (k == leftSize+1){  
            return root->val;  
        }else if (leftSize >= k){  
            return kthSmallest(root->left,k);  
        }else{  
            return kthSmallest(root->right, k-leftSize-1);  
        }  
    }  

【方法二】

因中序遍历为一个有序的数组,所以可以在中序遍历的过程中进行比较【在元素出栈的时候比较—-因数据肯定从小到大的出栈】。

int kthSmallest(TreeNode root, int k) {  
        Stack<TreeNode *> stack ;  
        if(root == NULL || K<=0)
              return -1;//表示不存在
        
        TreeNode *P = root;  
        while(P->left != NULL) {  //也是个循环
            stack.push(P);  
            P = P->left;   
        }  
          
        while(k>0 && (P != NULL || !stack.empty())) {  //注意k这个条件别忘记
            if(p==NULL) {  
                p = stack.top();
                stack.pop();
                //查找输出的过程中进行判断,是否为第k个元素
                 if(--k==0) 
                     return p->val;  
                
                  p = p->right;  
            } else {  
                stack.push(p);  
                p = p->left;  
            }  
        }   
    }  

时间复杂度:最多为O(N)

【注意】若求,第k大数,则方法一和方法二,均变为:先访问右子树—>再访问左子树即可。其他不变

参考:http://blog.csdn.net/sunao2002002/article/details/46726245

http://blog.csdn.net/xudli/article/details/46784263

    原文作者:suibianshen2012
    原文地址: https://blog.csdn.net/suibianshen2012/article/details/52051272
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞