Menu
CoddyTech

第 k 小的元素

Coddy AVL 树:数据结构系列 #10 课程的第 12 课,共 16 课。

接下来的挑战会使用你刚刚构建好的 AVLTree 类,它会以锁定文件的形式提供给你。每个挑战都会附带一个新的 solution 文件,你要在其中编写一个使用这棵树的函数。

对任何二叉搜索树进行中序遍历(旋转不会改变这一点),都会按排序后的顺序访问各个值。因此,第 k 小的值就是遍历结果中索引为 k - 1 的元素。

challenge icon

挑战

简单

编写一个函数 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 在线编译器