How many stretches add up to the figure
An investigator looks through a list of movements for every unbroken stretch that comes to a particular amount.
- Only consecutive movements count as a stretch.
- Stretches may overlap, and every one of them counts separately.
- Movements can be negative, so a longer stretch is not always a bigger total.
- Return how many stretches come to exactly the target.
stretchesTotalling(movements: list<int>, target: int) → int
C++ needs a compiler and Drill does not host one yet, so this page is the reference rather than an exercise: the problem, worked examples, and the solution in full. To type it out, the same problem runs in Python.
Where you start
int stretchesTotalling(std::vector<int> movements, int target) {
}Worked examples
| Call | Result |
|---|---|
stretchesTotalling(std::vector<int>{1, 1, 1}, 2) | 2 |
stretchesTotalling(std::vector<int>{1, 2, 3}, 3) | 2 |
stretchesTotalling(std::vector<int>{1, -1, 0}, 0) | 3 |
stretchesTotalling(std::vector<int>{}, 0) | 0 |
Hint
If the running total at two points differs by the target, the stretch between them is a hit. Keep a count of every running total you have seen and look up total minus target.
Reference solution in C++
int stretchesTotalling(std::vector<int> movements, int target) {
std::map<int, int> seen;
seen[0] = 1;
int running = 0, hits = 0;
for (int movement : movements) {
running += movement;
auto found = seen.find(running - target);
if (found != seen.end()) hits += found->second;
seen[running] += 1;
}
return hits;
}