2020-06-22
1.題目描述
給定一棵二叉搜索樹,請找出其中第k大的節點。
2.題解
中序遍歷二叉搜索樹即爲升序的結果,將其存儲在vector中,最後進行輸出即可。
3.代碼
/**
* 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 kthLargest(TreeNode* root, int k) {
if (!root) return -1;
stack<TreeNode*> mystack;
TreeNode* p=root;
vector<int> tmp;
while (p||!mystack.empty()){
if (p){
mystack.push(p);
p=p->left;
}else{
p=mystack.top();
mystack.pop();
tmp.push_back(p->val);
p=p->right;
}
}
int index=tmp.size()-k;
return tmp[index];
}
};