前缀和
地图上有 N 个目标,用整数Xi,Yi表示目标在地图上的位置,每个目标都有一个价值Wi。注意:不同目标可能在同一位置。
现在有一种新型的激光炸弹,可以摧毁一个包含 R×R 个位置的正方形内的所有目标。激光炸弹的投放是通过卫星定位的,但其有一个缺点,就是其爆炸范围,即那个正方形的边必须和x,y轴平行。
求一颗炸弹最多能炸掉地图上总价值为多少的目标。
前缀和的定义就是对于当前数组,第i个数以前的数的总和,形成一个新的数组,即前缀和数组。
s
u
m
[
i
]
=
∑
j
=
0
i
n
u
m
[
i
]
sum[i]=\sum_{j=0}^{i}num[i]
sum[i]=∑j=0inum[i]
以上是一维情况,对于二维数组,我们可以把它看成一个矩形,当前点的前缀和就是所在行以前和所在列以前所有数的总和
s
u
m
[
i
]
[
j
]
=
∑
m
=
0
i
∑
n
=
0
j
n
u
m
[
i
]
[
j
]
sum[i][j]=\sum_{m=0}^{i}\sum_{n=0}^{j}num[i][j]
sum[i][j]=∑m=0i∑n=0jnum[i][j]
可以表示成下图的形式,对于黄色框内的前缀和,就是黄色=绿色+红色-蓝色。
对于这题,我们就可以用二维前缀和进行快速求解
#include<iostream>
using namespace std;
//M是二维数组最大范围
const int M=5010;
//记录二维数组每个点的价值, 初始化为0,记作全局变量,存储在堆空间,防止栈满。
int g[M][M];
int main(){
//输入N行,每行输入一组数据,每组数据包括三个整数Xi,Yi,Wi,
//R表示爆炸范围
int N, R;
cin >> N >> R;
//n和m用来记录最大边界
int n = R, m = R;
for(int i=0, x, y, w; i<N; i++){
cin >> x >> y >> w;
//处理边界,从1开始
x++, y++;
n=max(n,x), m=max(m,y);
g[x][y] += w;
}
//计算前缀和
for(int i=1; i<=n; i++)
for(int j=1; j<=m; j++)
g[i][j] += g[i-1][j] + g[i][j-1] - g[i-1][j-1];
int res=0;
//遍历前缀和,求解最大值
for(int i=R; i<=n; i++)
for(int j=R; j<=m; j++)
res=max(res,g[i][j]-g[i-R][j]-g[i][j-R]+g[i-R][j-R]);
cout << res << endl;
return 0;
}
还可以进一步简化代码,在计算前缀和的同时求解最大值
#include<iostream>
using namespace std;
const int M=5010;
int g[M][M];
int main(){
int N, R;
cin >> N >> R;
int n = R, m = R;
for(int i=0, x, y, w; i<N; i++){
cin >> x >> y >> w;
x++, y++;
n=max(n,x), m=max(m,y);
g[x][y] += w;
}
int res=0;
int x, y;
for(int i=1; i<=n; i++){
for(int j=1; j<=m; j++){
g[i][j] += g[i-1][j] + g[i][j-1] - g[i-1][j-1];
x=max(i-R, 0);
y=max(j-R, 0);
res=max(res,g[i][j]-g[x][j]-g[i][y]+g[x][y]);
}
}
cout << res << endl;
return 0;
}
差分
差分的定义就是,对于给定的数组
n
u
m
num
num,第i个数和第i-1的差值所组成的新的数组就是差分数组,
d
i
f
[
i
]
=
n
u
m
[
i
]
−
n
u
m
[
i
−
1
]
dif[i]=num[i]-num[i-1]
dif[i]=num[i]−num[i−1](
d
i
f
[
0
]
=
n
u
m
[
0
]
dif[0]=num[0]
dif[0]=num[0])。
对于差分数组和原数组存在如下关系
n
u
m
[
i
]
=
d
i
f
[
0
]
+
d
i
f
[
1
]
+
⋅
⋅
⋅
+
d
i
f
[
i
]
=
n
u
m
[
0
]
+
n
u
m
[
1
]
−
n
u
m
[
0
]
+
⋅
⋅
⋅
+
n
u
m
[
i
]
−
n
u
m
[
i
−
1
]
num[i]=dif[0]+dif[1]+···+dif[i] =num[0]+num[1]-num[0]+···+num[i]-num[i-1]
num[i]=dif[0]+dif[1]+⋅⋅⋅+dif[i]=num[0]+num[1]−num[0]+⋅⋅⋅+num[i]−num[i−1]
给定一个长度为 n 的数列 a1,a2,…,an,每次可以选择一个区间 [l,r],使下标在这个区间内的数都加一或者都减一。
求至少需要多少次操作才能使数列中的所有数都一样,并求出在保证最少次数的前提下,最终得到的数列可能有多少种。
利用差分数组,对于某个区间(l, r)内加1,只需
b
[
l
]
+
=
1
,
b
[
r
]
−
=
1
b[l]+=1,b[r]-=1
b[l]+=1,b[r]−=1,这样就可以在
O
(
1
)
O(1)
O(1)时间复杂度更改,而不用遍历(l,r)范围内的数组。
借用差分数组,我们只需将差分数组全部变为0,就可以使原数组的数全部相同。而对于可执行的操作,我们可以分成如下4类:
- 1<=i,j<=n 该操作是针对i,j范围,必然伴随差分数组某两个数的加1以及减1
- i=0, 1<=j<=n 该操作是针对j以前的所有数,只伴随着差分数组某个数的加1或者减1
- 1<=i<=n, j=n+1 该操作是针对i以后的所有数,同上
- i=0, j=n+1 该操作是针对数组全部 差分数组不变
对于第一种情况一次操作可以更改差分数组的两个数,是我们应该优先选择的,通过加减两个差分数组的数,可以使得数组中的正数和负数相抵消,最后再用第二种和第三种方法对剩余的正数或者负数进行清零。如此满足最少操作的要求。第四种操作没有意义。
我们需要知道一点,当差分数组正数之和和负数之和相等的时候,利用第一种方式对差分数组清零,无论操作顺序是什么样的,整个数组得到的结果都是一样的,都是和初始数组首末值相等的数组。因此我们可以得出,对于最终数组的可能情况仅仅取决于对剩余正数或者负数的操作。
而对于剩余数的操作只有第二种和第三种方式,两种方式的可能组合的方式数,就是最终解。直观一点来看,例如我们最终剩余正数是下图左边所示:
黑线代表了差分数组还有两个正数差值,均是1,因此我们最终通过第2,3种方法,可以使原数组有红黄蓝三种方式使差分数组清零。最终代码如下:
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long LL;
const int N=100010;
int a[N];
int main(){
//接受数组
int n;
cin >> n;
for (int i=1; i<=n; i++) cin >> a[i];
//生成差分数组
for (int i=n; i>1; i--) a[i]-=a[i-1];
//计算差分数组正负数总和
LL pos=0, neg=0;
for(int i=2; i<=n; i++)
if(a[i]>0) pos+=a[i];
else neg-=a[i];
cout << max(pos, neg) << endl;
cout << abs(pos-neg)+1 <<endl;
return 0;
}
有 N 头牛站成一行,被编队为1、2、3…N,每头牛的身高都为整数。
当且仅当两头牛中间的牛身高都比它们矮时,两头牛方可看到对方。
现在,我们只知道其中最高的牛是第 P 头,它的身高是 H ,剩余牛的身高未知。
但是,我们还知道这群牛之中存在着 M 对关系,每对关系都指明了某两头牛 A 和 B 可以相互看见。
求每头牛的身高的最大可能值是多少。
这道题我们可以很明确的知道,数组首位的牛最高值就是H,因为给定的M对关系仅仅是数组内部的关系。所以根据于此,我们可以利用差分数组记录每对关系,在根据首位来确定原数组最理想的形式。
#include <iostream>
#include <set>
using namespace std;
const int N=10010;
int height[N];
int main(){
//n是牛的数量,p是第几头牛最高,h是最高的牛的高度,m是关系数。
int n, p, h, m;
cin >> n >> p >> h >> m;
height[1]=h;
//根据关系,对差分数组的边界进行处理
set<pair<int, int>> existed;
for(int i=0, a, b; i<m; i++){
cin >> a >> b;
if(a>b) swap(a,b);
if(!existed.count({a, b})){
existed.insert({a, b});
height[a+1]--, height[b]++;
}
}
//根据差分数组,求解原数组
for(int i=1; i<=n; i++){
height[i] += height[i-1];
cout << height[i] << endl;
}
return 0;
}