题解 | #二叉搜索树的第k个结点#
二叉搜索树的第k个结点
http://www.nowcoder.com/practice/ef068f602dde4d28aab2b210e859150a
三种做法
总体来说 采用非递归的效率比其他方式实现要好一点 占用内存稍微小一点
1、非递归,采用中序遍历,求出第k个节点
vector<TreeNode *> tp;
void inOrder(TreeNode *root) { //使用中序遍历得到有序数组
if(!root)
return;
inOrder(root->left);
tp.push_back(root);
inOrder(root->right);
}
TreeNode* KthNode(TreeNode* pRoot, int k) {
if(!pRoot)
return NULL;
inOrder(pRoot);
for(int i = 0; i < tp.size(); i++) {
if(i == k-1)
return tp[i];
}
return NULL;
}2、递归做法,无返回值的
TreeNode *tp = NULL; //保存最终的结果
void midOrder(TreeNode *pRoot, int &k) { //也是要用引用
if(!pRoot || tp) //tp不为空 直接返回 也是避免无谓
return;
midOrder(pRoot->left, k);
k--;
if(k == 0) {
tp = pRoot;
return;
}
midOrder(pRoot->right, k);
}
TreeNode* KthNode(TreeNode* pRoot, int k) {
if(!pRoot)
return NULL;
midOrder(pRoot, k);
return tp;
}3、递归做法,有返回值的
//也是递归使用中序遍历,当k为0时 找到节点 如果开始没有找到结果 递归返回的都是NULL
//一旦找到结果 就会把这个结果层层返回
TreeNode * getRes(TreeNode *pRoot, int &k) { //主要要用引用 保证递归的过程只能使用
//一个变量 因为每层递归的k值是不同的
if(!pRoot)
return NULL;
TreeNode *left = getRes(pRoot->left, k);
if(left){ //不为空 找到结果 及时返回 可以避免后面的递归
return left;
}
k--;
if(k == 0) {
return pRoot;
}
TreeNode *right = getRes(pRoot->right, k);
if(right){ //不为空 找到结果 及时返回 可以避免后面的递归
return right;
}
return left ? left : right ? right : NULL; //这里主要是返回NULL
}
TreeNode* KthNode(TreeNode* pRoot, int k) {
if(!pRoot)
return NULL;
int tk = k;
return getRes(pRoot, k);
}
小天才公司福利 1159人发布