120. Triangle

Approach 1: Storing row sums using DP

class Solution(object):
    def minimumTotal(self, triangle):
        """
        :type triangle: List[List[int]]
        :rtype: int
        """
        let = len(triangle)
        dp = [[0] * (i+1) for i in range(let)]
        dp[0][0] = triangle[0][0]
        counter = 2
        for i in range(1, let):
            for j in range(counter):
                if j == 0:
                    dp[i][j] = triangle[i][j] + dp[i-1][j]
                elif j == counter - 1:
                    dp[i][j] = triangle[i][j] + dp[i-1][j-1]
                else:
                    dp[i][j] = triangle[i][j] + min(dp[i-1][j-1], dp[i-1][j])
            counter += 1
        return min(dp[-1])

As like a standard DP problem, we will store the sums up to the previous row and use it to update the next row. The update condition here being: dp[i][j] = triangle[i][j] + min(dp[i-1][j-1], dp[i-1][j]) where (i,j) are the current co-ordinates of the triangle matrix for which the min sum is being evaluated, and dp[i][j] stores the sums up to row i and location j.

d[element] = element + min(d[left_parent], d[right_parent])

Time Complexity: O(N), where N is the total number of numbers in triangle

Space Complexity: O(N)

Approach 1: Same as above but O(1) space

In the above solution it is visible that the for row N we only need information from row N-1 and so on. This way we can optimize memory as below

class Solution(object):
    def minimumTotal(self, triangle):
        """
        https://leetcode.com/problems/triangle/description/
        :type triangle: List[List[int]]
        :rtype: int
        """
        lastRow = triangle[-1]

        for row in reversed(triangle[:-1]):
            tempRow = row[:]
            for i in range(0,len(row)):
                tempRow[i] += min(lastRow[i], lastRow[i+1])
            lastRow = tempRow
        return lastRow[0]

Time Complexity: O(N), where N is the total number of numbers in triangle

Space Complexity: O(n), where n is the number of rows. N = n(n+1)/2

Last updated