K-й наименьший элемент

Урок 12 из 16 курса AVL-дерево — Серия «Структуры данных» №10 на Coddy.

Следующие задачи используют готовый класс AVLTree, который вы только что создали, предоставленный вам в виде заблокированного файла. Каждая задача поставляется с новым файлом solution, в котором вы пишете функцию, ИСПОЛЬЗУЮЩУЮ дерево.

Центрированный обход (in-order traversal) любого бинарного дерева поиска (вращения этого не меняют) посещает значения в отсортированном порядке. Таким образом, 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;
}