Update 0044.开发商购买土地.md

problems/kamacoder/0044.开发商购买土地.md

@@ -166,8 +166,45 @@ int main () {
 }
 
 ```
+除了上述做法，还可采用的思路是直接求二维数组的前缀和，时间复杂度为 O(n^2)，
+```c++
+#include <iostream>
+#include <vector>
 
-
+using namespace std;
+int main() {
+    int n, m;
+    scanf("%d%d", &n, &m);
+    // 这里采用n+1,m+1，是为了避免处理边界时潜在的溢出问题
+    vector<vector<int>> val(n+1, vector<int>(m+1, 0));
+    vector<vector<int>> sum(n+1, vector<int>(m+1, 0));
+    for(int i = 1; i <= n; i++) {
+        for(int j = 1; j <= m; j++) {
+            scanf("%d", &val[i][j]);
+
+            // 代码的核心，初始化前缀和数组
+            sum[i][j] = val[i][j] + sum[i-1][j] + sum[i][j-1] - sum[i-1][j-1];
+        }
+    }
+
+    int min = INT32_MAX;
+    //按行进行划分求最小值
+    for(int i = 1; i <= n; i++) {
+        int sub = sum[n][m] - 2 * sum[i][m];
+        sub = sub > 0 ? sub : (-sub);
+        min = min < sub ? min : sub;
+    }
+    //按列进行划分求最小值
+    for(int j = 1; j <= m; j++) {
+        int sub = sum[n][m] - 2 * sum[n][j];
+        sub = sub > 0 ? sub : (-sub);
+        min = min < sub ? min : sub;
+    }
+
+    printf("%d", min);
+    return 0;
+}
+```
 
 ## 其他语言版本