タイトル説明:
二分探索木が与えられた場合、その中で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;
}
};