前缀和是一种重要的预处理,能大大降低查询的时间复杂度。
最简单的一道题就是给定 n 个数和 m 次询问,每次询问一段区间的和。求一个 O(n + m) 的做法。
用 O(n) 前缀和预处理,O(m) 询问。
主要代码
for (int i = 1; i <= n; i++) {
sum[i] = sum[i - 1] + a[i];
}
while (m-- > 0) {
int L, R;
printf("%d\n", sum[R] - sum[L - 1]);
}
}
针对于多次访问区间和问题
(1):
例题:
多次访问某区间内的素数个数,适用于前缀和
代码实现:
int n = sc.nextInt();
int m = sc.nextInt();
sum[1]=0;
prime[1]=1;
for (int i = 2; i <= m; i++) {
if (prime[i] == 0) {//判断素数
sum[i] = sum[i - 1] + 1;//前缀和
for (int j = i * i; j <= m; j += i) {//将合数标记
prime[j] = 1;//标记
}
} else {
sum[i] = sum[i - 1];//状态转移
}
}
全部代码:
import java.math.BigDecimal;
import java.util.*;
public class Main {
public static void main(String[] args) {
// TODO Auto-generated method stub
Scanner sc = new Scanner(System.in);
int[] prime = new int[1000001];
int[] sum = new int[1000001];
int n = sc.nextInt();
int m = sc.nextInt();
sum[1]=0;
prime[1]=1;
for (int i = 2; i <= m; i++) {
if (prime[i] == 0) {//判断素数
sum[i] = sum[i - 1] + 1;//前缀和
for (int j = i * i; j <= m; j += i) {//将合数标记
System.out.println(j+"----");
prime[j] = 1;//标记
}
} else {
sum[i] = sum[i - 1];//状态转移
}
}
for (int i = 0; i < n; i++) {
int l = sc.nextInt();
int r = sc.nextInt();
if(l<1|r>m) {
System.out.println("Crossing the line");
continue;
}
System.out.println(sum[r]-sum[l-1]);
}
}
}
(2):
看起来很简单:
import java.util.Scanner;
public class Main5 {
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
while (sc.hasNext()) {
int n = sc.nextInt();
int m = sc.nextInt();
int sum = 0;
if (n == 0 && m == 0) {
break;
}
for (int i = n; i <= m; i++) {
String str = String.valueOf(i);
if (!str.contains("4") &!str.contains("62")) {
sum++;
}
}
System.out.print(sum);
System.out.println();
}
}
}
测试:
超时!
分析:数据很大,而且可能超级多的测试数据,可以用前缀和提前将区间值处理出来,直接打表操作
import java.util.Scanner;
public class Main5 {
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int[] arr = new int[1000000];
/*定义一个前缀和数组
*
*/
for (int i = 1; i < 1000000; i++) {
String str = String.valueOf(i);
if (!str.contains("4") & !str.contains("62")) {
arr[i] = arr[i - 1] + 1;//如果号不吉利,加上1
} else {
arr[i] = arr[i - 1];//如果吉利,不增加·
}
}
while (sc.hasNext()) {
int n = sc.nextInt();
int m = sc.nextInt();
int sum = 0;
if (n == 0 && m == 0) {//判断
break;
}
System.out.println(arr[m]-arr[n-1]);//直接输出
}
}
}