前缀和差分算法

目录

前缀和

子矩阵的和

差分

差分矩阵

---

前缀和

 题目要求

 代码

#include<iostream>

using namespace std;

const int N=1e7+10;

int q[N];
int S[N];

int main()
{
    int n,m;
    cin>>n>>m;
    
    for(int i=1;i<=n;i++) scanf("%d",&q[i]);
    
    S[0]=0;
    for(int i=1;i<=n;i++)
    {
        S[i]=S[i-1]+q[i];
    }
    
    
    int l,r;
    int A;
    while(m--)
    {
        cin>>l>>r;
        A=S[r]-S[l-1];
        cout<<A<<endl;
    }
    
    return 0;
}

不算是模板，更像是一种思维，记住就好 

子矩阵的和

题目要求

 代码实现

#include<iostream>

using namespace std;

const int N=1010;

int n,m,q;
int a[N][N],s[N][N];

int main()
{
    scanf("%d%d%d",&n,&m,&q);
    
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {
            scanf("%d",&a[i][j]);
        }
    }
    
    s[0][0]=0;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {
            s[i][j]=s[i-1][j]+s[i][j-1]-s[i-1][j-1]+a[i][j];
        }
    }
    
    int x1,y1,x2,y2;
    while(q--)
    {
        scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
        int A=s[x2][y2]-s[x1-1][y2]-s[x2][y1-1]+s[x1-1][y1-1];
        printf("%d\n",A);
    }
    
    return 0;
}

差分

 

 

 b为差分数组

b1  = a1，b2 = a2 - a1,b3 = a3 - a2;

a1 = b1,a2 = b2 + b1,a3 = b3 + b2 + b1;

a是b数组的前缀和，b是a数组的差分

想要实现题目要求可以操作b数组，在l处加c即bl = bl + c，l之后的所有a就都加上了c

然后在r处打补丁，使得bl+1 = bl+1 - c，之后的所有a就还是a，没有加c

实现代码

#include <iostream>
#include <vector>
#include<algorithm>
using namespace std;


const int N = 100100;
int a[N], b[N];

void insert(int l, int r, int c)
{
	b[l] += c;
	b[r + 1] -= c;
	return;
}

int main()
{
	int n, m;
	cin >> n >> m;
	for (int i = 1; i <= n; i++)
		cin >> a[i];
	for (int i = 0; i <= n; i++)
		insert(i, i, a[i]);
	int l, r, c;
	while (m--)
	{
		cin >> l >> r >> c;
		insert(l, r, c);
	}
	for (int i = 1; i <= n; i++)
		b[i] += b[i - 1];
	for (int i = 1; i <= n; i++)
		printf("%d ", b[i]);
	
	return 0;
}

 要分清每次循环是从1开始还是0开始，对应不同的意思

 因为这里需要有个b[i - 1],所以i不能从0开始，当然，另外加一个if判断也可以

这里的第几个数第几个数并不是按照数组下标给的，要注意代码和日常使用的区别

差分矩阵

 

#include<iostream>

using namespace std;

const int N=1010;

int n,m,q;
int a[N][N],b[N][N];

void insert(int x1,int y1,int x2,int y2,int c)
{
    b[x1][y1]+=c;
    b[x2+1][y1]-=c;
    b[x1][y2+1]-=c;
    b[x2+1][y2+1]+=c;
}

int main()
{
    scanf("%d%d",&n,&m);
    scanf("%d",&q);
    
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {
            scanf("%d",&a[i][j]);
        }
    }
    
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {
            insert(i,j,i,j,a[i][j]);
        }
    }
    
    while(q--)
    {
        int x1,y1,x2,y2,c;
        scanf("%d%d%d%d%d",&x1,&y1,&x2,&y2,&c);
        insert(x1,y1,x2,y2,c);
    }
    
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {
            b[i][j]+=b[i-1][j]+b[i][j-1]-b[i-1][j-1];
        }
    }
    
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {
            printf("%d ",b[i][j]);
        }
        cout<<endl;
    }
    
    return 0;
}