一维前缀和
在输入的时候计算前面的和
再查询的时候前缀和就是当前r的位置的和减去l位置前面的和
k倍区间https://www.acwing.com/problem/content/submission/1232/
#include<iostream>
using namespace std;
const int N=1e5+10;
int n;
int k;
int a[N];
int f[N];
int main()
{
cin>>n>>k;
for(int i=1;i<=n;i++)
{
cin>>a[i];
f[i]=f[i-1]+a[i];
}
int cnt=0;
for(int i=1;i<=n;i++)
{
for(int j=i;j<=n;j++)
{
if((f[j]-f[i-1])%k==0) ++cnt//注意-1才是i到j的区间
}
}
cout<<cnt<<endl;
return 0;
}
//优化O(N)
#include<bits/stdc++.h>
using namespace std;
const int N=100010;
typedef long long ll;
int sum[N],a[N],res[N];
int n,k;
ll ans=0;
int main(){
cin>>n>>k;
for(int i=1;i<=n;i++){
cin>>a[i];
sum[i]=(sum[i-1]+a[i])%k;//前缀和取模
取模运算证明可以看下面acw_yu的证明
ans+=res[sum[i]];//更新答案
res[sum[i]]++;//两个相等的前缀和就能组成一个k倍区间
}
cout<<ans+res[0]<<endl;
return 0;
}
输入样例
5 3
2 1 3 6 4
1 2
1 3
2 4
输出样例:
3
6
10
思想:
#include<iostream>
using namespace std;
const int N=100010;
int s[N];
int c[N];
int n,m,l,r;
int main()
{
cin>>n>>m;
for(int i=1;i<=n;i++)
{
cin>>c[i];
s[i]=c[i]+s[i-1];
}
while(m--)
{
cin>>l>>r;
cout<<s[r]-s[l-1]<<endl;
}
return 0;
}
二维前缀和
给定一个n*m大小的矩阵a,有k次询问,每次询问给定x1,y1,x2,y2四个数,求以(x1,y1)为左上角坐标和(x2,y2)为右下角坐标的子矩阵的所有元素和。注意仍然包含左上角和右下角的元素。
上述粉红色区间为要求得的区间:
现在我们知道每个区域的和,所以粉红色区域的大小等于四个区域的值-绿色区域的-黑色区域+红色区域。所以ans= f [ x 2 ] [ y 2 ] − f [ x 1 − 1 ] [ y 2 ] − f [ x 2 ] [ y 1 − 1 ] + f [ x 1 − 1 ] [ y 1 − 1 ] f[x2][y2]-f[x1-1][y2]-f[x2][y1-1]+f[x1-1][y1-1] f[x2][y2]−f[x1−1][y2]−f[x2][y1−1]+f[x1−1][y1−1]
可以结合下面的例子进行体会,能够理解的更快速
现在我们知道了前缀和的作用,那么如何求这个前缀和呢?
递推公式如下:
f[i][j]=f[i-1][j]+f[i][j-1]-f[i][j-1]+a[i][j]
由于递推公式可知,第一行第一列要增加一个定义为0;
例如:数组a 更变后:
1 2 2 1 0 0 0 0 0
3 2 2 1 0 1 2 2 1
1 1 1 1 0 3 2 2 1
0 1 1 1 1
那么代码怎么写呢?
如下:
#include<iostream>
using namespace std;
const int N=1e9;
int a[N][N],s[N][N];
int n,m,k;
int main(){
cin>>n>>m>>k;
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++){
cin>>a[i][j];
s[i][j]=a[i][j]+s[i-1][j]+s[i][j-1]-s[i-1][j-1];
}
while(k--){
int x1,y1,x2,y2;
cin>>x1>>y1>>x2>>y2;
cout<<s[x2][y2]-s[x1-1][y2]-s[x2][y1-1]+s[x1-1][y1-1]<<endl;
}
return 0;
}
前缀和与窗口
窗口大小不变进行计算
壁画//本质上只能画(n+1)/2的画,而且画是连续的,所以求在这个长度的和哪个最大就可以
#include<iostream>
using namespace std;
const int N=5e6+10;
int n,t,s[N];
char str[N];
int main()
{
cin>>t;
for(int k=1;k<=t;k++)
{
cin>>n>>str+1; //把字符串从1的位置开始
for(int i=1;i<=n;i++) //预处理一下前缀和
s[i]=s[i-1]+str[i]-'0'; //将char转变为int
int res=0,m=(n+1)/2; //m可以理解为窗口大小
for(int i=m;i<=n;i++)
res=max(res,s[i]-s[i-m]);//更新最大值,取当前值与往后移动 m 个位置的值的差的最大值来更新 res
printf("Case #%d: %d\n",k,res);
}
return 0;
}