File tree Expand file tree Collapse file tree 1 file changed +28
-0
lines changed
solution/3500-3599/3548.Equal Sum Grid Partition II Expand file tree Collapse file tree 1 file changed +28
-0
lines changed Original file line number Diff line number Diff line change
1
+ from typing import List
2
+
3
+ class Solution :
4
+ def canPartitionGrid (self , grid : List [List [int ]]) -> bool :
5
+ m , n = len (grid ), len (grid [0 ])
6
+
7
+ total_sum = sum (sum (row ) for row in grid )
8
+
9
+ # Try horizontal cuts
10
+ row_prefix_sum = 0
11
+ for i in range (m - 1 ): # cut between row i and i+1
12
+ row_prefix_sum += sum (grid [i ])
13
+ if row_prefix_sum * 2 == total_sum :
14
+ return True
15
+
16
+ # Try vertical cuts
17
+ col_sums = [0 ] * n
18
+ for i in range (m ):
19
+ for j in range (n ):
20
+ col_sums [j ] += grid [i ][j ]
21
+
22
+ col_prefix_sum = 0
23
+ for j in range (n - 1 ): # cut between column j and j+1
24
+ col_prefix_sum += col_sums [j ]
25
+ if col_prefix_sum * 2 == total_sum :
26
+ return True
27
+
28
+ return False
You can’t perform that action at this time.
0 commit comments