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

embedded/2024/9/24 0:59: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/embedded/115827.html

相关文章

c语言中“二维数组传参”的本质

void print (int(*arr)[5] , int r,int c) { for(int i0;;i<j;i){ for(int j0;j<c;j){ pritnf("%d",arr[i][j]); } pritnf("\n"); } int main() { int arr[3][5]{{1,2,3,4,5},{2,3,4,5,6}{3,4,5,6,7}}; print (arr,3,5); //将数组内容打印…

计算机毕业设计选题推荐-基于python+Django的全屋家具定制服务平台

精彩专栏推荐订阅&#xff1a;在下方主页&#x1f447;&#x1f3fb;&#x1f447;&#x1f3fb;&#x1f447;&#x1f3fb;&#x1f447;&#x1f3fb; &#x1f496;&#x1f525;作者主页&#xff1a;计算机毕设木哥&#x1f525; &#x1f496; 文章目录 一、全屋家具定制…

python mysql pymysql 数据库操作,常用脚本,个人小工具

起因&#xff0c; 目的: 整理 mysql 工具 启动数据库 检查服务器是否启动了: Get-Service -Name ‘mysql*’ 如果没启动的话&#xff0c;那么就启动: net start MySQL80 (最好是开启管理员权限) 1, 日常最常用的&#xff0c;创建连接 --> 查看所有数据库 —> 查看所有…

Kotlin高阶函数func

Kotlin高阶函数func fun sum(a: Int, b: Int, someFunc: () -> Unit) {println("${a b}")someFunc() }fun myFunc() {println("计算成功") }fun main() {sum(1, 2, ::myFunc) } 输出&#xff1a; 3 计算成功 Kotlin函数作为参数指向不同逻辑_ketlin 将…

即插即用!高德西交的PriorDrive:统一的矢量先验地图编码,辅助无图自动驾驶

Driving with Prior Maps: Unified Vector Prior Encoding for Autonomous Vehicle Mapping 论文主页&#xff1a;https://misstl.github.io/PriorDrive.github.io/ 论文链接&#xff1a;https://arxiv.org/pdf/2409.05352 代码链接&#xff1a;https://github.com/missTL/Pr…

Llamaindex 使用过程中的常见问题 (FAQ)

导读 在使用LlamaIndex进行文档索引和查询时&#xff0c;您可能会发现需要根据特定的需求对基础设置进行调整。下面是一些常见的定制化需求及其对应的实现方式&#xff1a; 文档分割&#xff1a;为了更好地管理和查询大型文档&#xff0c;您可以选择将文档分割成更小的块。这可…

华为HarmonyOS地图服务 1 -- 如何实现地图呈现?

如何使用地图组件MapComponent和MapComponentController呈现地图&#xff0c;效果如下图所示。 MapComponent是地图组件&#xff0c;用于在您的页面中放置地图。MapComponentController是地图组件的主要功能入口类&#xff0c;用来操作地图&#xff0c;与地图有关的所有方法从此…

基于STM32的温度、电流、电压检测proteus仿真系统(OLED、DHT11、继电器、电机)

目录 一、主要功能 二、硬件资源 三、程序编程 四、实现现象 一、主要功能 基于STM32F103C8T6 采用DHT11读取温度、滑动变阻器模拟读取电流、电压。 通过OLED屏幕显示,设置电流阈值为80,电流小阈值为50,电压阈值为60,温度阈值为30 随便哪个超过预祝,则继电器切断,LE…