算法day02: 前缀和

58. 区间和

题目描述

给定一个整数数组 Array，请计算该数组在每个指定区间内元素的总和。

输入描述

第一行输入为整数数组 Array 的长度 n，接下来 n 行，每行一个整数，表示数组的元素。随后的输入为需要计算总和的区间下标：a，b （b > = a），直至文件结束。

输出描述

输出每个指定区间内元素的总和。

输入示例

5
1
2
3
4
5
0 1
1 3

输出示例

3
9

提示信息

数据范围：
0 < n <= 100000

思路

1.  这里我们设置一个数组nums用来接收控制台输入。

2. 此外同时设置一个数组sums记录nums前i位的总和。

3. 假如我们要求下标2 到 下标4的和，只需要求出sums[4] - sum[1]的值即可。

代码实现

import java.util.*;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int[] nums = new int[n];
        int[] sums = new int[n];
        int result = 0;
        int count = 0;
        for(int i = 0; i < n; i++) {
            nums[i] = sc.nextInt();
            count += nums[i];
            sums[i] = count;
        }
        while(sc.hasNextInt()) {
            int a = sc.nextInt();
            int b = sc.nextInt();
            if(a == 0) {
                result = sums[b];
            } else {
                result = sums[b] - sums[a - 1];
            }
            System.out.println(result);
        }
    }
}

44. 开发商购买土地

44. 开发商购买土地（第五期模拟笔试）

题目描述

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

思路：

这道题目和区间和其实有相似之处，由于只能横着分或者竖着分，所以我们可以求纵向的和与横向的和，最后比较最小差距。

1.  获取面积总和sum, 在给nums数组录入整数以后就可以得到。

2. 分别求出横向和纵向的各个分法的价值总和（拿纵向的举例）每一次for循环求一次和，比较一次差值 这里colcounts = colcount[0]是第一次for循环，colcounts = colcount[0] + colcount[1]是第二次for循环。

这里先求出横向和竖向的和以便于后续的求差值。

//横向求和。
        int[] rowcount = new int[nums.length];
        for(int i = 0; i < rowcount.length; i++) {
            for(int j = 0; j < nums[0].length; j++) {
                rowcount[i] += nums[i][j];
            }
        }

        //纵向求和。
        int[] colcount = new int[nums[0].length];
        for(int i = 0; i < colcount.length; i++) {
            for(int j = 0; j < nums.length; j++) {
                colcount[i] += nums[j][i];
            }
        }

3. 然后比较差值，这里的 差值是sum - colcounts - colcount 即 sum - 2 * colcounts。

于是有：

 //进行比较，取总价值的最小差值。
        int result = Integer.MAX_VALUE;
        int rowcounts = 0;
        for(int i = 0; i < rowcount.length; i++) {
            rowcounts += rowcount[i];
            result = Math.min(result, Math.abs(sum - 2 * rowcounts));
        }
        int colcounts = 0;
        for(int i = 0; i < colcount.length; i++) {
            colcounts += colcount[i];
            result = Math.min(result, Math.abs(sum - 2 * colcounts));
        }
        System.out.println(result);

代码实现

import java.util.*;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);

        int sum = 0;
        int a = sc.nextInt();
        int b = sc.nextInt();
        int[][] nums = new int[a][b];
        //求出土地价值总和。
        for(int i = 0; i < nums.length; i++) {
            for(int j = 0; j < nums[0].length; j++) {
                nums[i][j] = sc.nextInt();
                sum += nums[i][j];
            }
        }
        //横向求和。
        int[] rowcount = new int[nums.length];
        for(int i = 0; i < rowcount.length; i++) {
            for(int j = 0; j < nums[0].length; j++) {
                rowcount[i] += nums[i][j];
            }
        }

        //纵向求和。
        int[] colcount = new int[nums[0].length];
        for(int i = 0; i < colcount.length; i++) {
            for(int j = 0; j < nums.length; j++) {
                colcount[i] += nums[j][i];
            }
        }

        //进行比较，取总价值的最小差值。
        int result = Integer.MAX_VALUE;
        int rowcounts = 0;
        for(int i = 0; i < rowcount.length; i++) {
            rowcounts += rowcount[i];
            result = Math.min(result, Math.abs(sum - 2 * rowcounts));
        }
        int colcounts = 0;
        for(int i = 0; i < colcount.length; i++) {
            colcounts += colcount[i];
            result = Math.min(result, Math.abs(sum - 2 * colcounts));
        }
        System.out.println(result);
    }
}