数据结构与算法（开发商购买土地）

题目描述

在一个城市区域内，被划分成了n * m个连续的区块，每个区块都拥有不同的权值，代表着其土地价值。目前，有两家开发公司，A 公司和 B 公司，希望购买这个城市区域的土地。 

现在，需要将这个城市区域的所有区块分配给 A 公司和 B 公司。

然而，由于城市规划的限制，只允许将区域按横向或纵向划分成两个子区域，而且每个子区域都必须包含一个或多个区块。 为了确保公平竞争，你需要找到一种分配方式，使得 A 公司和 B 公司各自的子区域内的土地总价值之差最小。 

注意：区块不可再分。

输入描述

第一行输入两个正整数，代表 n 和 m。 

接下来的 n 行，每行输出 m 个正整数。

输出描述

请输出一个整数，代表两个子区域内土地总价值之间的最小差距。

输入示例

3 3
1 2 3
2 1 3
1 2 3

输出示例

0

提示信息

如果将区域按照如下方式划分：

1 2 | 3
2 1 | 3
1 2 | 3 

两个子区域内土地总价值之间的最小差距可以达到 0。

数据范围：

1 <= n, m <= 100；
n 和 m 不同时为 1。

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner scan=new Scanner(System.in);
        int n=0,m=0;
        n=scan.nextInt();
        m=scan.nextInt();
        int[][] nums=new int[n][m];
        int[] row=new int[n];//每个row[i]代表第i行元素的总和
        int[] col=new int[m];//每个col[i]代表第i列元素的总和
        int rowSum=0;
        int colSum=0;
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < m; j++) {
                nums[i][j]=scan.nextInt();
            }
        }

        for (int i = 0; i < n; i++) {//计算row[i]
            for (int j = 0; j < m; j++) {
                rowSum+=nums[i][j];
            }
            row[i]=rowSum;
        }

        for (int j = 0; j < m; j++) {//计算col[j]
            for (int i = 0; i < n; i++) {
                colSum+=nums[i][j];
            }
            col[j]=colSum;
        }

//对比是按行划分的差距小还是按列划分的差距小
        int rowResult=Main.result(row);
        int colResult=Main.result(col);
        System.out.println(Math.min(rowResult,colResult));
    }

    public static int result(int[] array){
        int total=array[array.length-1];//整个数组的和，array[i]代表前i行/列的和，包括第i行/列
        int res= Integer.MAX_VALUE;
//Math.abs((total-array[i])-array[i])计算两个区域划分后的差值
        for (int i = 0; i < array.length-1; i++) {
            res=Math.min(res, Math.abs((total-array[i])-array[i]));
        }
        return res;
    }
}