题解 | #第k轻的牛牛#
第k轻的牛牛
https://www.nowcoder.com/practice/d3b31f055b1640d9b10de0a6f2b8e6f3
/**
* 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 kthLighest(TreeNode* root, int k)
{
// write code here
vector<int> data;
dfs(root,data);
sort(data.begin(),data.end());
return data[k - 1];
}
};