一维度前缀和解题通用模板(java)

java 复制代码
public class PrefixSum1D {
    public static void main(String[] args) {
        // 原始数组(下标从1开始,方便计算)
        int[] nums = {0, 3, 1, 4, 1, 5, 9, 2, 6}; // 下标0不用,占位填0

        int n = nums.length - 1; // 实际元素个数(去掉占位的0)

        // ① 建立前缀和数组
        int[] pre = new int[n + 1]; // pre[i] 表示 nums[1..i] 的总和
        pre[0] = 0;                 // 前0个元素的和为0,这是边界

        for (int i = 1; i <= n; i++) {
            pre[i] = pre[i - 1] + nums[i]; // 当前前缀和 = 上一个前缀和 + 当前元素
        }

        // ② 查询区间 [l, r] 的和(下标从1开始)
        // 公式:sum(l, r) = pre[r] - pre[l - 1]
        int l = 2, r = 5;
        int rangeSum = pre[r] - pre[l - 1];
        // pre[5]=14, pre[1]=3, 所以区间[2,5]的和 = 14 - 3 = 11
        System.out.println("区间[" + l + "," + r + "]的和 = " + rangeSum); // 11
    }
}

总结一下

需要做的

1.初始化数据(需要开两个数组)

一个用来存数据(手动填数据,最前面填个0占位,开空间的话就用N+1,多开的1也是给占位0用的,否则会越界)

一个用来存前缀和(这个数组要开N+1,通常在for里面下标从1开始)

2.初始化左右边界值(相当于题目要求的区间范围)

看题目要求

例如:

int l = 2;

int r = 5;

2.核心解题公式

pre[i] = pre[i - 1] + nums[i];

在for里面执行,意思大概是:

当前前缀和 = 上一个前缀和 + 当前元素

int rangeSum = pre[r] - pre[l - 1];

无需在for中执行,意思大概是:

目标区间和 = 右边界前缀和 - 左边界上一位前缀和

重点!!!

pre[l-1]这里之所以要-1,是因为要确保左边界值不会被剪掉

反面例子,r=5,l=2,题目要求求[l,r] = [2,5]

如果不-1:

(1+2+3+4+5号的和) - (1+2号的和) = 3+4+5号的和

2号直接被吞掉了

相关推荐
无限进步_2 小时前
【C++】重载、重写和重定义的区别详解
c语言·开发语言·c++·ide·windows·git·github
许杰小刀2 小时前
Python网络请求库,从 requests 到 httpx
开发语言·python·httpx
weixin_513449962 小时前
walk_these_ways项目学习记录第十篇(通过行为多样性 (MoB) 实现地形泛化)--从仿真到部署
人工智能·学习·算法
历程里程碑2 小时前
1 . Git本地操作:版本控制 跨平台协作 仓库核心
java·开发语言·数据结构·c++·git·gitee·github
小欣加油2 小时前
leetcode 42 接雨水
c++·算法·leetcode·职场和发展
hekung2 小时前
maven的lifecycle与idea的run
java·maven
tankeven2 小时前
动态规划专题(14):石子合并问题(未完待续)
c++·算法·动态规划
阿维的博客日记2 小时前
为什么 ConcurrentHashMap 采用 synchronized 加锁而不采用ReentrantLock
java·juc
阿丰资源2 小时前
java项目(附资料)-基于SpringBoot+MyBatisPlus+MySQL+Layui的药品管理系统
java·spring boot·mysql