一维数组以及二位数组的差分
一维数组和二维数组的前缀和在上两个作品中,请先理解前缀和再来看差分
一、一维数组差分:
思路: 1、我们先定义一个数组d,它里面的值都是0,如果要让一个数组的某个 区间都加上val,比如说是[1,2](里面的是下标)区间上。
2、接下来我们要让d1]+val,d[3]-val。
3、让d数组进行前缀和,即d[i]=d[i-1]+d[i],(i要大于0)。
4 、让原数组a[i]=d[i]+a[i],这个结果就是加val值的结果。
上面a是原数组,d是差分数组,d’是思路中的第三个,a’是思路中的第四个。
下面是代码解释:
#include<stdio.h>
//前缀和:
const int n=5;
int arr[n]={1,5,2,6,4};
int sum[n];
//差分数组:
int d[n+1]={0};
void add(int l,int r,int v)
{
d[l] += v;
d[r+1] -=v;
}
int main()
{
//差分数组( (注意:这个要求询问次数少,要是再询问的话把d数组值为0!!!) ):
add(2,4,1);
add(1,4,-2);
for(int i=1;i<n;i++)
{
d[i] = d[i-1] + d[i]; //差分数组前缀合
}
for(int i=0;i<n;i++)
{
arr[i] += d[i];
printf("%d ",arr[i]);
}
return 0;
}
二、二维数组的差分:
(你要先明白二维数组的前缀和,如果你不会不用担心我上个作品你可以看看)
思路:1、老规矩和一维数组的思路差不多,先定义好差分数组,并且要算出来差分数组。
2、我们定义d[n][m]={0},接下来我们先看图片吧:
这里在x1,y1处加上val,则x1,y1后面的整个矩形都加上val,我们要得到是x1,y1到x2,y2这个矩形里面都加上val,那么剩下的三个矩形怎么办,重点来了:我们可以在x2+1,y1下标这里减去val,x1,y2+1这里减去val,这么一减,蓝色区域是不是多减了一次val,我们可以在x2+1,y2+1这里再加上个val是不是就好了(这里需要理解好二维数组的前缀和)
下面两个函数就是核心了(与一维数组差分进行联系着理解哦):
void add(int x1,int y1,int x2,int y2,int v)
{
d[x1][y1]+=v;
d[x2+1][y1]-=v;
d[x1][y2+1]-=v;
d[x2+1][y2+1]+=v;
}
void pre_d()
{
for(int i=1;i<n;i++) d[i][0]=d[i-1][0]+d[i][0]; //第一列
for(int i=1;i<m;i++) d[0][i]=d[0][i-1]+d[0][i]; //第一行
for(int i=1;i<n;i++)
{
for(int j=1;j<m;j++)
{
d[i][j]=d[i-1][j]+d[i][j-1]-d[i-1][j-1]+d[i][j];
}
}
}
基本上就是这了,完整代码是:
#include<bits/stdc++.h>
using namespace std;
const int n = 3,m = 4;
int a[n][m]={{1,2,3,4},
{2,5,6,9},
{6,4,2,3}};
int d[n][m]={0};
void add(int x1,int y1,int x2,int y2,int v)
{
d[x1][y1]+=v;
d[x2+1][y1]-=v;
d[x1][y2+1]-=v;
d[x2+1][y2+1]+=v;
}
//差分数组前缀和
void pre_d()
{
for(int i=1;i<n;i++) d[i][0]=d[i-1][0]+d[i][0]; //第一列
for(int i=1;i<m;i++) d[0][i]=d[0][i-1]+d[0][i]; //第一行
for(int i=1;i<n;i++)
{
for(int j=1;j<m;j++)
{
d[i][j]=d[i-1][j]+d[i][j-1]-d[i-1][j-1]+d[i][j];
}
}
}
//前缀和
int get_sum(int x1,int y1,int x2,int y2)
{
if(!x1&&!y1) return sum[x2][y2]; //表示x1,y1都是0
if(!x1) return sum[x2][y2]-sum[x2][y1-1];
if(!y1) return sum[x2][y2]-sum[x1-1][y2];
return sum[x2][y2]-sum[x2][y1-1]-sum[x1-1][y2]+sum[x1-1][y1-1];
}
int main()
{
add(0,0,1,2,3);
add(0,1,1,2,-1);
pre_d();
for(int i=0;i<n;i++)
{
for(int j=0;j<m;j++)
{
a[i][j]+=d[i][j];
cout<<a[i][j]<<' ';
if(j==m-1)
cout<<endl;
}
}
return 0;
}
希望可以帮到你,如果有疑问尽管问,我会回复的哦