Suppose that you want to support the following operations on a tree:
Update all nodes along the path from node to node .
Find the sum, maximum, minimum (or any other operation that satisfies the associative property) along the path from node to node .
Heavy Light Decomposition (or HLD) supports both operations efficiently.
Resources | |||||
---|---|---|---|---|---|
cp-algo | For an alternate implementation, see below | ||||
CF | blog + video for USACO Cowland. Binary jumping isn't necessary though. |
Optional: Tree Queries in O(NQ)
This is why you don't set problems where is intended ...
Tutorial
Definitions
- A heavy child of a node is the child with the largest subtree size rooted at the child.
- A light child of a node is any child that is not a heavy child.
- A heavy edge connects a node to its heavy child.
- A light edge connects a node to any of its light children.
- A heavy path is the path formed by a collection heavy edges.
- A light path is the path formed by a collection light edges.
Properties
Any path from node to node on the tree can pass through at most light edges.
Proof
Since a heavy path can only be broken by a light edge (or else the edge will be a part of the heavy path), we can know that there are at most heavy paths on any path from an arbitrary node to an arbitrary node .
In addition, by using segment trees (or any other RURQ data structure) we can calculate the value of any consecutive interval on any heavy path in time.
Since there are at most heavy paths and light edges, computing the value on the path from node to node will take = time. We can answer queries in time.
Here's an animation of how the algorithm works:
Implementation
Resources | |||||
---|---|---|---|---|---|
CF | |||||
CF | not complete | ||||
Benq | complete implementation following the above two articles with minor modifications |
Below is an example implementation of Heavy Light Decomposition based on the resources above. See the solution below, as well as the solutions for Subtrees & Paths and Query on a tree again!, for examples of how this implementation can be used.
C++
#include <bits/stdc++.h>using namespace std;template <class T, bool VALS_IN_EDGES> class HLD {private:int N, R, tim = 0; // n, root node, timevector<vector<int>> adj;vector<int> par, siz, depth, rt, pos; // parent, size, depth, root, position arraysLazySegtree<T> segtree; // Modify as needed
Path Queries II
Focus Problem – try your best to solve this problem before continuing!
View Internal SolutionExplanation
We can label each edge as either heavy or light, then use a segment tree to keep track of the maximum value in each heavy chain.
Now, to change the value at node to , we can just update the value in the segment tree. To query the maximum value in the path from to , we first find the Lowest Common Ancestor. We combine the path from to and the path from to to find our answer.
Implementation
Time Complexity: per query
C++
#include <bits/stdc++.h>using namespace std;Code Snippet: Segment Tree (Click to expand)Code Snippet: HLD (Click to expand)int main() {ios_base::sync_with_stdio(false);cin.tie(0);
Problems
Status | Source | Problem Name | Difficulty | Tags | ||
---|---|---|---|---|---|---|
CSES | Easy | Show TagsLCA | ||||
Gold | Easy | Show TagsHLD, PURS | ||||
SPOJ | Easy | Show TagsHLD | ||||
Gold | Normal | Show TagsHLD | ||||
Platinum | Normal | Show TagsHLD | ||||
HR | Normal | Show TagsHLD, RURQ | ||||
Old Gold | Normal | Show TagsHLD, PURS | ||||
YS | Normal | Show TagsHLD, SegTree | ||||
CF | Hard | Show TagsHLD | ||||
CF | Hard | Show TagsHLD | ||||
TLX | Hard | Show TagsHLD | ||||
JOI | Hard | Show TagsHLD | ||||
JOI | Very Hard | Show TagsHLD |
Warning!
"Grass Planting" isn't submittable on the USACO website. Use this link to submit.
Module Progress:
Join the USACO Forum!
Stuck on a problem, or don't understand a module? Join the USACO Forum and get help from other competitive programmers!