前缀和的概念
对于一个给定的数组A,它的前缀和数组S中S[i]表示从第1个元素到第i个元素的总和,用公式表示为:
Si=A1+A2+A3+...+An
前缀和的作用
在O(1)的时间求出数组任意区间的区间和。
降低求解的复杂度
算法模板
int n=10; int [] arr=new int [n]; long [] sum=new long[n+1]; for(int i=1;i<=n;i++){ sum[i]=sum[i-1]+arr[i-1]; }
例题解析
给定一个长度为n的数组a以及m个查询
每个查询包含三个整数l,r,k表示询问l~r之间所有元素的k次方和
请对每个查询输出一个答案,答案对10^9+7取模
输入格式
第一含输入两个整数n,m其含义如上所述。
第二行输入n个整数a[1],a[2],...a[n]
接下来m行,每行输入三个整数l,r,k表示一个查询
输出格式
输出m行,每行一个整数,表示查询的答案对10^9+7取模的结果。
package Accommodate; import java.util.*; public class Main1 { public static void main(String[] args) { Scanner scan=new Scanner(System.in); int n=scan.nextInt(); int q=scan.nextInt(); int arr[]=new int [n]; for(int i = 0;i<n;i++) { arr[i]=scan.nextInt(); } long mod=(long)1e9+7; long [][]sum=new long[6][n+1]; for(int i=1;i<6;i++) { for(int j=1;j<=n;j++) { sum[i][j]=sum[i][j-1]+(long)Math.pow(arr[j-1],i); sum[i][j]%=mod; } } for(int i=0;i<q;i++) { int l=scan.nextInt()-1; int r=scan.nextInt()-1; int k=scan.nextInt(); long ans=sum[k][r+1]+mod-sum[k][l]; ans%=mod; System.out.println(ans); } } }