前缀和与差分

前缀和

1. 前缀和

输入一个长度为n的整数序列。

接下来再输入m个询问，每个询问输入一对l, r。

对于每个询问，输出原序列中从第l个数到第r个数的和。

输入格式

第一行包含两个整数n和m。

第二行包含n个整数，表示整数数列。

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

输出格式

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

数据范围

$1\le l\le r\le n$
$1\le n,m\le 100000$
$-1000\le 数列中元素的值\le 1000$

输入样例：

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

输出样例：

3
6
10

代码:

#include <iostream>

using namespace std;

//
const int maxn = 100005;
int n,m;
int l,r;
int q[maxn];
long long sum[maxn];  //存储前i个数的和
//


int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&q[i]);	//题目输入较大 建议用scanf 或者 cin关闭同步
        if(i==1) sum[i] = q[i];  //第一个数的前缀和就是他本身
        if(i>1) sum[i] = sum[i-1]+q[i]; //等于前一个数的前缀和加上本身
    }
    for(int i=0;i<m;i++) //m次询问
    {
        scanf("%d%d",&l,&r);
        printf("%lld\n",sum[r] - sum[l-1]);  // l ~ r的和就是前r个数的前缀和 减去前 l-1个数的前缀和
    }
}

2. 子矩阵的和

输入一个n行m列的整数矩阵，再输入q个询问，每个询问包含四个整数x1, y1, x2, y2，表示一个子矩阵的左上角坐标和右下角坐标。

对于每个询问输出子矩阵中所有数的和。

输入格式

第一行包含三个整数n，m，q。

接下来n行，每行包含m个整数，表示整数矩阵。

接下来q行，每行包含四个整数x1, y1, x2, y2，表示一组询问。

输出格式

共q行，每行输出一个询问的结果。

数据范围

$1\le n,m\le 1000$
$1\le q\le 200000$
$1\le x1\le x2\le n$
$1\le y1\le y2\le m$
$-1000\le 矩阵内元素的值\le 1000$

输入样例：

3 4 3
1 7 2 4
3 6 2 8
2 1 2 3
1 1 2 2
2 1 3 4
1 3 3 4

输出样例：

17
27
21

代码:

#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;

//
    const int maxn = 1005;
    int n , m , k;
    int q[maxn][maxn];
    long long matSum[maxn][maxn];
//

int main()
{
    memset(matSum,0,sizeof(matSum));
    cin>>n>>m>>k;
    for(int i=1;i<=n;i++)
    for(int j=1;j<=m;j++)
    {
        cin>>q[i][j];
        /*求matSum*/
        matSum[i][j] = q[i][j] + matSum[i-1][j] + matSum[i][j-1] - matSum[i-1][j-1];
    }

    int x1,y1,x2,y2;
    for(int i=0;i<k;i++)
    {
        cin>>x1>>y1>>x2>>y2;
        printf("%lld\n",matSum[x2][y2] - matSum[x1-1][y2] - matSum[x2][y1-1] + matSum[x1-1][y1-1]);
    }
}

差分

1. 差分

输入一个长度为n的整数序列。

接下来输入m个操作，每个操作包含三个整数l, r, c，表示将序列中[l, r]之间的每个数加上c。

请你输出进行完所有操作后的序列。

输入格式

第一行包含两个整数n和m。

第二行包含n个整数，表示整数序列。

接下来m行，每行包含三个整数l，r，c，表示一个操作。

输出格式

共一行，包含n个整数，表示最终序列。

数据范围

$1\le n,m\le 100000$
$1\le l\le r\le n$
$-1000\le c\le 1000$
$-1000\le 整数序列中元素的值\le 1000$

输入样例：

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

输出样例：

3 4 5 3 4 2

$q_{i-1}=di{f}_1+di{f}_2+di{f}_3+...+di{f}_{i-1}$
$q_i=di{f}_1+di{f}_2+di{f}_3+...+di{f}_{i-1}+di{f}_i$
$那么di{f}_i=q_i-q_{i-1}$

$dif[n]$叫做$q[n]$的差分数组

$q[n]$叫做$dif[n]$的前缀和

---

如果$di{f}_i$加上一个数$c$,那么它的前缀和数组q中,从$i$到$n$每一项都加上$c$.

代码:

#include <bits/stdc++.h>
using namespace std;

//
	int n,m;
	int r,l,c;
	int q[100005];
	int dif[100005]; //q[]的差分数组 :
	//q_{i-1} = dif_1 + dif_2 + dif_3 + ...+dif_{i-1} 
	//q_i     = dif_1 + dif_2 + dif_3 + ...+dif_{i-1} + dif_i 
	//那么dif_i = q_i - q_{i-1} 
//

int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
		scanf("%d",&q[i]);
	//构造差分数组
	dif[1] = q[1]; 
	for(int i=2;i<=n;i++) 
        dif[i] = q[i] - q[i-1];
	for(int i=0;i<m;i++)
	{
		scanf("%d%d%d",&l,&r,&c);
		dif[l] += c;
		dif[r+1] -= c;
	}
	//求dif的前缀和
	q[1] = dif[1];
	printf("%d ",q[1]);
	for(int i=2;i<=n;++i)
	{
		q[i] = q[i-1] + dif[i];
		printf("%d ",q[i]);
	}
	return 0;
}

2. 差分矩阵

输入一个n行m列的整数矩阵，再输入q个操作，每个操作包含五个整数x1, y1, x2, y2, c，其中(x1, y1)和(x2, y2)表示一个子矩阵的左上角坐标和右下角坐标。

每个操作都要将选中的子矩阵中的每个元素的值加上c。

请你将进行完所有操作后的矩阵输出。

输入格式

第一行包含整数n,m,q。

接下来n行，每行包含m个整数，表示整数矩阵。

接下来q行，每行包含5个整数x1, y1, x2, y2, c，表示一个操作。

输出格式

共 n 行，每行 m 个整数，表示所有操作进行完毕后的最终矩阵。

数据范围

$1\le n,m\le 1000$
$1\le q\le 100000$
$1\le x1\le x2\le n$
$1\le y1\le y2\le m$
$-1000\le c\le 1000$
$-1000\le 矩阵内元素的值\le 1000$

输入样例：

3 4 3
1 2 2 1
3 2 2 1
1 1 1 1
1 1 2 2 1
1 3 2 3 2
3 1 3 4 1

输出样例：

2 3 4 1
4 3 4 1
2 2 2 2

代码:

#include <iostream>
using namespace std;

//
    const int maxn = 1005;
    int n,m;
    int q[maxn][maxn];
    int dif[maxn][maxn]; //差分数组
    //q[i][j] = 从[1,1] 到 [i,j] 的 矩阵的元素和
//
void insert(int x1,int y1,int x2,int y2,int c)
{
    dif[x1][y1] += c;
    dif[x1][y2+1] -= c;
    dif[x2+1][y1] -= c;
    dif[x2+1][y2+1] += c;
}
int main()
{
    int k;
    scanf("%d%d%d",&n,&m,&k);
    for(int i=1;i<=n;i++)
    for(int j=1;j<=m;j++)
    {
        scanf("%d",&q[i][j]);
        insert(i,j,i,j,q[i][j]);
    }
    
    int x1, y1, x2, y2 ,c;
    for(int i=0;i<k;i++) //k次操作 
    {
        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++)
        {
            dif[i][j] += dif[i-1][j] + dif[i][j-1] - dif[i-1][j-1];
            printf("%d ",dif[i][j]);
        }
        printf("\n");
    }
    
}