#1043

Partition Array for Maximum Sum

Medium
ArrayDynamic ProgrammingDynamic ProgrammingArray
LeetCode ↗

Approaches

Brute ForceOptimal
Complexity Comparison
Brute ForceOptimal Solution
Time
O(n²)
O(n * k)
Space
O(1)
O(n)
💡

Intuition

Time O(n * k)Space O(n)

The optimal solution uses dynamic programming to build the maximum sum iteratively. We maintain a dp array where dp[i] represents the maximum sum we can achieve using the first i elements of the array.

⚙️

Algorithm

3 steps
  1. 1Step 1: Initialize a dp array of size n+1 with all zeros.
  2. 2Step 2: Iterate through the array from 1 to n.
  3. 3Step 3: For each position i, check the last j elements (up to k), calculate the maximum value in that segment, and update dp[i] accordingly.
solution.py9 lines
1def maxSumAfterPartitioning(arr, k):
2    n = len(arr)
3    dp = [0] * (n + 1)
4    for i in range(1, n + 1):
5        max_val = 0
6        for j in range(1, min(k, i) + 1):
7            max_val = max(max_val, arr[i - j])
8            dp[i] = max(dp[i], dp[i - j] + max_val * j)
9    return dp[n]

Complexity note: The time complexity is O(n * k) because we iterate through each element and for each element, we may check up to k previous elements, leading to a nested loop structure.

  • 1Dynamic programming allows us to build solutions incrementally and efficiently.
  • 2Understanding how to maintain state (like max values) is crucial in optimizing solutions.

Solutions and explanations are original Tejav content. Problem titles © LeetCode — use the LeetCode button above for the full problem statement.