BCCs and 2CCs

Catalan Numbers

DP on Trees - Combining Subtrees

Querying for the minimum and number of occurrences of minimum in a range

Counting Minimums with Segment Tree

Finding nodes that must be visited along any path.

Critical

Dynamic programming on grid boundary states.

DP on Broken Profile

Additional dynamic programming techniques and optimizations like Knuth's optimization.

Additional DP Optimizations and Techniques

Visiting all edges of a graph exactly once.

Eulerian Tours

A formula for finding the number of faces in a planar graph.

Euler's Formula

Extended Euclidean Algorithm

More Complex Operations Using FFT

Introduction to Fast Fourier Transform

Flow with Lower Bounds

A simple solution to "Robotic Cow Herd" that generalizes.

Fracturing Search

Solving games that are usually two-player to find the winner.

Game Theory

Interactive and Communication Problems

Link Cut Tree

Lagrangian Relaxation

LineContainer

Matroid Intersection

Introduces maximum flow as well as flow with lower bounds.

Maximum Flow

Triangle Inequality, Johnson's Algorithm, and Min Cost Flow

Minimum Cost Flow

Minimum Cut

Erasing from non-amortized insert-only data structures.

Offline Deletion

What if data structures could time travel?

Persistent Data Structures

Prefix Sums of Number-Theoretic Functions (Part 1)

Prefix Sums of Number-Theoretic Functions (Part 2)

Subsets of nodes in directed graphs where each node in a subset can reach each other node in the subset.

Strongly Connected Components

Returning to Bellman-Ford and Floyd-Warshall.

Shortest Paths with Negative Edge Weights

Segment Tree Beats

Slope trick refers to a way to manipulate piecewise linear convex functions. Includes a simple solution to USACO Landscaping.

Slope Trick

Max Suffix Query with Insertions Only

Knuth-Morris-Pratt and Z Algorithms (and a few more related topics).

String Searching

Suffix Automata, Suffix Trees, and Palindromic Trees

String Suffix Structures

Quickly sorting suffixes of a string and its applications

Suffix Array

Treaps

Vectorization in C++

Efficiently querying for the kth minimum element in a subarray.

Wavelet Tree

XOR Basis

Using randomized algorithms to solve problems.

Randomness

Extending Range Queries to 2D (and beyond).

2D Range Queries

Binary Jumping

Examples of how bitsets give some unintended solutions on recent USACO problems.

(Optional) Bitsets

Decomposing a tree to facilitate path computations.

Centroid Decomposition

Final tips for Platinum and additional practice problems.

Additional Practice for USACO Platinum

Smallest convex polygon containing a set of points on a grid.

Convex Hull

A way to find the maximum or minimum value of several convex functions at given points.

Convex Hull Trick

Using Divide & Conquer as a DP Optimization.

Divide & Conquer - DP

Using Divide & Conquer to answer offline or online range queries on a static array.

Divide & Conquer - SRQ

Solving subset sum problems efficiently with DP.

Sum over Subsets DP

Geometry Primitives

Heavy-Light Decomposition

Repeatedly multiplying a square matrix by itself.

Matrix Exponentiation

Small-To-Large Merging

The inclusion-exclusion principle is a counting technique that generalizes the formula for computing the size of union of n finite sets.

Inclusion-Exclusion Principle

Lazy updates on segment trees and two binary indexed trees in conjunction.

Range Update Range Query

Solving 2D grid problems using 1D range queries.

Range Queries with Sweep Line

Walking on a Segment Tree, Non-Commutative Combiner Functions

More Applications of Segment Tree

Sparse Segment Trees

Splitting up data into smaller chunks to speed up processing.

Square Root Decomposition

Sweep Line

Compressing a tree to only the necessary nodes.

Virtual Tree

Decomposing Kruskal's algorithm to solve problems about minimum/maximum edge weights.

Kruskal Reconstruction Tree

DP on Trees - Solving For All Roots

Combinatorics

Final tips for Gold and additional practice problems.

Additional Practice for USACO Gold

Incorporating custom comparators into standard library containers.

(Optional) Sorted Sets with Custom Comparators

DP problems that require iterating over subsets.

Bitmask DP

Solving a DP problem on every contiguous subarray of the original array.

Range DP

DP on Trees - Introduction

The Disjoint Set Union (DSU) data structure, which allows you to add edges to a graph and test whether two vertices of the graph are connected.

Disjoint Set Union

Finding the number of integers in a range that have a property.

Digit DP

Using the information that one integer evenly divides another.

Divisibility

Quickly testing equality of substrings or sets with a small probability of failure.

Hashing

Maintaining collections of distinct elements with hashing.

(Optional) Hashmaps

Speeding up naive recursive solutions with memoization.

Introduction to DP

Finding the next element smaller or larger than a specified key in a sorted set, and using iterators with sorted sets.

More Operations on Sorted Sets

Problems that can be modeled as filling a limited-size container with items.

Knapsack DP

Finding and using the longest increasing subsequence of an array.

Longest Increasing Subsequence

Finding a subset of the edges of a connected, undirected, edge-weighted graph that connects all the vertices to each other of minimum total weight.

Minimum Spanning Trees

Problems involving dividing the search space into two.

Meet In The Middle

Modular Arithmetic

Segment Tree, Binary Indexed Tree, and Order Statistic Tree (in C++).

Point Update Range Sum

Counting the number of "special" paths on a grid, and how some string problems can be solved using grids.

Paths on Grids

Bellman-Ford, Floyd-Warshall, and Dijkstra.

Shortest Paths with Non-Negative Edge Weights

Maintaining data over consecutive subarrays.

Sliding Window

A data structure that only allows insertion and deletion at one end.

Stacks

Using ternary or binary search to find the mode of unimodal functions.

Optimizing Unimodal Functions

Ordering the vertices of a directed acyclic graph such that each vertex is visited before its children.

Topological Sort

Flattening a tree into an array to easily query and update subtrees.

Euler Tour Technique

Introduces how BFS can be used to find shortest paths in unweighted graphs.

Shortest Paths with Unweighted Edges

Binary searching on arbitrary monotonic functions.

Binary Search

Efficiently searching for a value in a sorted array.

Binary Search on a Sorted Array

Final tips for Silver and additional practice problems.

Additional Practice for USACO Silver

Finding connected components in a graph represented by a grid.

Flood Fill

Directed graphs in which every vertex has exactly one outgoing edge.

Introduction to Functional Graphs

Traversing a graph with depth first search and breadth first search.

Graph Traversal

Solving greedy problems by sorting the input.

Greedy Algorithms with Sorting

Six bitwise operators and the common ways they are used.

Intro to Bitwise Operators

Introducing a special type of graph: trees.

Introduction to Tree Algorithms

Introduction to Prefix Sums

Computing range sum queries in constant time over a fixed 1D array.

Subarray Sums II

"use strict";
const {Fragment: _Fragment, jsx: _jsx, jsxs: _jsxs} = arguments[0];
const frontmatter = undefined;
function _createMdxContent(props) {
 const _components = {
 a: "a",
 code: "code",
 h2: "h2",
 p: "p",
 pre: "pre",
 strong: "strong",
 ...props.components
 }, {CPPSection, HeaderLink, JavaSection, LanguageSection, MATHSPAN, PySection, Warning} = _components;
 if (!CPPSection) _missingMdxReference("CPPSection", true);
 if (!HeaderLink) _missingMdxReference("HeaderLink", true);
 if (!JavaSection) _missingMdxReference("JavaSection", true);
 if (!LanguageSection) _missingMdxReference("LanguageSection", true);
 if (!MATHSPAN) _missingMdxReference("MATHSPAN", true);
 if (!PySection) _missingMdxReference("PySection", true);
 if (!Warning) _missingMdxReference("Warning", true);
 return _jsxs(_Fragment, {
 children: [_jsxs(_components.h2, {
 id: "problem",
 children: [_jsx(_components.a, {
 "aria-hidden": "true",
 tabIndex: "-1",
 className: "anchor before",
 href: "#problem",
 children: _jsx(HeaderLink, {})
 }), "Problem"]
 }), "\n", _jsxs(_components.p, {
 children: ["We are asked to find the number of subarrays that sum up to ", _jsx(MATHSPAN, {
 className: "language-math math-inline",
 latex: "x",
 children: "<math xmlns=\"http://www.w3.org/1998/Math/MathML\"><semantics><mrow><mi>x</mi></mrow><annotation encoding=\"application/x-tex\">x</annotation></semantics></math>x"
 }), " given the size\nof the array and its elements."]
 }), "\n", _jsxs(_components.h2, {
 id: "explanation",
 children: [_jsx(_components.a, {
 "aria-hidden": "true",
 tabIndex: "-1",
 className: "anchor before",
 href: "#explanation",
 children: _jsx(HeaderLink, {})
 }), "Explanation"]
 }), "\n", _jsxs(_components.p, {
 children: ["We can have a map that keeps track of the prefix sums. At each index ", _jsx(MATHSPAN, {
 className: "language-math math-inline",
 latex: "i",
 children: "<math xmlns=\"http://www.w3.org/1998/Math/MathML\"><semantics><mrow><mi>i</mi></mrow><annotation encoding=\"application/x-tex\">i</annotation></semantics></math>i"
 }), ", we can\ncount the number of prefixes with sum equal to ", _jsx(MATHSPAN, {
 className: "language-math math-inline",
 latex: "\\texttt{prefixSum}[i] - x",
 children: "<math xmlns=\"http://www.w3.org/1998/Math/MathML\"><semantics><mrow><mtext mathvariant=\"monospace\">prefixSum</mtext><mo stretchy=\"false\">[</mo><mi>i</mi><mo stretchy=\"false\">]</mo><mo>−</mo><mi>x</mi></mrow><annotation encoding=\"application/x-tex\">\\texttt{prefixSum}[i] - x</annotation></semantics></math>prefixSum[i]−x"
 }), ". This\nwill ensure that we can remove a prefix from our current prefix to build a\nsubarray with sum ", _jsx(MATHSPAN, {
 className: "language-math math-inline",
 latex: "x",
 children: "<math xmlns=\"http://www.w3.org/1998/Math/MathML\"><semantics><mrow><mi>x</mi></mrow><annotation encoding=\"application/x-tex\">x</annotation></semantics></math>x"
 }), ". After every iteration, we just add our new prefix sum to\nthe map."]
 }), "\n", _jsxs(_components.h2, {
 id: "implementation",
 children: [_jsx(_components.a, {
 "aria-hidden": "true",
 tabIndex: "-1",
 className: "anchor before",
 href: "#implementation",
 children: _jsx(HeaderLink, {})
 }), "Implementation"]
 }), "\n", _jsxs(LanguageSection, {
 children: [_jsxs(CPPSection, {
 children: [_jsxs(_components.p, {
 children: [_jsx(_components.strong, {
 children: "Time Complexity:"
 }), " ", _jsx(MATHSPAN, {
 className: "language-math math-inline",
 latex: "\\mathcal{O}(N\\log N)",
 children: "<math xmlns=\"http://www.w3.org/1998/Math/MathML\"><semantics><mrow><mi mathvariant=\"script\">O</mi><mo stretchy=\"false\">(</mo><mi>N</mi><mi>log</mi><mo>⁡</mo><mi>N</mi><mo stretchy=\"false\">)</mo></mrow><annotation encoding=\"application/x-tex\">\\mathcal{O}(N\\log N)</annotation></semantics></math>O(NlogN)"
 })]
 }), _jsxs(_components.p, {
 children: ["In C++, because ", _jsx(_components.code, {
 children: "std::unordered_map"
 }), " is vulnerable to collisions,\nwe use ", _jsx(_components.code, {
 children: "std::map"
 }), " at the cost of adding an extra log factor to the complexity."]
 }), _jsx(_components.pre, {
 children: _jsx(_components.code, {
 className: "language-cpp",
 children: "#include <iostream>\n#include <map>\n#include <vector>\n\nusing namespace std;\n\nint main() {\n\tios_base::sync_with_stdio(0);\n\tcin.tie(0);\n\n\tint N, X;\n\tcin >> N >> X;\n\tvector<int> T(N);\n\tfor (int i = 0; i < N; i++) { cin >> T[i]; }\n\n\tlong long prefix_sum = 0;\n\tlong long ans = 0;\n\tmap<long long, int> sums;\n\tsums[0] = 1;\n\tfor (int x : T) {\n\t\tprefix_sum += x;\n\t\t/*\n\t\t * If there is a subarray with a prefix sum of prefix_sum - X,\n\t\t * we can exclude it from our current subarray to get the desired sum.\n\t\t * Thus, we can add the number of those subarrays to our answer.\n\t\t */\n\t\tans += sums[prefix_sum - X];\n\t\t// Increment the amount of prefix sums with a sum of prefix_sum\n\t\tsums[prefix_sum]++;\n\t}\n\tcout << ans << endl;\n}\n"
 })
 })]
 }), _jsxs(JavaSection, {
 children: [_jsxs(_components.p, {
 children: [_jsx(_components.strong, {
 children: "Time Complexity:"
 }), " ", _jsx(MATHSPAN, {
 className: "language-math math-inline",
 latex: "\\mathcal{O}(N\\log N)",
 children: "<math xmlns=\"http://www.w3.org/1998/Math/MathML\"><semantics><mrow><mi mathvariant=\"script\">O</mi><mo stretchy=\"false\">(</mo><mi>N</mi><mi>log</mi><mo>⁡</mo><mi>N</mi><mo stretchy=\"false\">)</mo></mrow><annotation encoding=\"application/x-tex\">\\mathcal{O}(N\\log N)</annotation></semantics></math>O(NlogN)"
 })]
 }), _jsx(_components.pre, {
 children: _jsx(_components.code, {
 className: "language-java",
 children: "import java.io.*;\nimport java.util.*;\n\npublic class SubarraySumsII {\n\tpublic static void main(String[] args) throws IOException {\n\t\tKattio io = new Kattio();\n\t\tint arraySize = io.nextInt();\n\t\tint target = io.nextInt();\n\t\tint[] array = new int[arraySize];\n\n\t\tfor (int x = 0; x < arraySize; x++) { array[x] = io.nextInt(); }\n\n\t\tlong prefixSum = 0;\n\t\tlong answer = 0;\n\t\tMap<Long, Integer> sums = new HashMap<>();\n\t\tsums.put((long)0, 1);\n\n\t\tfor (int x : array) {\n\t\t\tprefixSum += x;\n\t\t\t/*\n\t\t\t * If there is a subarray with a prefix sum of prefix_sum - X,\n\t\t\t * we can exclude it from our current subarray to get the desired\n\t\t\t * sum. Thus, we can add the number of those subarrays to our\n\t\t\t * answer.\n\t\t\t */\n\t\t\tif (sums.containsKey(prefixSum - target)) { // check if it is in our map\n\t\t\t\tanswer += sums.get(prefixSum - target);\n\t\t\t}\n\n\t\t\t// Increment the amount of prefix sums with a sum of prefix_sum\n\t\t\tif (!sums.containsKey(prefixSum)) { // not yet in map, so add it\n\t\t\t\tsums.put(prefixSum, 1);\n\t\t\t} else { // already in map, add one to it\n\t\t\t\tsums.put(prefixSum, sums.get(prefixSum) + 1);\n\t\t\t}\n\t\t}\n\t\tio.println(answer);\n\t\tio.close();\n\t}\n\t//BeginCodeSnip{Kattio}\n\tstatic class Kattio extends PrintWriter {\n\t\tprivate BufferedReader r;\n\t\tprivate StringTokenizer st;\n\t\t// standard input\n\t\tpublic Kattio() { this(System.in, System.out); }\n\t\tpublic Kattio(InputStream i, OutputStream o) {\n\t\t\tsuper(o);\n\t\t\tr = new BufferedReader(new InputStreamReader(i));\n\t\t}\n\t\t// USACO-style file input\n\t\tpublic Kattio(String problemName) throws IOException {\n\t\t\tsuper(problemName + \".out\");\n\t\t\tr = new BufferedReader(new FileReader(problemName + \".in\"));\n\t\t}\n\t\t// returns null if no more input\n\t\tpublic String next() {\n\t\t\ttry {\n\t\t\t\twhile (st == null || !st.hasMoreTokens())\n\t\t\t\t\tst = new StringTokenizer(r.readLine());\n\t\t\t\treturn st.nextToken();\n\t\t\t} catch (Exception e) { }\n\t\t\treturn null;\n\t\t}\n\t\tpublic int nextInt() { return Integer.parseInt(next()); }\n\t\tpublic double nextDouble() { return Double.parseDouble(next()); }\n\t\tpublic long nextLong() { return Long.parseLong(next()); }\n\t}\n\t//EndCodeSnip\n}\n"
 })
 })]
 }), _jsxs(PySection, {
 children: [_jsxs(_components.p, {
 children: [_jsx(_components.strong, {
 children: "Time Complexity:"
 }), " ", _jsx(MATHSPAN, {
 className: "language-math math-inline",
 latex: "\\mathcal{O}(N)",
 children: "<math xmlns=\"http://www.w3.org/1998/Math/MathML\"><semantics><mrow><mi mathvariant=\"script\">O</mi><mo stretchy=\"false\">(</mo><mi>N</mi><mo stretchy=\"false\">)</mo></mrow><annotation encoding=\"application/x-tex\">\\mathcal{O}(N)</annotation></semantics></math>O(N)"
 }), " (expected)"]
 }), _jsx(Warning, {
 children: _jsxs(_components.p, {
 children: ["Without incorporating randomness, the code below will fail to pass\ncases specifically designed to make Python dict run slowly. See the ", _jsx(_components.a, {
 href: "/bronze/intro-sets",
 children: "Introduction to Sets"
 }), " module for more information."]
 })
 }), _jsx(_components.pre, {
 children: _jsx(_components.code, {
 className: "language-py",
 children: "import random\n\nRANDOM = random.randrange(2**62)\n\n\ndef Wrapper(x):\n\treturn x ^ RANDOM\n\n\ndef main():\n\tN, X = map(int, input().split())\n\tprefix, res = 0, 0\n\tmp = {Wrapper(0): 1} # mp[0] = 1\n\n\tfor x in input().split():\n\t\tprefix += int(x)\n\t\tres += mp.get(Wrapper(prefix - X), 0) # if not in dict, return 0\n\t\tmp[Wrapper(prefix)] = mp.get(Wrapper(prefix), 0) + 1\n\tprint(res)\n"
 })
 })]
 })]
 })]
 });
}
function MDXContent(props = {}) {
 const {wrapper: MDXLayout} = props.components || ({});
 return MDXLayout ? _jsx(MDXLayout, {
 ...props,
 children: _jsx(_createMdxContent, {
 ...props
 })
 }) : _createMdxContent(props);
}
return {
 frontmatter,
 default: MDXContent
};
function _missingMdxReference(id, component) {
 throw new Error("Expected " + (component ? "component" : "object") + " `" + id + "` to be defined: you likely forgot to import, pass, or provide it.");
}

Table of Contents

Table of Contents

Problem

Explanation

Implementation

Join the USACO Forum!