Kth Smallest Element in a BST
来源:互联网 发布:中国省份数据库 编辑:程序博客网 时间:2024/06/18 07:39
这道题不难。
但是卡了一会,原因是if(count=k-1) 我没写==。 然后突然想起吉林大学的一位C++老师,碰到 这种情况,我们要写k-1 ==count!!!
递归思路如下
1.若左子树有k-1个节点,返回根的值
2.若左子树有多余k-1个,则在左子树找第k小的
3.若左子树少于k-1个,则在右子树找 k-左子树节点数-1小的
代码如下:class Solution {
public:
int kthSmallest(TreeNode* root, int k) {
if(!root)
return 0;
int count = count_subtree_nodes(root->left);
//cout<<count<<endl;
if(count==k-1)
return root->val;
else
if(count>k-1)
return kthSmallest(root->left,k);
else
return kthSmallest(root->right,k-1-count);
}
int count_subtree_nodes(TreeNode *root)
{
if(!root)
return 0;
return count_subtree_nodes(root->left)+count_subtree_nodes(root->right)+1;
}
};
public:
int kthSmallest(TreeNode* root, int k) {
if(!root)
return 0;
int count = count_subtree_nodes(root->left);
//cout<<count<<endl;
if(count==k-1)
return root->val;
else
if(count>k-1)
return kthSmallest(root->left,k);
else
return kthSmallest(root->right,k-1-count);
}
int count_subtree_nodes(TreeNode *root)
{
if(!root)
return 0;
return count_subtree_nodes(root->left)+count_subtree_nodes(root->right)+1;
}
};
0 0
- #leetcode#Kth Smallest Element in a BST
- 230.Kth Smallest Element in a BST
- leetcode--Kth Smallest Element in a BST
- Kth Smallest Element in a BST
- Kth Smallest Element in a BST
- Leetcode|Kth Smallest Element in a BST
- Kth Smallest Element in a BST
- [leetcode] Kth Smallest Element in a BST
- 【LeetCode】Kth Smallest Element in a BST
- Kth Smallest Element in a BST
- LeetCode Kth Smallest Element in a BST
- Kth Smallest Element in a BST
- 【leetcode】Kth Smallest Element in a BST
- Kth Smallest Element in a BST
- Kth Smallest Element in a BST
- LeetCode Kth Smallest Element in a BST
- Kth Smallest Element in a BST
- leetcode: Kth Smallest Element in a BST
- dp计算
- 山东省第七届ACM省赛 Triple Nim
- 删除数据
- Android 根据文件Uri找文件路径 和根据路径找文件Uri
- 往mysql中插入 自增id不连续(曾经的坑)
- Kth Smallest Element in a BST
- DSP2808与ARM STM32F103的SPI通讯例程及详解
- PowerDesigner16.5快速入门显示,注释comment配置方法,以及创建sql文件过程中需要注意的一些问题
- Unknown module(s) in QT: multimedia
- CCF NOI1015 星期几
- 几种决策树学习算法比较
- Leetcode 290. Word Pattern
- oracle存储过程中的while loop循环示例
- 刷新当前目录下所有文件的shell脚本