差分算法（一维加二维解析）

BOF_dcb

已于 2024-02-18 10:37:04 修改

阅读量321

点赞数 6

分类专栏：基础算法文章标签：算法 c++ 数据结构

于 2024-01-16 10:03:33 首次发布

本文链接：https://blog.csdn.net/m0_62119368/article/details/135616187

版权

基础算法专栏收录该内容

8 篇文章 0 订阅

订阅专栏

本文介绍了如何使用一维和二维差分数组在C++中高效地对数组进行操作，包括给子矩阵加常数以及计算前缀和，通过构造差分数组来简化计算过程。

摘要由CSDN通过智能技术生成

概念：给一个数组a[N], 假设b[N]是a[N]的差分数组，则a[N]为b[N]的前缀和数组。

一维差分

目标：让a[L~R]加常数c。

如果b[L]+=c, 则求数组b的前缀和后，a[L~N]都在原基础上加了c
因此，我们让b[R+1]-=c,就实现了 让a[L~R]加常数c

差分数组构造：
b[1]=a[1]
b[2]=a[2]-a[1]
b[3]=a[3]-a[2]
…
b[n]=a[n]-a[n-1]

#include<iostream>

using namespace std;

const int N=1e6+10;
int n,m,l,r,c,a[N],b[N];

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

int main()
{
    scanf("%d%d",&n,&m);
    
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        insert(i,i,a[i]);
    }
    
    while(m--)
    {
        scanf("%d%d%d",&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++)
    cout<<b[i]<<" ";
    
    return 0;
    
    
}

二维差分

原数组a[N][N] , 差分数组b[N][N]；

目标：将原数组a[N][N]的一个子矩阵( (x1,y1)到(x2,y2) )整体加c,
其中 (x1,y1) 和 (x2,y2)表示一个子矩阵的左上角坐标和右下角坐标。

实现：
同理，如果b[x1[y1]+=c后，求前缀和, 原数组a[N][N]的子矩阵( (x1,y1)到(n,n) ) 整体加c,
因此，只要让

b[x1][y2+1]-=c;
b[x2+1][y1]-=c;
b[x2+1][y2+1]+=c;

然后再求前缀和，就可以使原数组a[N][N]的一个子矩阵( (x1,y1)到(x2,y2) )整体加c了。

#include<iostream>
#include<cstdio>

using namespace std;

const int N=1010;
int n,m,q;
int a[N][N] , b[N][N] , s[N][N];

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


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