Menu
Coddy logo textTech

Сумма в диапазоне

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

Поскольку дерево сохраняет все значения упорядоченными, вам не нужно проверять каждый узел, чтобы просуммировать диапазон: когда значение узла меньше low, все его левое поддерево также меньше low и его можно пропустить, и та же логика применима к правому поддереву, когда значение узла больше high.

Простого центрированного обхода, который рекурсивно переходит только в те поддеревья, которые могут содержать значения из диапазона, достаточно, хотя для этой задачи полный обход с фильтром тоже подойдет.

challenge icon

Задание

Легко

Напишите функцию rangeSum(tree, low, high), которая возвращает сумму всех значений в дереве, находящихся в диапазоне между low и high включительно.

Попробуйте сами

#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 p1 = atoi(strtok(NULL, " \n"));
    int result = rangeSum(tree, p0, p1);
    printf("%d\n", result);
    return 0;
}

Все уроки раздела AVL-дерево — Серия «Структуры данных» №10