solves Number of Ways of Cutting a Pizza (#1444) in python

Author: sumanth-botlagunta

python/number_of_ways_of_cutting_pizza.py

@@ -0,0 +1,24 @@
+# https://leetcode.com/problems/number-of-ways-of-cutting-a-pizza/
+# T: O(k*r*c*(m+n)) where k is the number of cuts, r and c are the number of rows and columns in the pizza, and m and n are the number of rows and columns in the preSum array.
+# S: O(k*r*c) where k is the number of cuts, and r and c are the number of rows and columns in the pizza.
+
+class Solution:
+    def ways(self, pizza: List[str], K: int) -> int:
+        m, n, MOD = len(pizza), len(pizza[0]), 10 ** 9 + 7
+        preSum = [[0] * (n + 1) for _ in range(m + 1)]
+        for r in range(m - 1, -1, -1):
+            for c in range(n - 1, -1, -1):
+                preSum[r][c] = preSum[r][c + 1] + preSum[r + 1][c] - preSum[r + 1][c + 1] + (pizza[r][c] == 'A')
+        @lru_cache(None)
+        def dp(k, r, c):
+            if preSum[r][c] == 0: return 0
+            if k == 0: return 1
+            ans = 0
+            for nr in range(r + 1, m):
+                if preSum[r][c] - preSum[nr][c] > 0:
+                    ans = (ans + dp(k - 1, nr, c)) % MOD
+            for nc in range(c + 1, n):
+                if preSum[r][c] - preSum[r][nc] > 0:
+                    ans = (ans + dp(k - 1, r, nc)) % MOD
+            return ans
+        return dp(K - 1, 0, 0)