summaryrefslogtreecommitdiff
path: root/progs/a832.py
diff options
context:
space:
mode:
authorSanto Cariotti <santo@dcariotti.me>2024-05-28 10:29:13 +0200
committerSanto Cariotti <santo@dcariotti.me>2024-05-28 10:29:13 +0200
commitf05d888a0b621ca4e99e2b0fb6e23c097006fe41 (patch)
treeeebbb2489144112d3288393e354d19375a0aa088 /progs/a832.py
Init
Diffstat (limited to 'progs/a832.py')
-rw-r--r--progs/a832.py15
1 files changed, 15 insertions, 0 deletions
diff --git a/progs/a832.py b/progs/a832.py
new file mode 100644
index 0000000..5e11874
--- /dev/null
+++ b/progs/a832.py
@@ -0,0 +1,15 @@
+def max_profit(price, k):
+ n = len(price)
+ final_profit = [[None for x in range(n)] for y in range(k + 1)]
+ for i in range(k + 1):
+ for j in range(n):
+ if i == 0 or j == 0:
+ final_profit[i][j] = 0
+ else:
+ max_so_far = 0
+ for x in range(j):
+ curr_price = price[j] - price[x] + final_profit[i-1][x]
+ if max_so_far < curr_price:
+ max_so_far = curr_price
+ final_profit[i][j] = max(final_profit[i][j-1], max_so_far)
+ return final_profit[k][n-1] \ No newline at end of file