第 k 小的元素
Coddy AVL 树:数据结构系列 #10 课程的第 12 课,共 16 课。
接下来的挑战会使用你刚刚构建好的 AVLTree 类,它会以锁定文件的形式提供给你。每个挑战都会附带一个新的 solution 文件,你要在其中编写一个使用这棵树的函数。
对任何二叉搜索树进行中序遍历(旋转不会改变这一点),都会按排序后的顺序访问各个值。因此,第 k 小的值就是遍历结果中索引为 k - 1 的元素。
挑战
简单编写一个函数 kthSmallest(tree, k),返回树中第 k 小的值(从 1 开始计数)。你可以假设 k 是有效的。
亲手试试
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "avltree.h"
#include "solution.h"
int main(void) {
AVLTree* tree = AVLTree_create();
char line1[4096];
fgets(line1, sizeof(line1), stdin);
char* tok = strtok(line1, " \n");
while (tok != NULL) {
AVLTree_insert(tree, atoi(tok));
tok = strtok(NULL, " \n");
}
char line2[256];
fgets(line2, sizeof(line2), stdin);
int p0 = atoi(strtok(line2, " \n"));
int result = kthSmallest(tree, p0);
printf("%d\n", result);
return 0;
}
AVL 树:数据结构系列 #10中的所有课
自己练习: C 在线编译器