Skip to content

Commit f80f674

Browse files
Pooja SharmaPooja Sharma
Pooja Sharma
authored and
Pooja Sharma
committed
added pre-commit requirements
1 parent e742e3b commit f80f674

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

dynamic_programming/matrix_chain_multiplication.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@
2929
arr = [40, 20, 30, 10, 30]
3030
output : 26000
3131
"""
32+
import sys
3233

3334

3435
def matrix_chain_multiply(arr: list[int]) -> int:
@@ -57,7 +58,8 @@ def matrix_chain_multiply(arr: list[int]) -> int:
5758
return 0
5859
# initialising 2D dp matrix
5960
n = len(arr)
60-
dp = [[float("inf") for j in range(n)] for i in range(n)]
61+
int_max = sys.maxsize
62+
dp = [[int_max for j in range(n)] for i in range(n)]
6163
# we want minimum cost of multiplication of matrices
6264
# of dimension (i*k) and (k*j). This cost is arr[i-1]*arr[k]*arr[j].
6365
for i in range(n - 1, 0, -1):
@@ -69,6 +71,7 @@ def matrix_chain_multiply(arr: list[int]) -> int:
6971
dp[i][j] = min(
7072
dp[i][j], dp[i][k] + dp[k + 1][j] + arr[i - 1] * arr[k] * arr[j]
7173
)
74+
7275
return dp[1][n - 1]
7376

7477

0 commit comments

Comments
 (0)