SwordはOffer54(バイナリ検索ツリーのk番目に大きいノード)を指します-簡単

質問リンク

タイトル説明:

二分探索木が与えられた場合、その中でk番目に大きいノードを見つけてください。

たとえば、次のように入力します。

输入: root = [3,1,4,null,2], k = 1
   3
  / \
 1   4
  \
   2
输出: 4

输入: root = [5,3,6,2,4,null,null,1], k = 3
       5
      / \
     3   6
    / \
   2   4
  /
 1
输出: 4

データ範囲:

1 ≤ k ≤ 二叉搜索树元素个数

問題解決のアイデア:

二分探索木では、左側のサブツリーのすべてのノードがルートノードよりも小さく、右側のサブツリーのすべてのノードがルートノードよりも大きくなります。左右のサブツリーのすべてのノードがこのルールを満たします。
二分探索木の中次走査シーケンスは、増加するシーケンスです。しかし、この質問で必要なのは、k番目に大きいノードです。したがって、ミドルオーダートラバーサルの逆の順序が必要です。
再帰的走査シーケンス:1。右サブツリー2.ルートノード3.左サブツリー
numを使用して、現在の値の数を記録します。num== kの場合、再帰を早期に終了できます。

ACコード(c ++)

/**
 * 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:
    void dfs(TreeNode * root,int k,int &num,int &ans){
    
    
        if(!root)
            return;
        dfs(root->right,k,num,ans);
        num++;
        if(num==k)
        {
    
    
            ans = root->val;
            return;
        }
        dfs(root->left,k,num,ans);
        
        
    }
    int kthLargest(TreeNode* root, int k) {
    
    
        int ans=0;
        int num=0;
        dfs(root,k,num,ans);
        return ans;
    }
};

おすすめ

転載: blog.csdn.net/Yang_1998/article/details/113049112
おすすめ