算法:前缀和

3 阅读1分钟

前缀和

一维前缀和: 原数组:a[n] 一维前缀和数组:s[n]

s[i]=s[i-1]+a[i]

题目描述

输入一个长度为 n 的整数序列。 接下来再输入 m 个询问,每个询问输入一对 l,r。 对于每个询问,输出原序列中从第 l 个数到第 r 个数的和。

输入格式

第一行包含两个整数 n 和 m。 第二行包含 n 个整数,表示整数数列。 接下来 m 行,每行包含两个整数 l 和 r,表示一个询问的区间范围。

输出格式

共 m 行,每行输出一个询问的结果。

数据范围

1≤l≤r≤n, 1≤n,m≤100000, −1000≤数列中元素的值≤1000

输入样例:

5 3 2 1 3 6 4 1 2 1 3 2 4

输出样例:

3 6 10

AC代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
ll n,m;
ll a[1000],s[1000];
int main()
{
	cin>>n>>m;
	for(int i=1;i<=n;i++)
	cin>>a[i];
	for(int i=1;i<=n;i++)
	s[i]=s[i-1]+a[i];
	ll l,r;
	while(m--){
		cin>>l>>r;
		cout<<s[r]-s[l-1]<<endl;
	}
	return 0;
}

字符串前缀和例题

十四届蓝桥杯省赛 F题 子串简写

先找出字符串中a的个数,并将其进行前缀和,根据题目要求,确定字符串大于 n 的情况。

#include<bits/stdc++.h>
using namespace std;
int n,l,c[500005];
long long sum;
char s[500005],a,b;
int main(){
  	scanf ("%d\n%s %c %c",&n,s+1,&a,&b);
  	l=strlen(s+1);
  	for(int i=1;i<=l;i++){
    	if (s[i]==a) c[i]++;
    		c[i]+=c[i-1];
    		//cout<<c[i]<<endl;
    	if (i>=n&&s[i]==b){
      		sum += c[i-n+1];
    	}
  	}
  	cout<<sum;
  	return 0;
}