1537. Get the Maximum Score

HardGreedyArrayTwo PointersDynamic Programming

Problem Description

You are presented with two distinct sorted arrays nums1 and nums2. Your task is to find the maximum score you can achieve by traversing these arrays according to a set of rules. A valid path through the arrays is defined by starting at the first element of either nums1 or nums2 and then moving from left to right. If, during your traversal, you encounter a value that exists in both nums1 and nums2, you are allowed to switch to the other array—but only once for each value. The score is the sum of all unique values that you encounter on this path. The goal is to find the maximum possible score for all valid paths. As the final score might be large, you should return the result modulo 10^9 + 7.

Intuition

To solve this problem, one needs to understand that the maximum score is obtained by traversing through the maximum values from both arrays. Since you are allowed to switch between arrays at common elements, you should do so in a way that maximizes the score.

The key is to walk through both arrays in parallel and track two sums: one for nums1 and another for nums2. As long as the numbers in both arrays are distinct, keep adding them to their respective sums. When a common element is met, you have a choice to switch paths. You want to take the path with the greater sum up to this point since that gives you the maximum score. From that common element, reset both sums to the greater sum and continue the process.

By doing this, you effectively collect the maximum values from both arrays and switch paths at the optimal times to ensure you are always on the path with the greatest potential score. When you reach the end of both arrays, the maximum sum at that point will be the maximum score that can be obtained.

Learn more about Greedy, Two Pointers and Dynamic Programming patterns.

Solution Approach

The solution is an elegant application of the two-pointer technique, which is well-suited for problems involving sorted arrays. Because the two arrays are sorted and the problem description allows us to switch paths at common elements, the two-pointer method allows us to efficiently compare elements from both arrays without needing additional space. Here's how the given solution works step by step:

Initialize two pointers, i and j, to start at the beginning of nums1 and nums2, respectively.
Initialize two sums, f and g, to keep track of the running sum for nums1 and nums2. These variables will also help decide when to switch paths.
Use a while loop to continue processing until both pointers have reached the end of their respective arrays (i < m or j < n).
Inside the loop, there are several cases to consider:

a. If pointer i has reached the end of nums1, add the current element in nums2 to sum g, and increment j.

b. If pointer j has reached the end of nums2, add the current element in nums1 to sum f, and increment i.

c. If the current element in nums1 (nums1[i]) is less than the current element in nums2 (nums2[j]), add nums1[i] to sum f and increment i.

d. If the current element in nums1 is greater than the current element in nums2, add nums2[j] to sum g and increment j.

e. If the current elements in both arrays are equal, a common element is encountered. The max of f and g (which represents the best score up to this point from either array) is added to the common element, then this value is assigned to both f and g since switching paths here is allowed and should yield the maximum score. Increment both i and j after this operation.
After the loop ends (both arrays have been fully traversed), the final answer is the maximum of the two sums, f and g, since you could end in either array.
Return this maximum value modulo 10^9 + 7 as per the problem statement requirements to account for a potentially large score.

In terms of data structures, no additional storage is needed apart from a few variables to keep track of the indices and the sums. This solution's space complexity is O(1), only requiring constant space, and the time complexity is O(m+n), where m and n are the lengths of the two input arrays. Because each element in the arrays is examined at most once by each pointer, the algorithm is highly efficient.

This solution is a demonstration of cleverly optimizing the process of path selection in a way that continually maximizes the potential score at each step.

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 take two sorted arrays to illustrate the solution approach:

nums1 = [2, 4, 5, 8, 10]
nums2 = [4, 6, 8, 9]

Following the proposed solution, here's the walkthrough:

We initialize two pointers i = 0 for nums1 and j = 0 for nums2. We also initialize two sums f = 0 and g = 0.
Since nums1[0] < nums2[0], we add nums1[i] (which is 2) to f and increment i. Now, f = 2, g = 0.
Now we compare nums1[i] (which is 4) with nums2[j] (also 4). Since they are equal, we have a common element.

a. We switch to the array with the higher sum, which are equal at this moment (f = g = 2), so the path doesn't change.

b. We add the maximum of f and g to the common value and assign it to both f and g. Thus, f = g = 4 + 2 = 6.

c. Increment both i and j. Now, i = 2, j = 1.
Now, nums1[i] is 5 and nums2[j] is 6. Since 5 < 6, we add nums1[i] to f and increment i. Now, f = 11, g = 6.
We have nums1[i] as 8 and nums2[j] also 8. Another common element encountered.

a. We choose the path with the higher sum, which is f (11 > 6).

b. We add the max of f and g to the common value, which gives f = g = 8 + 11 = 19.

c. Increment both i and j. Now, i = 4, j = 3.
Now nums1[i] is 10 and nums2[j] is 9. Since 9 < 10, we add nums2[j] to g and increment j. Now, g = 19 + 9 = 28.
We add nums1[i] to f (since j has reached the end of nums2) and increment i. Now, f = 19 + 10 = 29.
Both pointers have reached the end of their arrays. We take the maximum of f and g, which in this case is f = 29.

So, the maximum score that can be achieved is 29, and we will return this value modulo 10^9 + 7 to get the final answer which is 29 itself since it's much less than the modulo value.

Solution Implementation

1from typing import List
2
3class Solution:
4    def maxSum(self, nums1: List[int], nums2: List[int]) -> int:
5        # Initialize variables
6        MOD = 10**9 + 7  # The modulo value for the result
7        len_nums1, len_nums2 = len(nums1), len(nums2)  # Lengths of the input arrays
8        index_nums1 = index_nums2 = 0  # Index pointers for nums1 and nums2
9        sum_nums1 = sum_nums2 = 0  # Running sums for nums1 and nums2
10
11        # Process both arrays until we reach the end of one of them
12        while index_nums1 < len_nums1 or index_nums2 < len_nums2:
13            if index_nums1 == len_nums1:  # nums1 is exhausted, continue with nums2
14                sum_nums2 += nums2[index_nums2]
15                index_nums2 += 1
16            elif index_nums2 == len_nums2:  # nums2 is exhausted, continue with nums1
17                sum_nums1 += nums1[index_nums1]
18                index_nums1 += 1
19            elif nums1[index_nums1] < nums2[index_nums2]:  # Current element of nums1 is smaller
20                sum_nums1 += nums1[index_nums1]
21                index_nums1 += 1
22            elif nums1[index_nums1] > nums2[index_nums2]:  # Current element of nums2 is smaller
23                sum_nums2 += nums2[index_nums2]
24                index_nums2 += 1
25            else:  # Elements in both arrays are equal
26                # Update both sums to the maximum of the two sums plus the current element
27                sum_nums1 = sum_nums2 = max(sum_nums1, sum_nums2) + nums1[index_nums1]
28                # Move past this common element in both arrays
29                index_nums1 += 1
30                index_nums2 += 1
31
32        # Return the maximum of the two sums modulo MOD
33        return max(sum_nums1, sum_nums2) % MOD
34

1class Solution {
2
3    public int maxSum(int[] nums1, int[] nums2) {
4        final int MODULO = (int) 1e9 + 7; // Define the modulo value as a constant
5        int lengthNums1 = nums1.length; // Length of the first array
6        int lengthNums2 = nums2.length; // Length of the second array
7        int indexNums1 = 0; // Current index in nums1
8        int indexNums2 = 0; // Current index in nums2
9        long sumNums1 = 0; // Running sum of segments from nums1
10        long sumNums2 = 0; // Running sum of segments from nums2
11
12        // While there are elements left to consider in either array
13        while (indexNums1 < lengthNums1 || indexNums2 < lengthNums2) {
14            if (indexNums1 == lengthNums1) {
15                // If nums1 is exhausted, add remaining nums2 elements to sumNums2
16                sumNums2 += nums2[indexNums2++];
17            } else if (indexNums2 == lengthNums2) {
18                // If nums2 is exhausted, add remaining nums1 elements to sumNums1
19                sumNums1 += nums1[indexNums1++];
20            } else if (nums1[indexNums1] < nums2[indexNums2]) {
21                // If current element in nums1 is smaller, add it to sumNums1
22                sumNums1 += nums1[indexNums1++];
23            } else if (nums1[indexNums1] > nums2[indexNums2]) {
24                // If current element in nums2 is smaller, add it to sumNums2
25                sumNums2 += nums2[indexNums2++];
26            } else {
27                // If elements are the same, add the max of the two running sums to both sums
28                // and move forward in both arrays
29                sumNums1 = sumNums2 = Math.max(sumNums1, sumNums2) + nums1[indexNums1];
30                indexNums1++;
31                indexNums2++;
32            }
33        }
34
35        // Calculate max of both sums and apply modulo
36        int result = (int) (Math.max(sumNums1, sumNums2) % MODULO);
37
38        return result; // Return the result
39    }
40}
41

1#include <vector>
2#include <algorithm>
3
4class Solution {
5public:
6    int maxSum(std::vector<int>& nums1, std::vector<int>& nums2) {
7        const int MODULO = 1e9 + 7; // Define the modulo constant.
8        int size1 = nums1.size(), size2 = nums2.size();
9        int index1 = 0, index2 = 0; // Initialize pointers for the two arrays.
10        long long sum1 = 0, sum2 = 0; // Initialize sums to 0 as long long to prevent overflow.
11
12        // Iterate through both arrays simultaneously.
13        while (index1 < size1 || index2 < size2) {
14            if (index1 == size1) {
15                // If nums1 is exhausted, add remaining elements of nums2 to sum2.
16                sum2 += nums2[index2++];
17            } else if (index2 == size2) {
18                // If nums2 is exhausted, add remaining elements of nums1 to sum1.
19                sum1 += nums1[index1++];
20            } else if (nums1[index1] < nums2[index2]) {
21                // If the current element in nums1 is less than in nums2, add it to sum1.
22                sum1 += nums1[index1++];
23            } else if (nums1[index1] > nums2[index2]) {
24                // If the current element in nums2 is less than in nums1, add it to sum2.
25                sum2 += nums2[index2++];
26            } else {
27                // When both elements are equal, move to the next elements and add the maximum
28                // of sum1 and sum2 to both sums. 
29                sum1 = sum2 = std::max(sum1, sum2) + nums1[index1];
30                index1++;
31                index2++;
32            }
33        }
34
35        // Calculate the maximum sum encountered, modulo the defined number.
36        return std::max(sum1, sum2) % MODULO;
37    }
38};
39

1function maxSum(nums1: number[], nums2: number[]): number {
2    // Define the modulo value as per the problem's constraint to avoid overflow. 
3    const mod = 1e9 + 7;
4
5    // Store the lengths of the input arrays.
6    const nums1Length = nums1.length;
7    const nums2Length = nums2.length;
8
9    // Initialize two variables to keep track of the maximum sum path for each of the arrays.
10    let nums1MaxSum = 0;
11    let nums2MaxSum = 0;
12
13    // Initialize two pointers for traversing through the arrays.
14    let nums1Index = 0;
15    let nums2Index = 0;
16
17    // Iterate over the arrays until the end of both is reached.
18    while (nums1Index < nums1Length || nums2Index < nums2Length) {
19        if (nums1Index === nums1Length) {
20            // If nums1 is exhausted, continue adding the elements from nums2 to nums2's max sum.
21            nums2MaxSum += nums2[nums2Index++];
22        } else if (nums2Index === nums2Length) {
23            // If nums2 is exhausted, continue adding the elements from nums1 to nums1's max sum.
24            nums1MaxSum += nums1[nums1Index++];
25        } else if (nums1[nums1Index] < nums2[nums2Index]) {
26            // If the current element in nums1 is less than in nums2,
27            // add it to the max sum path of nums1.
28            nums1MaxSum += nums1[nums1Index++];
29        } else if (nums1[nums1Index] > nums2[nums2Index]) {
30            // If the current element in nums2 is less than in nums1,
31            // add it to the max sum path of nums2.
32            nums2MaxSum += nums2[nums2Index++];
33        } else {
34            // If the elements are equal, add the maximum of the two paths plus the current element,
35            // and increment both pointers as the path merges.
36            const maxOfBoth = Math.max(nums1MaxSum, nums2MaxSum) + nums1[nums1Index];
37            nums1MaxSum = maxOfBoth;
38            nums2MaxSum = maxOfBoth;
39            nums1Index++;
40            nums2Index++;
41        }
42    }
43
44    // Return the maximum sum path lesser array modulo the defined mod value.
45    return Math.max(nums1MaxSum, nums2MaxSum) % mod;
46}
47

Time and Space Complexity

The given code aims to find the maximum sum of two non-decreasing arrays where the sum includes each number exactly once, and where an element present in both arrays can only be counted in one of the arrays at any intersection point.

Time Complexity

The time complexity of the code is O(m + n), where m is the length of nums1 and n is the length of nums2. This is because the algorithm uses two pointers i and j to traverse both arrays simultaneously. In the worst case, each pointer goes through the entirety of its corresponding array, resulting in a complete traversal of both arrays. Since the traversal involves constant-time checks and updates, no element is visited more than once, and there are no nested loops, the time complexity is linear with respect to the total number of elements in both arrays.

Space Complexity

The space complexity of the code is O(1). Aside from the input arrays nums1 and nums2, we only use a constant amount of extra space for the variables i, j, f, g, and mod. No additional space is allocated that is dependent on the input size, hence the constant space complexity.

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

Which of the tree traversal order can be used to obtain elements in a binary search tree in sorted order?

Pre-order traversal

Post-order traversal

In-order traversal

Level-order traversal

I don’t know