Uzupełnienie odpowiedzi ngoaho91.
Najlepszym sposobem rozwiązania tego problemu jest użycie struktury danych drzewa segmentów. To pozwala ci odpowiadać na takie zapytania w O (log (n)), co oznaczałoby, że całkowita złożoność twojego algorytmu wynosiłaby O (Q logn), gdzie Q jest liczbą zapytań. Jeśli użyjesz naiwnego algorytmu, całkowita złożoność wyniesie O (Q n), co jest oczywiście wolniejsze.
Wadą drzew segmentowych jest jednak wada. Zajmuje dużo pamięci, ale często mniej zależy ci na pamięci niż na szybkości.
Pokrótce opiszę algorytmy używane przez ten DS:
Drzewo segmentów jest tylko specjalnym przypadkiem drzewa wyszukiwania binarnego, w którym każdy węzeł przechowuje wartość zakresu, do którego jest przypisany. Węzłu root przypisano zakres [0, n]. Lewemu dziecku przypisano zakres [0, (0 + n) / 2], a prawemu dziecku [(0 + n) / 2 + 1, n]. W ten sposób drzewo zostanie zbudowane.
Utwórz drzewo :
/*
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.
}
}
Drzewo zapytań
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.
}
Jeśli potrzebujesz dodatkowych wyjaśnień, daj mi znać.
BTW, Drzewo segmentów obsługuje także aktualizację pojedynczego elementu lub zakresu elementów w O (log n)