[C++]Kth Smallest Element in a BST 在一个二叉排序树中找第k小的元素
2015-08-29 22:12
337 查看
leetcode 原题:https://leetcode.com/problems/kth-smallest-element-in-a-bst/
Given a binary search tree, write a function
smallest element in it.
Note:
You may assume k is always valid, 1 ≤ k ≤ BST's total elements.
简要分析:题目的意思很明确,需要我们在二叉排序树中找到第k小元素
因此,根据二叉排序树最大的特点就是左子树的元素小于根节点,右子树的元素大于根节点,中序遍历就可以得到一个从小到大排序的数组。
这样的时间复杂度是O(N)。
但是,如果k = 1呢? 此时你还把所有元素都排序一遍再输出最小的元素是不是就很浪费时间呢?
其实,我们根据中序遍历的特点可以知道,中序遍历输出的元素就是从小到大的,那么如果我们能够记录其遍历输出的元素个数,那么我们就能知道,第k个元素的输出位置。此时就可以终止程序了。
实现代码
#include <iostream>
using namespace std;
//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 kthSmallest(TreeNode* root, int k)
{
int result = 0;
bool flag = false;
int times = 0;
inorder(root, k, result, flag, times);//root表示节点,k表示需要找到第k小的元素,result保存我们需要的结果,flag判断是否找到了最小元素,times表示当前元素是第times小的元素
return result;
}
void inorder(TreeNode *root, int k, int &result, bool &flag, int ×)
{
if (root == NULL && times >= k)
return;
if (root->left != NULL)
{
inorder(root->left, k, result, flag, times);
} else
{
flag = true;
}
if (flag)
{
times++;
}
if (k == times)
{
result = root->val;
return;
}
if (root->right != NULL)
{
inorder(root->right, k, result, flag, times);
}
return;
}
};
Given a binary search tree, write a function
kthSmallestto find the kth
smallest element in it.
Note:
You may assume k is always valid, 1 ≤ k ≤ BST's total elements.
简要分析:题目的意思很明确,需要我们在二叉排序树中找到第k小元素
因此,根据二叉排序树最大的特点就是左子树的元素小于根节点,右子树的元素大于根节点,中序遍历就可以得到一个从小到大排序的数组。
这样的时间复杂度是O(N)。
但是,如果k = 1呢? 此时你还把所有元素都排序一遍再输出最小的元素是不是就很浪费时间呢?
其实,我们根据中序遍历的特点可以知道,中序遍历输出的元素就是从小到大的,那么如果我们能够记录其遍历输出的元素个数,那么我们就能知道,第k个元素的输出位置。此时就可以终止程序了。
实现代码
#include <iostream>
using namespace std;
//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 kthSmallest(TreeNode* root, int k)
{
int result = 0;
bool flag = false;
int times = 0;
inorder(root, k, result, flag, times);//root表示节点,k表示需要找到第k小的元素,result保存我们需要的结果,flag判断是否找到了最小元素,times表示当前元素是第times小的元素
return result;
}
void inorder(TreeNode *root, int k, int &result, bool &flag, int ×)
{
if (root == NULL && times >= k)
return;
if (root->left != NULL)
{
inorder(root->left, k, result, flag, times);
} else
{
flag = true;
}
if (flag)
{
times++;
}
if (k == times)
{
result = root->val;
return;
}
if (root->right != NULL)
{
inorder(root->right, k, result, flag, times);
}
return;
}
};
相关文章推荐
- 使用C++实现JNI接口需要注意的事项
- 关于指针的一些事情
- 搜狗百度360市值齐跌:搜索引擎们陷入集体焦虑?
- AVL树-自平衡二叉查找树(Java实现)
- c++ primer 第五版 笔记前言
- 本人即将筹备败家日志,敬请期待!
- share_ptr的几个注意点
- IE:使用搜索助手
- 动易2006序列号破解算法公布
- Ruby实现的矩阵连乘算法
- C#插入法排序算法实例分析
- Lua中调用C++函数示例
- Lua教程(一):在C++中嵌入Lua脚本
- Lua教程(二):C++和Lua相互传递数据示例
- 超大数据量存储常用数据库分表分库算法总结
- C#数据结构与算法揭秘二
- C#冒泡法排序算法实例分析
- 算法练习之从String.indexOf的模拟实现开始
- C#算法之关于大牛生小牛的问题
- C++联合体转换成C#结构的实现方法