前缀和:
对于一个给定的数列,它的前缀和序列可以通过相加一一进行递推
一个部分和,数列A某个下标区间内的数的和可以表示为前缀和相见的方式
从而可以以线性的速度算出数组某一部分的和
例题 激光炸弹
一种新型的激光炸弹,可以摧毁一个边长为 R 的正方形内的所有的目标。
现在地图上有 N 个目标,用整数Xi,Yi表示目标在地图上的位置,每个目标都有一个价值Wi。
激光炸弹的投放是通过卫星定位的,但其有一个缺点,就是其爆炸范围,即那个边长为 R 的正方形的边必须和x,y轴平行。
若目标位于爆破正方形的边上,该目标不会被摧毁。
求一颗炸弹最多能炸掉地图上总价值为多少的目标。
输入格式
第一行输入正整数 N 和 R ,分别代表地图上的目标数目和正方形的边长,数据用空格隔开。
接下来N行,每行输入一组数据,每组数据包括三个整数Xi,Yi,Wi,分别代表目标的x坐标,y坐标和价值,数据用空格隔开。
输出格式
输出一个正整数,代表一颗炸弹最多能炸掉地图上目标的总价值数目。
数据范围
0<N≤10000
0≤Xi,Yi≤5000
输入样例:
2 1
0 0 1
1 1 1
输出样例:
1
解题思路:
按从左到右、从上到下的顺序遍历每个方格,依次算出每个方格的前缀和
s[i][j]处前缀和 = 图④ + s[i][j]处本身的值 (即图②+图③ - 图①的值,再加上s[i][j])
解模板
#include<iostream> using namespace std; int g[5005][5005]; int main(){ int N,R; cin >> N >> R; int n = R,m = R; if(R>5005){ //测试数据中有一个R大于边长 int res = 0; for(int i = 0,x,y,w;i < N;i++){ scanf("%d%d%d",&x,&y,&w); res += w; } cout << res <<endl; return 0; } for(int i = 0,x,y,w;i < N;i++){ scanf("%d%d%d",&x,&y,&w); x++; //坐标从1开始 避免边界判断 所以要+1 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; }
差分:
a[1],a[2],...,a[n] b[i] = a[i] - a[i - 1],b[1] = a[1] a[i] = b[1] + b[2] + ... + b[i] = a[1] + a[2] - a[1] + a[3] - a[2] + ... + a[i] - a[i - 1] = a[i] 由此证明 a是b的前缀和序列 b是a的差分序列
例题 IncDec序列
题目描述
给定一个长度为 n 的数列 ,
,…,
,每次可以选择一个区间 [l,r],使下标在这个区间内的数都加一或者都减一。
求至少需要多少次操作才能使数列中的所有数都一样,并求出在保证最少次数的前提下,最终得到的数列可能有多少种。
输入格式
第一行输入正整数n。
输出格式
第一行输出最少操作次数。
第二行输出最终能得到多少种结果。
数据范围
输入样例:
4
1
1
2
2
输出样例:
1
2
解题思路:
差分+贪心
修改数列区间[ l , r],整个区间内的数只加一或只减一,用最少的修改次数使得整个数列的数都一样。
结合差分的思想,把序列A的区间[ l , r]加d,也就是把,
…
都加上d,其实就是它的差分序列B中,
,
,其他的位置统统不改变。
要求A序列中所有的数相同,而不在意这些方案具体是什么,所以说我们就可以转化题目,也就是将对A序列的+1,−1+1,−1操作,让A序列相同,相当于每次选出…
中的任意两个数,一个加一一个减一,把
…
变成全0即可,也就是A序列全部相等。而且最后得到的序列,就是n个
。
从 ,
,… ,
中任选两个数的方法可分为四类:
1.选和
,其中 2≤i, j ≤n。这种操作会改变
,
,…,
中两个数的值。应该在保证
和
一正一负的前提下,尽量多地采取这种操作,更快地接近目标。
为何要正负配对:最终目的是要B序列下标为2~n的都要为0,所以负数加,正数减,可以最快到达目标为0的状态。无法配对的数可以选
或者
这两个不影响的数,进行修改。
2.选和
,其中 2≤j ≤n。
3.选和
,其中 2≤i≤n。
4.选和
。这种情况没有意义,因为它不会改变
,
,…,
的值,相当于浪费了一次操作,一定不是最优解。
解模板:
#include<bits/stdc++.h> 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){ //不要输入 if (c) 因为c是指不为0就好了,如果c为-1,那么最后的布尔值也为1,if(c)的意思是,只要c不为0,那么条件的布尔值都为1 pos += a[i]; }else { neg -= a[i]; } } cout << min(pos,neg) + abs(pos - neg) << endl; cout << abs(pos - neg) + 1 << endl; return 0; }
例题 最高的牛
题目描述
有 N 头牛站成一行,被编队为1、2、3 … N,每头牛的身高都为整数。
当且仅当两头牛中间的牛身高都比它们矮时,两头牛方可看到对方。
现在,我们只知道其中最高的牛是第 P 头,它的身高是 H ,剩余牛的身高未知。
但是,我们还知道这群牛之中存在着 M 对关系,每对关系都指明了某两头牛 A 和 B 可以相互看见。
求每头牛的身高的最大可能值是多少。
输入格式
第一行输入整数 N,P,H,M,数据用空格隔开。
接下来M行,每行输出两个整数 A 和 B ,代表牛 A 和牛 B 可以相互看见,数据用空格隔开。
输出格式
一共输出 N 行数据,每行输出一个整数。
第 i 行输出的整数代表第 i 头牛可能的最大身高。
数据范围
1≤N≤10000,1≤H≤1000000,1≤A,B≤10000,0≤M≤10000
输入样例:
9 3 5 5
1 3
5 3
4 3
3 7
9 8
输出样例:
5
4
5
3
4
4
5
5
5
解题思路:
M对关系中A、B两头牛可以看见说明A、B两头牛是等高度的
题目中要求牛的身高最高,因此可以将每一组关系(A、B),看作[A+1,B-1]这组牛身高只比A,B这两头牛矮1
图片来源:https://www.acwing.com/solution/content/817/
结合前缀和与差分的思想,设置一个数组hight[ i ],用于记录第i位上的牛比最高的牛矮多少,hight[P]为0。
牛身高序列A的区间(a,b)减一,其差分序列height的变化为height[a]减一,height[b+1]加一,其他位置不变。
最后再根据其差分序列,推出原本序列即可。
解模板:
#include<bits/stdc++.h> using namespace std; const int N = 10010; int height[N]; int main(){ 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; }