1.区域和检索——数组不可变
题目:
给定一个整数数组 nums
,处理以下类型的多个查询:
计算索引 left
和 right
(包含 left
和 right
)之间的 nums
元素的和,其中 left <= right
实现 NumArray
类:
NumArray(int[] nums)
使用数组nums
初始化对象int sumRange(int i, int j)
返回数组nums
中索引left
和right
之间的元素的 总和 ,包含left
和right
两点(也就是nums[left] + nums[left + 1] + ... + nums[right]
)
解题思路:
创建新的数组sum[],数组长度比nums[]多一个,sum[]用来储存nums[]每项累加的和,例如:int[] nums={1,2,3},则int[] sum={1,3,6},推出sumRange(left,right)=sum[right+1]-sum[left],sum[x+1]=sum[x]+nums[x],sum[x+1]就是所求的元素之和。
代码实现:
class NumArray {
int[] sum;
public NumArray(int[] nums) {
int n=nums.length;
sum=new int[n+1];
for(int x=0;x<n;x++){
sum[x+1]=sum[x]+nums[x];
}
}
public int sumRange(int left, int right) {
return sum[right+1]-sum[left];
}
}
结果:
2.所有奇数长度字数组的和
题目:
给你一个正整数数组 arr
,请你计算所有可能的奇数长度子数组的和。
子数组 定义为原数组中的一个连续子序列。
请你返回 arr
中 所有奇数长度子数组的和 。
解题思路:
运用前缀和算法,与第一题有些相似,在第一题的基础上解题,sum[]数组长度比arr[]多一个,sum[]用来储存nums[]每项累加的和,两个for嵌套进行遍历,头下标从前开始,尾下标end=头下标s+字数组的长度l-1,sum[end+1]-sum[s]算出子数组的和赋值给sums,返回sums。
代码实现:
class Solution {
public int sumOddLengthSubarrays(int[] arr) {
int n=arr.length;
int[] sum=new int[n+1];
for(int x=0;x<n;x++){
sum[x+1]=sum[x]+arr[x];
}
int sums=0;
for(int s=0;s<n;s++){
for(int l=1;s+l<=n;l+=2){
//length=end-start+1;
int end=s+l-1;
sums+=sum[end+1]-sum[s];
}
}
return sums;
}
}
结果:
3.和相同的二元子数组
题目:
给你一个二元数组 nums
,和一个整数 goal
,请你统计并返回有多少个和为 goal
的 非空 子数组。
子数组 是数组的一段连续部分。
解题思路:
使用前缀和算法,两个for循环套用,n>end>=start,end从start开始递增,arr[end+1]-arr[start]就是子数组的和。
代码实现:
class Solution {
public int numSubarraysWithSum(int[] nums, int goal) {
int n=nums.length;
int[] arr=new int[n+1];
for(int x=0;x<n;x++){
arr[x+1]=arr[x]+nums[x];
}
int num=0;
for(int start=0;start<n;start++){
for(int end=start;end<n;end++){
if(arr[end+1]-arr[start]==goal){
num++;
}
}
}
return num;
}
}
结果: