45. Jump Game II


Problem Description

In this problem, you're given an array of integers called nums where each element represents the maximum length of a forward jump you can make from that index. Your goal is to find out the minimum number of jumps required to reach the last element of the array, starting from the first element.

Imagine standing on a series of stepping stones stretched across a river, where each stone represents an index in the array and the number on the stone is the maximum number of stones you can skip forward with your next step. You want to cross the river using as few steps as possible, and all the stones are placed so it's guaranteed you can reach the other side.

The constraints are as follows:

  • You are always starting at the first index.
  • You can choose any number of steps to move forward as long as it doesn't exceed the number specified in the current index.
  • You must reach to or beyond the final index of the array.

This is a path-finding problem where we're not only interested in reaching the end but also in minimizing the number of steps we take to get there.

Intuition

The solution to this problem lies in a greedy strategy, which means taking the best immediate step without worrying about the global optimum at each move.

First, understand that since you can always reach the last index, there's no need to worry about getting stuck -- it's only a matter of how quickly you can get there. Also, the farthest you can reach from the current step should always be considered for making a jump, as it gives more options for the next move.

With this in mind, you keep track of two important positions:

  • The farthest you can reach (mx) from the current index or from a group of indexes you consider during a jump.
  • The last position you jumped to (last), which is your starting line for considering where to jump next.

As you progress through the array, you keep updating mx to be the maximum of itself and the farthest you can reach from the current index (i + nums[i]). When you reach the last index, it means you've exhausted all possibilities for the current jump and must make another jump. This is when you set last to mx, because mx is the new maximum range of positions you can jump to from your current position, and increment the number of jumps you've made.

By iterating through the array with this strategy, you can determine the minimum number of jumps needed to reach the end of the array.

Learn more about Greedy and Dynamic Programming patterns.

Solution Approach

The implementation of this solution uses the greedy algorithm approach to determine the minimum number of jumps. The greedy algorithm makes the locally optimal choice at each step with the hope of finding the global optimum, and in this problem, it aims to reach the farthest index possible with each jump, thereby minimizing the total number of jumps.

The following steps and variables are used to implement this strategy:

  • ans is a counter to keep track of the number of jumps made.
  • mx is a variable that holds the maximum distance that can be reached at the current position.
  • last is a variable that keeps track of the farthest index reached with the last jump.

The algorithm works as follows:

  1. Initialize ans, mx, and last to 0. These will store the current number of jumps, the maximum reachable index, and the last jumped-to index, respectively.
  2. Iterate over the array nums except for the last element, because from the second to last element, you are always one jump away from the end:
    • Update mx to be the maximum of itself and the sum of the current index i and the current element's value nums[i]. This effectively sets mx to the farthest reachable index from the current index.
    • Check if the current index i is equal to last. If true, it means you've reached the maximum index from the previous jump, and you have to jump again.
    • When jumping again (the if condition is satisfied):
      • Increment ans because you've made a jump.
      • Update last to mx, as now you will calculate the reachable distance from this new index.

By working through the array with these rules, the ans variable reflects the minimum number of jumps needed when the loop ends. This greedy approach ensures that you make a jump only when necessary, which coincides with reaching the maximum distance from the previous jump.

The final result is obtained by returning the value of ans after the loop has finished executing. Since the greedy algorithm here always takes the farthest reaching option, it guarantees that the number of jumps is minimized.

Here is the solution code with comments corresponding to each step:

class Solution:
    def jump(self, nums: List[int]) -> int:
        ans = mx = last = 0  # Step 1: Initialize variables
        for i, x in enumerate(nums[:-1]):  # Step 2: Iterate over the array except the last element
            mx = max(mx, i + x)  # Update mx to the farthest reachable index
            if last == i:  # Check if we've reached the max distance from the last jump
                ans += 1  # Increment ans to record the jump
                last = mx  # Update last to the new maximum reachable index
        return ans  # Return the minimum number of jumps required

By faithfully following these steps, the code realizes the greedy solution approach and successfully solves the problem of finding the minimum number of jumps.

Ready to land your dream job?

Unlock your dream job with a 2-minute evaluator for a personalized learning plan!

Start Evaluator

Example Walkthrough

Let's consider a simple example to illustrate the solution approach using the greedy algorithm as described:

Suppose we have the array nums = [2, 3, 1, 1, 4]. We want to find the minimum number of jumps to reach the last element.

  • We're starting at the first element (index 0), which is 2. This means that from index 0, we can jump to index 1 or index 2. Our goal is to reach index 4 (the last index) with the minimum number of jumps.

Here's how the greedy algorithm will walk through this example:

  1. Initialize ans (number of jumps), mx (maximum reachable index), and last (last jumped-to index) to 0.

    • ans = 0
    • mx = 0
    • last = 0
  2. Begin iterating from index 0 to the second-to-last index, which is index 3 in this case. For each index i, we will do the following:

  • Index 0:

    • The maximum we can reach from here is 0 + nums[0] = 0 + 2 = 2. So mx is now 2.
    • We haven't reached last yet, so we don't increment ans or update last.
  • Index 1:

    • Here, we can reach 1 + nums[1] = 1 + 3 = 4, which is the last index. Now mx is updated to 4.
    • We've reached last (which is still 0), so it's time to make a jump.
      • Increment ans to 1. (ans = 1)
      • Update last to mx, which is 4. (last = 4)
  • Index 2:

    • We can reach 2 + nums[2] = 2 + 1 = 3, but mx is already 4, so there's no need to update mx.
    • We have not yet reached the new last (which is now 4), so no jumps are made.
  • Index 3:

    • We can reach 3 + nums[3] = 3 + 1 = 4, but again, mx is already 4.
    • We have not yet reached the new last (which is still 4), so no jumps are made.

Using the greedy approach, we've found that we can reach the last index with only 1 jump from index 1 directly to index 4. Therefore, the minimum number of jumps required to reach the last element is 1.

  • The final answer ans is returned, which contains the value 1, indicating the minimum number of jumps needed to reach the end of the array.

The greedy algorithm works optimally in this case by maximizing the reach with every jump and only increasing the jump count when it is necessary.

Solution Implementation

1from typing import List
2
3class Solution:
4    def jump(self, nums: List[int]) -> int:
5        # Initialize the jump count, the maximum reach, and the edge of the current range to 0.
6        jump_count = max_reach = last_reach = 0
7      
8        # Iterate over the list excluding the last element as it's unnecessary to jump from the last position.
9        for index, value in enumerate(nums[:-1]):
10            # Update the maximum reach with the furthest position we can get to from the current index.
11            max_reach = max(max_reach, index + value)
12          
13            # If we have reached the furthest point to which we had jumped previously,
14            # Increment the jump count and update the last reached position to the current max_reach.
15            if last_reach == index:
16                jump_count += 1
17                last_reach = max_reach
18      
19        # Return the minimum number of jumps needed to reach the end of the list.
20        return jump_count
21
1class Solution {
2    public int jump(int[] nums) {
3        int steps = 0;           // Initialize a step counter to keep track of the number of jumps made
4        int maxReach = 0;        // Initialize the maximum reach from the current position
5        int lastJumpMaxReach = 0;  // Initialize the maximum reach of the last jump
6
7        // Iterate through the array except for the last element, 
8        // because we want to reach the last index, not jump beyond it
9        for (int i = 0; i < nums.length - 1; ++i) {
10            // Update the maximum reach by taking the maximum between the current maxReach 
11            // and the position we could reach from the current index (i + nums[i])
12            maxReach = Math.max(maxReach, i + nums[i]);
13          
14            // If the current index reaches the last jump's maximum reach,
15            // it means we have to make another jump to proceed further
16            if (lastJumpMaxReach == i) {
17                steps++;               // Increment the step counter because we're making another jump
18              
19                lastJumpMaxReach = maxReach;  // Update the last jump's max reach to the current maxReach
20              
21                // There's no need to continue if the maximum reach is already beyond the last index,
22                // as we are guaranteed to end the loop
23                if (maxReach >= nums.length - 1) {
24                    break;
25                }
26            }
27        }
28      
29        // Return the minimum number of jumps needed to reach the last index
30        return steps;
31    }
32}
33
1#include <vector>
2using namespace std;
3
4class Solution {
5public:
6    // Function to return the minimum number of jumps to reach the end of the array.
7    int jump(vector<int>& nums) {
8        int numJumps = 0;           // Number of jumps made.
9        int currentMaxReach = 0;    // Max index we can reach with the current number of jumps.
10        int lastMaxReach = 0;       // Max index we can reach from the last jump.
11      
12        // Iterate over each element in the array except the last one.
13        for (int i = 0; i < nums.size() - 1; ++i) {
14            // Update the furthest index we can reach from here.
15            currentMaxReach = max(currentMaxReach, i + nums[i]);
16          
17            // If we're at the last index reached by the last jump,
18            // it's time to make another jump.
19            if (lastMaxReach == i) {
20                numJumps++;
21                lastMaxReach = currentMaxReach;  // Update the max reach for the new jump.
22            }
23        }
24      
25        return numJumps;  // Return the total number of jumps made to reach the end.
26    }
27};
28
1function jump(nums: number[]): number {
2    // Initialize the variables:
3    // jumps: to keep track of the minimum number of jumps needed
4    // maxReach: the max index that can be reached
5    // currentEnd: the last index of the current jump
6    let [jumps, maxReach, currentEnd] = [0, 0, 0];
7
8    // Iterate over the array excluding the last element because
9    // no jump is needed from the last element
10    for (let index = 0; index < nums.length - 1; ++index) {
11        // Update maxReach to be the furthest index reachable from here
12        maxReach = Math.max(maxReach, index + nums[index]);
13
14        // When we reach the currentEnd, it means we've made a jump
15        if (currentEnd === index) {
16            // Increment the jump count
17            ++jumps;
18            // Move the currentEnd to the furthest index reachable
19            currentEnd = maxReach;
20        }
21    }
22
23    // Return the minimum number of jumps needed to reach the end of the array
24    return jumps;
25}
26

Time and Space Complexity

The given Python code implements a greedy algorithm to solve the jump game problem, where you're required to find the minimum number of jumps needed to reach the last index in the array.

The time complexity is O(n) because there is a single for loop that goes through the nums array once. Even though there is a maximum calculation within the loop, this does not change the overall linear time complexity, as it only takes constant time to calculate the maximum of two numbers.

The space complexity is O(1) because the algorithm only uses a fixed number of variables (ans, mx, last) and does not allocate any additional data structures that grow with input size. This means that the space required by the algorithm does not increase with the size of the input array nums.

Learn more about how to find time and space complexity quickly using problem constraints.


Discover Your Strengths and Weaknesses: Take Our 2-Minute Quiz to Tailor Your Study Plan:
Question 1 out of 10

What's the output of running the following function using the following tree as input?

1def serialize(root):
2    res = []
3    def dfs(root):
4        if not root:
5            res.append('x')
6            return
7        res.append(root.val)
8        dfs(root.left)
9        dfs(root.right)
10    dfs(root)
11    return ' '.join(res)
12
1import java.util.StringJoiner;
2
3public static String serialize(Node root) {
4    StringJoiner res = new StringJoiner(" ");
5    serializeDFS(root, res);
6    return res.toString();
7}
8
9private static void serializeDFS(Node root, StringJoiner result) {
10    if (root == null) {
11        result.add("x");
12        return;
13    }
14    result.add(Integer.toString(root.val));
15    serializeDFS(root.left, result);
16    serializeDFS(root.right, result);
17}
18
1function serialize(root) {
2    let res = [];
3    serialize_dfs(root, res);
4    return res.join(" ");
5}
6
7function serialize_dfs(root, res) {
8    if (!root) {
9        res.push("x");
10        return;
11    }
12    res.push(root.val);
13    serialize_dfs(root.left, res);
14    serialize_dfs(root.right, res);
15}
16

Recommended Readings

Want a Structured Path to Master System Design Too? Don’t Miss This!


Load More