这道题是很经典,暴力求解肯定超时,我们可以用前缀和的思想,首先我们来讲解一下:

sum[i]表示前i个数字的和,如果求i到j之间所有数的和,我们之间sum[j]-sum[i-1]就可以得出来,但是这道题需要算的结果中每个数都乘了(r-l+1),我们可以先求出multiply[i]=a1*1+a2*2+a3*3+...+ai*i,

例如l=x,r=y,我们可以先求multiply[y]-multiply[x-1]=ax*x+ax+1*(x+1)+...+ay*y,这已经很接近结果了,只是每个数都多算几次,而且多算的次数是相同的都是(l-1)次,我们再让刚刚求出来的数减去(l-1)*(sum[y]-sum[x-1])就能求出来了。

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

public class Main {
    public static void main(String[] args) throws IOException {
        long MOD= 998244353;
        BufferedReader br=new BufferedReader(new InputStreamReader(System.in));//快读
        String[] s = br.readLine().split(" ");
        int n=Integer.parseInt(s[0]),q=Integer.parseInt(s[1]);
        long he[]=new long[n+1];//前缀和
        long cheng[]=new long[n+1];//相乘后的前缀和
        String[] s1 = br.readLine().split(" ");
        for (int i=1;i<=n;i++){
            he[i]=(he[i-1]+Long.parseLong(s1[i-1]))%MOD;
            cheng[i]=(cheng[i-1]+(Long.parseLong(s1[i-1])*i)%MOD)%MOD;
        }
        while (q-->0){
            String[] s2 = br.readLine().split(" ");
            int l=Integer.parseInt(s2[0]),r=Integer.parseInt(s2[1]);
            long count=(cheng[r]+MOD-cheng[l-1])%MOD;//减的时候需要加MOD防止出现负数
            long cha=((l-1)*(he[r]+MOD-he[l-1])%MOD)%MOD;
            long result=(count+MOD-cha)%MOD;
            System.out.println(result);
        }
    }
}

更多推荐