day10（续）

仲夏是荔枝味

已于 2024-05-19 11:28:03 修改

阅读量907

点赞数 7

文章标签：算法

于 2024-05-19 11:27:29 首次发布

本文链接：https://blog.csdn.net/2202_76101487/article/details/139039005

版权

第一题

DP35 【模板】二维前缀和

综上所述：



题目要求在一个二维数组中，给出两个哈坐标（x1，y1），（x2，y2），要求求其中位置的和，如上图所示；

本题我们头发前缀和数组的形式来解决这道题；

步骤一：预处理一个前缀和矩阵

dp[i][j]表示原始数组从（1,1）到（i，j）的前缀和；

下面来分析一下如何求解 dp[i][j]：

dp【i】【j】是从坐标（1,1）到（i，j）的元素的和，所以就是上述图形里面的A+B+C+D;

其中：A+B=dp【i-1】【j】；

A+C = dp【i】【j-1】;

                A =  dp【i-1】【j-1】；

所以dp【i】【j】=（A+B）+（A+C）+D-A=

=dp【i-1】【j】+ dp【i】【j-1】+arr【i】【j】- dp【i-1】【j-1】；

由此我们就可求出每一个dp【i】【j】的值；

步骤二：使用前缀和矩阵：

        （x1，y1），（x2，y2）的前缀和矩阵分析如下所示：

如图所示，上述区域被封为A,B,C,D 四个区域；我们需要的数值的d的范围：

故此D= （A+B+C+D） -(A+B)-(A+C)+A

= dp【x2】【y2】- dp【x1-1】【y2】- dp【x2】【y1-1】+ dp【x1-1】【y1-1】;

故此代码如下：
import java.util.Scanner;

// 注意类名必须为 Main, 不要有任何 package xxx 信息
public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        // 注意 hasNext 和 hasNextLine 的区别
        //1、输入数据
        int n = in.nextInt(),m = in.nextInt(),q = in.nextInt();
        int[][] arr = new int[n+1][m+1];
        for(int i =1;i<=n;i++){
            for(int j = 1;j<= m;j++){
                arr[i][j] = in.nextInt();
            }
        }
        //2、预处理一个数组
        long[][] dp = new long[n+1][m+1];
        for(int i =1;i<=n;i++){
            for(int j = 1;j<= m;j++){
                dp[i][j] = dp[i-1][j]+dp[i][j-1]-dp[i-1][j-1]+arr[i][j];
            }
        }
        //3、使用前缀和矩阵
        while(q>0){
            int x1 = in.nextInt();
            int y1 = in.nextInt();
            int x2 = in.nextInt();
            int y2 = in.nextInt();
            System.out.println(dp[x2][y2] - dp[x1-1][y2] - dp[x2][y1-1] + dp[x1-1][y1-1]);
            q--;
        }
    }
}
    

第二题：

724. 寻找数组的中心下标

综上所述：

本题可以采用前缀和数组的思想来解决问题：

如上述图所示，我们来分析前缀和数组和后缀和数组：

f：表示前缀和数组，f【i】表示【0，i-1】的区间的所有元素和；

f【i】= f【i-1】+nums【i】；

g:表示后缀和数组，g【i】表示【i+1，n-1】的区间的所有元素和；

g【i】= g【i+1】+nums【i+1】；

当我们所求的前缀和和后缀和相等时，当前的i元素就是我们所需要的元素；

关于细节要求：

f（0）= g（n-1）= 0，防止指针越界；

代码如下：
class Solution {
    public int pivotIndex(int[] nums) {
        int n = nums.length;
        int[] f = new int[n];
        int[] g = new int[n];
        
        //1、预处理前缀和数组和后缀和数组
        for(int i=1;i<n;i++){
            f[i] = f[i-1]+nums[i-1];
        }
        for(int i=n-2;i>=0;i--){
            g[i] = g[i+1]+nums[i+1];
        }
        //2、使用
        for(int i=0;i<n;i++){
            if(f[i] == g[i]){
                return i;
            }
            
        }
        return -1;
        

    }
}