代码随想录算法训练营第三天（2）|44. 开发商购买土地

文档讲解：代码随想录

状态：有点难度

44. 开发商购买土地

本题为代码随想录后续扩充题目，还没有视频讲解，顺便让大家练习一下ACM输入输出模式（笔试面试必备）

题目链接(opens new window)

【题目描述】

在一个城市区域内，被划分成了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。

思路

可以通过前缀和的思路来简化：

在数字输入的过程中求其和(即:sum),

由于只允许将区域按横向或纵向划分成两个子区域，而且每个子区域都必须包含一个或多个区块。所有直接将每行和每列的和计算出来.

之后再进行分块:将第1行认为第一次分配,去计算差,并将其值赋给result,horizontalCut自叠加,再通过Math.abs返回  int 值的绝对值,之后用Math.min取最小的差

verticalCut同理

 Math.abs
返回  int 值的绝对值。如果参数为非负数，则返回该参数。如果参数为负数，则返回该参数的相反数。
注意，如果参数等于 Integer.MIN_VALUE 的值即能够表示的最小负 int 值），那么结果与该值相同且为负。
        
        

1. 计a公司的区域为a,计b公司的区域为b,
2. 可以知道:
3. sum=a+b
4. b=sum-a
5. 差是b-a的绝对值
6. 即[sum-(2*a)]的绝对值

代码如下：

import java.util.Scanner;
public class Main {
    public static void main(String[] args) {
        Scanner scanner=new Scanner(System.in);
        
        int n=scanner.nextInt();
        int m=scanner.nextInt();
        int[][] nums=new int[n][m];
        int sum=0;
        
        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                nums[i][j]=scanner.nextInt();
                sum+=nums[i][j];
            }
        }
        
        //横向
        int[] horizontal = new int[n];
        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                horizontal[i]+=nums[i][j];
            }
        }
        
        //纵向
        int[] vertical = new int[m];
        for(int j=0;j<m;j++){
            for(int i=0;i<n;i++){
                vertical[j]+=nums[i][j];
            }
        }
        
        int result = Integer.MAX_VALUE;
        int horizontalCut = 0;
        for (int i = 0; i < n; i++) {
            horizontalCut += horizontal[i];
            result = Math.min(result, Math.abs(sum - 2 * horizontalCut));
        }
        // Math.abs
        // 返回  int 值的绝对值。
        // 如果参数为非负数，则返回该参数。
        // 如果参数为负数，则返回该参数的相反数。
        // 注意，如果参数等于 Integer.MIN_VALUE 的值
        // （即能够表示的最小负 int 值），那么结果与该值相同且为负。
        
        int verticalCut = 0;
        for(int j=0;j<m;j++){
            verticalCut+= vertical[j];
            result = Math.min(result, Math.abs(sum - 2 * verticalCut));
        }
         System.out.println(result);
    }
}

时间复杂度： O(n^2)

欢迎各位大佬纠正。

如果有问题可以在评论区讨论。

引用：代码随想录