【算法训练】day6前缀和的简单场景（数组总结）

蜡笔小新..

已于 2024-10-04 11:08:31 修改

阅读量338

点赞数 5

分类专栏： # 算法题解文章标签：算法数据结构

于 2024-09-12 02:00:00 首次发布

本文链接：https://blog.csdn.net/nyxdsb/article/details/142145927

版权

算法题解专栏收录该内容

18 篇文章 0 订阅

订阅专栏

44. 开发商购买土地

题目描述

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

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

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

注意：区块不可再分。

输入描述

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

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

输出描述

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

输入示例

输出示例

提示信息

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

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 scanner = new Scanner(System.in);
        int n = scanner.nextInt();
        int m = scanner.nextInt();
        //区间总面积为sum
        int sum = 0;
        //数组存放区块
        int[][] ret = new int[n][m];
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < m; j++) {
                ret[i][j] = scanner.nextInt();
                sum += ret[i][j];
            }
        }

        // 统计横向前缀和
        int[] row = new int[n];
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < m; j++) {
                row[i] += ret[i][j];
            }
        }

        // 统计纵向前缀和
        int[] col = new int[m];
        for (int j = 0; j < m; j++) {
            for (int i = 0; i < n; i++) {
                col[j] += ret[i][j];
            }
        }
        //打印结果result
        int result = Integer.MAX_VALUE;
        //横向切割
        int rowCut = 0;
        for (int i = 0; i < n; i++) {
            rowCut += row[i];
            result = Math.min(result, Math.abs(sum - 2 * rowCut));
        }
        //纵向切割
        int colCut = 0;
        for (int j = 0; j < m; j++) {
            colCut += col[j];
            result = Math.min(result, Math.abs(sum - 2 * colCut));
        }

        System.out.println(result);
        scanner.close();
    }
}