Форум программистов, компьютерный форум CyberForum.ru

Медленное дерево отрезков - C++

Восстановить пароль Регистрация
 
cat4xan
0 / 0 / 1
Регистрация: 30.07.2015
Сообщений: 2
30.07.2015, 09:08     Медленное дерево отрезков #1
Приветствую. Пишу дерево отрезков для задачи нахождения сумм на отрезках. Оно работает, даже вроде правильно, но при массиве 105 и количестве запросов 105 не укладывается в 1 секунду. Подскажите пожалуйста, как его можно ускорить, переписать, дописать. Код может вызвать шок!
C++
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
#include<iostream>
#include<vector>
#include<algorithm>
 
using namespace std;
 
vector<long long> a;
vector<long long> t;
const int neutralV = 0;
 
void build_tree()
{
    for (int i = a.size() - 1; i < 2 * a.size() - 1; i++)
        t[i] = a[i - (a.size() - 1)];
    for (int i = a.size() - 2; i >= 0; i--)
        t[i] = t[(i << 1) + 1] + t[(i << 1) + 2];
}
 
long long rsq(int v, int cl, int cr, int left, int right)
{
    if (left >= right)
        return neutralV;
    if (cl == left && cr == right)
        return t[v];
 
    int cm = (cl + cr) >> 1;
    long long neutral = neutralV;
    neutral += rsq((v << 1) + 1, cl, cm, left, min(cm, right));
    neutral += rsq((v << 1) + 2, cm, cr, max(left, cm), right);
    return neutral;
}
 
int main()
{
    // freopen("input.in", "r", stdin);
    int n;
    cin >> n;
    int old_n = n;
    n = 2 << ((int)(log2(n) + 0.99999999) - 1);
    a.assign(n, neutralV);
    t.assign(2 * n - 1, neutralV);
    for (int i = 0; i < old_n; i++)
    {
        cin >> a[i];
    }
    build_tree();
    int q;
    cin >> q;
    for (int i = 0; i < q; i++)
    {
        int left, right;
        cin >> left >> right;
        left--;
        cout << rsq(0, 0, n, left, right) << " ";
    }
}
Лучшие ответы (1)
Similar
Эксперт
41792 / 34177 / 6122
Регистрация: 12.04.2006
Сообщений: 57,940
30.07.2015, 09:08     Медленное дерево отрезков
Посмотрите здесь:

Напишите программу, которая бы читала дерево в формате (а) и затем печатала бы это дерево в формате (б). C++
Дерево отрезков C++
Дерево отрезков, редактирование куска и поиск суммы C++
C++ На прямой своими концами заданы N отрезков. Найти точку принадлежащую максимальному числу отрезков
Слишком медленное чтение wstring из файла C++
C++ Дерево дерево, странное дерево
Дерево отрезков, проверьте код C++

Искать еще темы с ответами

Или воспользуйтесь поиском по форуму:
После регистрации реклама в сообщениях будет скрыта и будут доступны все возможности форума.
_Ivana
2189 / 1394 / 124
Регистрация: 01.03.2013
Сообщений: 4,151
Записей в блоге: 2
30.07.2015, 14:34     Медленное дерево отрезков #2
Недавно тоже писал олимпиадную задачку на деревья отрезков, правда на Хаскеле. Да, так же добивал нейтральными элементами (для суммы - нулями) до целой степени двойки, но потом то это все надо заполнить именно в бинарное дерево - где у каждого узла будет свое значение и ссылка на левое и правое поддеревья такой же структуры, у терминальных листьев - на воиды. В вашем коде не разбирался, но навскидку не увидел даже типа, позволяющего хранить подобную структуру.
cat4xan
0 / 0 / 1
Регистрация: 30.07.2015
Сообщений: 2
31.07.2015, 07:45  [ТС]     Медленное дерево отрезков #3
Сообщение было отмечено автором темы, экспертом или модератором как ответ
Здесь дерево храниться на обычном массиве: корень дерева - 0, левый сын i вершины - i * 2 + 1, правый сын - i * 2 + 2. Предполагается, что традиционно в дереве отрезков само дерево в явном виде не хранится.

Добавлено через 11 часов 54 минуты
Проблема решилась! Оказалось, что дело не в алгоритме, а в моей упёртости. Всегда ото всех слышал, что надо отключать синхронизацию с stdio, но не придавал этому значения, сейчас ещё раз услышал и решил отключить и вуаля 0,6 сек на макс тесте.
Yandex
Объявления
31.07.2015, 07:45     Медленное дерево отрезков
Ответ Создать тему
Опции темы

Текущее время: 02:44. Часовой пояс GMT +3.
КиберФорум - форум программистов, компьютерный форум, программирование
Powered by vBulletin® Version 3.8.9
Copyright ©2000 - 2016, vBulletin Solutions, Inc.
Рейтинг@Mail.ru