基础算法(4)——前缀和

news/2024/9/23 7:24:45/

1. 前缀和

题目描述:

解法一:暴力解法

直接模拟实现题目流程即可

时间复杂度为O(N*q),根据题目给出的条件q <= 10^{5},肯定会超时

解法二:前缀和(适用题型:快速 求出数组中某一个 连续区间 的 

时间复杂度:O(q)+O(n)

算法思路:

细节问题:为什么数组下标从 1 开始

当数组下标从 0 开始,dp[l - 1] 就成了 dp[-1] 造成数组越界异常

因此数组下标从 1 开始,若是数组下标从 0 开始时,需要处理边界情况

代码实现:

import java.util.Scanner;// 注意类名必须为 Main, 不要有任何 package xxx 信息
public class Main {public static void main(String[] args) {Scanner in = new Scanner(System.in);//1.读入数据int n = in.nextInt();int q = in.nextInt();int[] arr = new int[n + 1];for (int i = 1; i < n + 1; i++) {arr[i] = in.nextInt();}//2.预处理一个前缀和数组long[] dp = new long[n + 1]; //防溢出dp[1] = arr[1];for (int j = 2; j < n + 1; j++) {dp[j] += dp[j - 1] + arr[j];}//3.使用前缀和数组while (q > 0) {int l = in.nextInt();int r = in.nextInt();System.out.println(dp[r] - dp[l - 1]);q--;}}
}

2. 二维前缀和

题目描述:

解法一:暴力解法(模拟)

时间复杂度:O(n*m*q),会超时

解法二:前缀和

时间复杂度:O(m*n)+O(q)

算法思路:

代码实现:

import java.util.Scanner;// 注意类名必须为 Main, 不要有任何 package xxx 信息
public class Main {public static void main(String[] args) {Scanner in = new Scanner(System.in);//1.读入数据int n = in.nextInt();int m = in.nextInt();int q = in.nextInt();int[][] arr = new int[n + 1][m + 1];for (int i = 1; i < n + 1; i++) {for (int j = 1; j < m + 1; j++) {arr[i][j]= in.nextInt();}}//2.预处理一个前缀和矩阵long[][] dp = new long[n + 1][m + 1];for (int i = 1; i < n + 1; i++) {for (int j = 1; j < m + 1; j++) {dp[i][j] = dp[i - 1][j] + dp[i][j - 1] + arr[i][j] - dp[i - 1][j - 1];}}//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--;}}
}

3. 寻找数组的中心下标

题目描述:

解法:前缀和

算法思路:

从中心下标的定义可知,除中心下标的元素外,该元素左边的【前缀和】等于该元素右边的【后缀和】

因此我们可以先预处理出来两个数组,一个表示前缀和,一个表示后缀和

然后用一个 for 循环枚举可能的中心下标,判断每一个位置的【前缀和】以及【后缀和】,如果二者相等,就返回当前下标

代码实现:

class Solution {public int pivotIndex(int[] nums) {int n = nums.length;//f[i] 表示:[0, i - 1] 区间所有元素的和//g[i] 表示:[i + 1, n - 1] 区间所有元素的和int[] f = new int[n]; //前缀和数组int[] g = new int[n]; //后缀和数组//预处理前缀和后缀和数组(优化前)//这里当i等于0时,f[0]本来就是0,没必要多加判断,直接从1位置开始遍历即可// for (int i = 0; i < n; i++) {//     if (i == 0) {//         f[i] = 0;//     } else {//         f[i] = f[i - 1] + nums[i - 1];//     }// }//当i等于n-1时本来就是0,无需处理,直接从n-2处开始遍历即可// for (int i = n - 1; i >= 0; i--) {//     if (i == n - 1) {//         g[i] = 0;//     } else {//         g[i] = g[i + 1] + nums[i + 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];}//使用前后缀和数组进行判断for (int i = 0; i < n; i++) {if (f[i] == g[i]) {return i;}}return -1;}
}

4. 除自身以外数组的乘积

题目描述:

解法:前缀积

算法思路:

根据题意,对于每一个位置的最终结果 answer[i] 都由两部分组成:

第一部分:nums[0] * nums[1] * nums[2] * ... * nums[i - 1]

第二部分:nums[i + 1] * nums[i + 2] * ... * nums[n - 1]

可以利用前缀和的思想,使用两个数组 f 和 g 分别处理出来两个信息:

代码实现:

class Solution {public int[] productExceptSelf(int[] nums) {int n = nums.length;//f[i] 表示[0, i-1] 区间内所有元素的乘积//g[i] 表示[i+1, n-1] 区间内所有元素的乘积int[] f = new int[n];int[] g = new int[n];int[] answer = new int[n];//细节问题,这两个下标处的值需设为1,设为0的话任何数乘0都得0了f[0] = 1;g[n - 1] = 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];}//处理结果数组for (int i = 0; i < n; i++) {answer[i] = f[i] * g[i];}return answer;}
}

5. 和为 K 的子数组

题目描述:

解法一:暴力枚举

时间复杂度:O(N)

算法思路:

解法二:前缀和

算法思路:

代码实现:

class Solution {public int subarraySum(int[] nums, int k) {Map<Integer, Integer> hash = new HashMap<Integer, Integer>();hash.put(0, 1); //对应整个数组元素和为 k 的情况int sum = 0; //记录上一个位置的元素和int ret = 0; //记录前缀和为 sum - k 出现的次数for (int x : nums) {sum += x; //计算当前位置的前缀和ret += hash.getOrDefault(sum - k, 0); //统计结果hash.put(sum, hash.getOrDefault(sum, 0) + 1); //把当前的前缀和放入哈希表}return ret;}
}

6. 和可被 K 整除的子数组

题目描述:

知识补充:

算法思路:

代码实现:

class Solution {public int subarraysDivByK(int[] nums, int k) {Map<Integer, Integer> hash = new HashMap<Integer, Integer>();hash.put(0 % k, 1);int sum = 0; //标记前一个位置的前缀和int ret = 0; //表示最终结果for (int x : nums) {sum += x; //计算当前位置的前缀和int r = (sum % k + k) % k;ret += hash.getOrDefault(r, 0); //统计结果hash.put(r, hash.getOrDefault(r, 0) + 1);}return ret;}
}

7. 连续数组

题目描述:

解法:前缀和 + 哈希表

代码实现:

class Solution {public int findMaxLength(int[] nums) {Map<Integer, Integer> hash = new HashMap<>();hash.put(0, -1); //默认存在一个前缀和为 0 的情况int sum = 0;int ret = 0;for (int i = 0; i < nums.length; i++) { //此处要求下标,所以不能用 foreach//计算当前位置前缀和//无需修改原数组,仅需判断当前位置为 0 时,加上 -1 即可sum += (nums[i] == 0 ? -1 : 1);if (hash.containsKey(sum)) { //判断哈希表中是否存在前缀和ret = Math.max(ret, i - hash.get(sum));} else { //若哈希表中不存在前缀和,则更新hash.put(sum, i);}}return ret;}
}

实例:

8. 矩阵区域和

题目描述:

题目解析:

解法:二维前缀和

二维前缀和递推公式推导:

使用前缀和

算法思路:

此处将 answer 简写为 ans

第一步:当我们要求 ans[i][j] 时,仅需知道其对应的 x1、y1、x2、y2

第二步:下标的映射关系

代码实现:

class Solution {public int[][] matrixBlockSum(int[][] mat, int k) {int m = mat.length;int n = mat[0].length;//1.预处理前缀和矩阵int[][] dp = new int[m + 1][n + 1]; //此处 +1 操作就是为 dp 表增加一行一列for (int i = 1; i <= m; i++) { //dp 表从(1,1)开始for (int j = 1; j <= n; j++) {dp[i][j] = dp[i - 1][j] + dp[i][j - 1] - dp[i - 1][j - 1] + mat[i - 1][j - 1];}}//2.使用int[][] ret = new int[m][n];for (int i = 0; i < m; i++) {for (int j = 0; j < n; j++) {int x1 = Math.max(0, i - k) + 1;int y1 = Math.max(0, j - k) + 1;int x2 = Math.min(m - 1, i + k) + 1;int y2 = Math.min(n - 1, j + k) + 1;ret[i][j] = dp[x2][y2] - dp[x1 - 1][y2] - dp[x2][y1 - 1] + dp[x1 - 1][y1 - 1];}}return ret;}
}

http://www.ppmy.cn/news/1529207.html

相关文章

文件上传-php

查找方式 ***(1) 黑盒 查找(upload) 扫描 (2) 应用型 窗口 上传中心或者后台中心 上传 Ps:后台是后台 权限是权限 (3) 会员中心 (4) 白盒 基本函数定义 写前端的 Enctype 上传类型Method 提交方式Onsubmit 鼠标的时间Action"放在指定文件"Php 接受表单数据 isset(…

Android 新增目录怎么加入git

一服务器创建仓库 一个新的目录增加到仓库中&#xff0c;需要仓库有对应的仓库地址&#xff0c;这个让管理员在服务器段创建仓库地址和对应的文件夹名称&#xff0c; 例如&#xff1a;需求是系统需要增加一个中文输入法&#xff0c;我增加一个输入法的目录&#xff0c;管理员创…

网络安全(黑客技术)2024新版自学手册路线

&#x1f91f; 基于入门网络安全/黑客打造的&#xff1a;&#x1f449;黑客&网络安全入门&进阶学习资源包 前言 什么是网络安全 网络安全可以基于攻击和防御视角来分类&#xff0c;我们经常听到的 “红队”、“渗透测试” 等就是研究攻击技术&#xff0c;而“蓝队”、…

【工具】Java Excel转图片

【工具】Java Excel转图片 package com.yj.luban.modules.office.excel;import org.apache.poi.ss.usermodel.*; import org.apache.poi.xssf.usermodel.XSSFWorkbook;import javax.imageio.ImageIO; import java.awt.Color; import java.awt.Font; import java.awt.*; import …

Android 命令行关机

在 Android 设备上&#xff0c;可以通过以下命令行命令来关机&#xff1a; adb shell reboot -p其中&#xff1a; adb shell&#xff1a;通过 ADB 进入设备的命令行环境。reboot -p&#xff1a;执行关机操作&#xff0c;-p 表示关机而不是重启。 如果你是在设备本地的终端上而…

怎么在路由器上使用tcpdump抓包

怎么在路由器上使用tcpdump抓包 1&#xff09;下载tcpdump包到桌面&#xff0c;路由器界面进入 /tmp目录&#xff0c;/目录无法安装 2&#xff09;开启ftp服务—可以安装IPOP4.1.exe软件&#xff0c;打开服务–ftp列 3&#xff09;连接本机的ftp服务器—ftp 192.168.1.2 &#…

论文阅读:Omni-Kernel Network for Image Restoration

论文地址&#xff1a;https://ojs.aaai.org/index.php/AAAI/article/view/27907 项目地址&#xff1a;https://github.com/c-yn/OKNet 发表时间&#xff1a;2024 图像恢复的目的是从一个退化的低质量的观测中重建一个高质量的图像。最近&#xff0c;Transformer模型由于其强大…

ROS第五梯:ROS+VSCode+C++单步调试

解决问题&#xff1a;在ROS项目中进行断点调试。 第一步&#xff1a;创建一个ROS项目或者打开一个现有的ROS项目。 第二步&#xff1a;修改c_cpp_properties.json 增加一段命令: "compileCommands": "${workspaceFolder}/build/compile_commands.json"第三…