3.6 KiB
id | title | author | prerequisites | ||
---|---|---|---|---|---|
data-structures-gold | Data Structures (Gold) | Michael Cao |
|
More advanced applications of data structures introduced in earlier divisions.
Monotonic Stack
Consider the following problem:
Given an array,
a
, ofN (1 \le N \le 10^5)
integers, for every indexi
, find the rightmost indexj
such thatj < i
anda_i > a_j
.
To solve this problem, let's store a stack of pairs of <value, index>
and iterate over the array from left to right. For some index i
, we will compute ans_i
, the rightmost index for i
, as follows:
- Keep popping the top element off the stack as long as
value \ge a_i
. This is because we know that the pair containingvalue
will never be the solution to any indexj > i
, sincea_i
is less than or equal to thanvalue
and has an index further to the right. - If
value < a_i
, setans[i]
toindex
, because a stack stores the most recently added values first (or in this case, the rightmost ones),index
will contain the rightmost value which is less thana_i
. Then, pop the top element off the stack, becauseindex
can't be the solution for two elements.
The stack we used is called a "monotonic stack" because we keep popping off the top element of the stack which maintains it's monotonicity (the same property needed for algorithms like binary search) because the elements in the stack are increasing.
Further Reading
- "Nearest Smallest Element" from CPH 8
- CP Algorithms (Min Stack)
- Medium
Problems
(add more once codeforces comes up)
Sliding Window
Let's envision a sliding window (or constant size subarray) of size K
moving left to right along an array, a
. For each position of the window, we want to compute some information.
Let's store a std::set
of integers representing the integers inside the window. If the window currently spans the range i \dots j
, we observe that moving the range forward to i+1 \dots j+1
only removes a_i
and adds a_{j+1}
to the window. We can support these two operations and query for the minimum/maximum in the set in O(\log N)
.
To compute the sum in the range, instead of using a set, we can store a variable s
representing the sum. As we move the window forward, we update s
by performing the operations s
-= a_i
and s
+= a_{j+1}
.
Further Reading
- "Sliding Window" from CPH 8
- Read this article to learn about the "min queue" that CPH describes.
- Medium
- G4G
Problems
- Haybale Feast (Gold)
- Direct application of sliding window.
- Train Tracking 2 (Plat)
- Extremely difficult.
(add more once codeforces comes up)