一. 前缀和的简单介绍
1.前缀和的数组型简单定义
对于数组nums来说,前缀和pre[i] = pre[i-1] + nums[i],即每个位置上存储的是前i个数组元素的和,用数学公式表示为:
数组的典型案例:
其他类型的前缀和多是最终能变成和数量相关,即转换成int[] 数组类型。
那么,对于后缀和是一样的道理。
2.前缀和的典型使用场景/触发关键词
在一些题目或者场景中,出现了子数组*,连续一段区间,并且和求数量或者判断数量相关,很大程度上可以向前缀和的方去思考。
二. 使用前缀和的经典案例
1.leetcode1413 逐步求和得到正数的最小值
2.leetcode2383 赢得比赛需要的最少训练时长
3.leetcode2389 和有限的最长子序列
给你一个长度为 n 的整数数组 nums ,和一个长度为 m 的整数数组 queries 。
返回一个长度为 m 的数组 answer ,其中 answer[i] 是 nums 中 元素之和小于等于 queries[i] 的 子序列 的 最大 长度 。
子序列 是由一个数组删除某些元素(也可以不删除)但不改变剩余元素顺序得到的一个数组。
输入:nums = [4,5,2,1], queries = [3,10,21]
输出:[2,3,4]
解释:queries 对应的 answer 如下:
- 子序列 [2,1] 的和小于或等于 3 。可以证明满足题目要求的子序列的最大长度是 2 ,所以 answer[0] = 2 。
- 子序列 [4,5,1] 的和小于或等于 10 。可以证明满足题目要求的子序列的最大长度是 3 ,所以 answer[1] = 3 。
- 子序列 [4,5,2,1] 的和小于或等于 21 。可以证明满足题目要求的子序列的最大长度是 4 ,所以 answer[2] = 4 。
class Solution {
public int[] answerQueries(int[] nums, int[] queries) {
int[] pre = new int[nums.length];
Arrays.sort(nums);
pre[0] = nums[0];
for(int i = 1; i < nums.length; i++){
pre[i] = pre[i-1] + nums[i];
}
int[] ans = new int[queries.length];
for(int i = 0; i < queries.length; i++){
int len = find(pre,queries[i]);
ans[i] = len;
}
return ans;
}
public int find(int[] pre, int target){
return erfen(pre,target,0,pre.length-1);
}
public int erfen(int[] pre, int target, int left, int right){
if(left <= right){
int mid = left + ((right-left)>>1);
if(pre[mid] > target){
return erfen(pre,target,left,mid-1);
}
else if(pre[mid] < target){
return erfen(pre,target,mi