题解 | #二叉树之寻找第k大#
二叉树之寻找第k大
https://www.nowcoder.com/practice/8e5f73fa3f1a407eb7d0b0d7a105805e
/**
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* };
*/
class Solution {
public:
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param root TreeNode类
* @param k int整型
* @return int整型
*/
void dfs(TreeNode* root,vector<int>& data)
{
if (root == NULL)
{
return;
}
data.push_back(root->val);
dfs(root->left,data);
dfs(root->right,data);
}
int kthLargest(TreeNode* root, int k)
{
// write code here
vector<int> data;
dfs(root,data);
sort(data.begin(),data.end(),[](const int a,const int b)
{
return a > b;
});
return data[k - 1];
}
};
韶音科技公司氛围 643人发布