(一)问题描述
这道题不是leetcode啦,是代码随想录(卡码网)的补充题目。这个问题思路很简单,比较特殊的是需要使用ACM模式来编写代码。
卡码网 58. 区间和(第九期模拟笔试)https://kamacoder.com/problempage.php?pid=1070
给定一个整数数组 Array,请计算该数组在每个指定区间内元素的总和。第一行输入为整数数组 Array 的长度 n,接下来 n 行,每行一个整数,表示数组的元素。随后的输入为需要计算总和的区间下标:a,b (b > = a),直至文件结束。输出每个指定区间内元素的总和。
输入示例:
5 1 2 3 4 5 0 1 1 3
输出示例:
3 9
提示:数据范围0<n<=100000
(二)关键词提取
- 求解数组区间和:前缀和是有效的解决方式。
(三)解决思路
前缀和
这个问题在卡码网中使用ACM模式编写,需要自己读入和输出数据。我使用Java来编写代码。
暴力算法很简单:根据给出的下标搜索对应的区间,for循环求和。但是这种解法会超时。
import java.util.Scanner;
public class Main{//这里命名只能叫Main,因为默认这个文件名叫Main.java
public static void main(String args[]){
Scanner scanner = new Scanner(System.in);
int n=0;
if(scanner.hasNextInt()){
n=scanner.nextInt();
}
int[] vec=new int[n];
for(int i=0;i<n;i++){
if(scanner.hasNextInt()){
vec[i]=scanner.nextInt();
}
}
while(scanner.hasNextInt()){
int a=scanner.nextInt();
int b=scanner.nextInt();
int sum=0;
if(b>=a){
for(int i=a;i<=b;i++){
sum+=vec[i];
}
}
System.out.println(sum);
}
scanner.close();
}
}
此时应当使用前缀和的思想来解决这一问题。前缀和的思想是重复利用计算过的子数组之和,从而降低区间查询需要累加计算的次数。在计算涉及区间和的问题时非常有用。例如,想要求解第四个元素到第十个元素的和,那么只需要用前十个的和减去前三个元素的和就可以了。
import java.util.Scanner;
public class Main{
public static void main(String args[]){
Scanner scanner = new Scanner(System.in);
int n=0;
if(scanner.hasNextInt()){
n=scanner.nextInt();
}
int presum=0;
int[] p=new int[n];
int[] vec=new int[n];
for(int i=0;i<n;i++){
if(scanner.hasNextInt()){
vec[i]=scanner.nextInt();
presum+=vec[i];
p[i]=presum;
}
}
while(scanner.hasNextInt()){
int a=scanner.nextInt();
int b=scanner.nextInt();
int result;
if(a==0){
result=p[b];
}
else{
result=p[b]-p[a-1];
}
System.out.println(result);
}
scanner.close();
}
}
(四) 易错点
- 注意计算区间和的最后一步是p[b]-p[a-1],而不是p[b]-p[a]。
- 使用第一个循环读取数组时,要用for循环,设置循环条件是i<n(因为已经给了数组长度为n)。不能和第二个循环一样使用条件为scanner.hasNextInt()的while循环,因为不是读完数组输入就停止了,还会一直输入区间边界的,用while循环会一直往下读。
- while循环当中的if判断,要排除a=0的情况。只写b>a不行,b>a是包含了a=0的情况的。