一维前缀和,二维前缀和,窗口描述

一维前缀和

在输入的时候计算前面的和

再查询的时候前缀和就是当前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)为右下角坐标的子矩阵的所有元素和。注意仍然包含左上角和右下角的元素。

捕获.PNG

上述粉红色区间为要求得的区间:

现在我们知道每个区域的和,所以粉红色区域的大小等于四个区域的值-绿色区域的-黑色区域+红色区域。所以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[x11][y2]f[x2][y11]+f[x11][y11]

可以结合下面的例子进行体会,能够理解的更快速

现在我们知道了前缀和的作用,那么如何求这个前缀和呢?

递推公式如下:

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;
}

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值