Java 一维二维 前缀和问题

前缀和问题

当某些问题需要求得某片数据中某个区间的和，此时我们可以选择暴力求和，也可以选择使用预处理的前缀和求得结果。

问题

子矩阵和：给你一个m×n的整数矩阵，在上面找一个x×y的子矩阵，使子矩阵中所有元素的和最大。

输入格式:

输入数据的第一行为一个正整数T，表示有T组测试数据。每一组测试数据的第一行为四个正整数m,n,x,y（0<m,n<1000 AND 0<x<=m AND 0<y<=n），表示给定的矩形有m行n列。接下来这个矩阵，有m行，每行有n个不大于1000的正整数。

输出格式:

对于每组数据，输出一个整数，表示子矩阵的最大和。

输入样例:

1
4 5 2 2
3 361 649 676 588
992 762 156 993 169
662 34 638 89 543
525 165 254 809 280

输出样例:

2474

思路：一开始看到上述题目，我的第一思路就是模拟xy这样的框遍历这些数据的所有可能，从而求得数据中所有可能的最大求和值，显然这个方法是需要在遍历这个二维数组的同时再次使用两个for遍历小区域的数据来求得和，这样在最坏的情况下时间复杂度为O(n₄)

有一天学长推荐了一个算法，可以更优的解决这样的题，首先介绍一下一维线性数组的前缀和。

一维数组前缀和

当我们有一个一维数组 [ 1、 2、 3、 8、 0、 7、 9、 23 ]，这里让求得 x一点到y一点的区间和，首先需要得到预处理后的前缀和数组，我们需要在原数组的下标0处扩充一位，用于向后的求和操作，我们每次拿到一位数字都与前一位求和放在当前位置上，这样就得到一个预处理前缀和数组，每一项的递推公式如下 F(x，y) = s(i - 1) + s(i)，预处理前缀数组如下[ 0、 1、 3、 6、 14、 14、 21、 30、 53 ]，当我们需要求得x-y区间内和的值时，可以利用坐标轴得到

因为预处理前缀和数组时线性的当前位置与前一位置的数字相加的，所以预处理前缀和也是逐渐递增的，x-y区间和的递推公式 S(x,y) = a(y) - a(x - 1)，这样可以有效的抹去不必要的区间遍历，时间复杂度为O(1)。

import java.util.Scanner;

/**
 * 一维前缀和
 * @author sean
 * @version 1.0
 * @date 2024-3-11 19:49
 */
public class LinePrefix {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int[] arr = new int[scanner.nextInt() + 1];
        for (int i = 1; i < arr.length; i++) {
            arr[i] = arr[i - 1] + scanner.nextInt();
        }
        int x = scanner.nextInt();
        int y = scanner.nextInt();
        System.out.printf("%d - %d 的和是 %d",x,y,arr[y] - arr[x - 1]);
    }
}

结果：
5
1 2 3 4 5
2 4
2 - 4 的和是 9

一维数组的前缀和是不是很好理解呢，那么二维数组的前缀和也好理解，接下来我们理解一下。

二维数组前缀和

问题抛出：一开始的二维数组区间求和问题，我们再来看一下，我们同样可以使用前缀和的预处理形式来进行对问题的解决。

同样的我们也需要在二维数组的x轴和y轴上单独开一个为0的行和列用于预处理操作。在二维数组中任何一个数据的位置向**点（1，1）**靠拢求得这一快区域的和放在当前位置，那么这个区域的和到底应该怎么求呢，我们可以画一个坐标图来解释。

为零的行和列就是为了前缀和的预处理做准备，当前位置的值存放当前位置到（1，1）的区间和。那么当前位置的前缀和怎么求呢？通过图中可以看看到，设当前位置为α，β，加上α-1和β-1处的前缀和，再减去重复的部分α-1，β-1，递推公式S(x,y) = a(i,j) + a(i-1)(j) + a(i)(j-1) - a(i-1)(j-1)。

得到预处理的前缀和数组之后又如何拿到某个区间的和呢？还是画图说明问题。

可以看出假设我们需要求得，浅绿色区间和，我们需要找到区间的上一行（i - 1）且需要是区间的最后一列（j + y - 1）,还需要找到左侧区间最后一个位置的元素数据区间的前一列（j - 1）且需要是区间的最后一行 （i + x - 1），看图就应该知道我们只需要减去这两个区域的前缀和即可，但是我们会注意到左上区间会有重复计算的一部分，很简单找到他的前缀和减去就可以啦坐标为s(i-1)(j-1)。这样我们就可以得到求得某一区间和的递推公式S(x,y) = a(i+x-1)(j+y-1) - a(i-1)(j+y-1) - a(i+x-1)(j-1) + a(i-1)(j-1)。

##问题代码

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.*;

/**
 * @author sean
 * @version 1.0
 * @date 2024-3-9 15:43
 */
public class W13 {
    public static void main(String[] args) throws Exception{
        Scanner scanner = new Scanner(System.in);
        int t = scanner.nextInt();
        List<Integer> res = new ArrayList<>();
        while (t-- != 0){
          	// 所有有可能被覆盖的区间和列表
            int n = scanner.nextInt();
            int m = scanner.nextInt();
            int x = scanner.nextInt();
            int y = scanner.nextInt();
          	// 预处理前缀和数组
            int[][] arr = new int[n + 1][m + 1];
          	// 根据上述公式拿到前缀和数组
            for (int i = 1; i < arr.length; i++) {
                for (int j = 1; j < arr[i].length; j++) {
                    arr[i][j] = arr[i - 1][j] + arr[i][j - 1] - arr[i - 1][j - 1] + scanner.nextInt();
                }
            }
          	// 所有有可能被覆盖区域的前缀和列表
            List<Integer> maxRes = new ArrayList<>();
            // 求所有可能的区间和
            for (int i = 1; i < arr.length - x + 1; i++) {
                for (int j = 1; j < arr[i].length - y + 1; j++) {
                    int sum = arr[i + x - 1][j + y - 1] + arr[i - 1][j - 1]
                            - arr[i - 1][j + y - 1] - arr[i + x - 1][j - 1];
                    maxRes.add(sum);
                }
            }
          	// 降序排
            maxRes.sort((o1, o2) -> o2 - o1);	
          	// 将当前所有区间和的最大和放在结果列表中
            res.add(maxRes.get(0));
        }
      	// 遍历最大区间和列表
        for (int i = 0; i < res.size() ; i++) {
            System.out.println(res.get(i));
        }
    }
}