Щоб доповнити відповідь ngoaho91.
Найкращий спосіб вирішити цю проблему - це використовувати структуру даних Сегментного дерева. Це дозволяє відповідати на такі запити в O (log (n)), це означатиме, що загальна складність вашого алгоритму буде O (Q logn), де Q - кількість запитів. Якби ви використовували алгоритм наївності, загальна складність була б O (Q n), яка явно повільніше.
Однак є недолік використання сегментарних дерев. Це займає багато пам’яті, але багато разів ви дбаєте менше про пам’ять, ніж про швидкість.
Я коротко опишу алгоритми, використовувані цим DS:
Дерево сегмента - це лише окремий випадок дерева бінарного пошуку, де кожен вузол містить значення діапазону, якому він призначений. Кореневому вузлу, присвоюється діапазон [0, n]. Лівій дитині присвоюється діапазон [0, (0 + n) / 2], а правому - [(0 + n) / 2 + 1, n]. Таким чином дерево буде побудовано.
Створіть дерево :
/*
A[] -> array of original values
tree[] -> Segment Tree Data Structure.
node -> the node we are actually in: remember left child is 2*node, right child is 2*node+1
a, b -> The limits of the actual array. This is used because we are dealing
with a recursive function.
*/
int tree[SIZE];
void build_tree(vector<int> A, int node, int a, int b) {
if (a == b) { // We get to a simple element
tree[node] = A[a]; // This node stores the only value
}
else {
int leftChild, rightChild, middle;
leftChild = 2*node;
rightChild = 2*node+1; // Or leftChild+1
middle = (a+b) / 2;
build_tree(A, leftChild, a, middle); // Recursively build the tree in the left child
build_tree(A, rightChild, middle+1, b); // Recursively build the tree in the right child
tree[node] = max(tree[leftChild], tree[rightChild]); // The Value of the actual node,
//is the max of both of the children.
}
}
Дерево запитів
int query(int node, int a, int b, int p, int q) {
if (b < p || a > q) // The actual range is outside this range
return -INF; // Return a negative big number. Can you figure out why?
else if (p >= a && b >= q) // Query inside the range
return tree[node];
int l, r, m;
l = 2*node;
r = l+1;
m = (a+b) / 2;
return max(query(l, a, m, p, q), query(r, m+1, b, p, q)); // Return the max of querying both children.
}
Якщо вам потрібно додаткове пояснення, просто дайте мені знати.
BTW, Сегментне дерево також підтримує оновлення одного елемента або діапазону елементів у O (log n)