前缀和与差分

前缀和

地图上有 N 个目标，用整数Xi,Yi表示目标在地图上的位置，每个目标都有一个价值Wi。注意：不同目标可能在同一位置。
现在有一种新型的激光炸弹，可以摧毁一个包含 R×R 个位置的正方形内的所有目标。激光炸弹的投放是通过卫星定位的，但其有一个缺点，就是其爆炸范围，即那个正方形的边必须和x，y轴平行。
求一颗炸弹最多能炸掉地图上总价值为多少的目标。

前缀和的定义就是对于当前数组，第i个数以前的数的总和，形成一个新的数组，即前缀和数组。$sum[i]={\sum }_{j=0}^{i}num[i]$
以上是一维情况，对于二维数组，我们可以把它看成一个矩形，当前点的前缀和就是所在行以前和所在列以前所有数的总和$sum[i][j]={\sum }_{m=0}^i{\sum }_{n=0}^{j}num[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;
}

差分

差分的定义就是，对于给定的数组$num$，第i个数和第i-1的差值所组成的新的数组就是差分数组，$dif[i]=num[i]-num[i-1]$（$dif[0]=num[0]$)。
对于差分数组和原数组存在如下关系
$num[i]=dif[0]+dif[1]+\cdot \cdot \cdot +dif[i]=num[0]+num[1]-num[0]+\cdot \cdot \cdot +num[i]-num[i-1]$

给定一个长度为 n 的数列 a1,a2,…,an，每次可以选择一个区间 [l,r]，使下标在这个区间内的数都加一或者都减一。
求至少需要多少次操作才能使数列中的所有数都一样，并求出在保证最少次数的前提下，最终得到的数列可能有多少种。

利用差分数组，对于某个区间（l, r）内加1，只需$b[l]+=1,b[r]-=1$，这样就可以在$O(1)$时间复杂度更改，而不用遍历（l，r）范围内的数组。
借用差分数组，我们只需将差分数组全部变为0，就可以使原数组的数全部相同。而对于可执行的操作，我们可以分成如下4类：

1. 1<=i，j<=n 该操作是针对i，j范围，必然伴随差分数组某两个数的加1以及减1
2. i=0, 1<=j<=n 该操作是针对j以前的所有数，只伴随着差分数组某个数的加1或者减1
3. 1<=i<=n, j=n+1 该操作是针对i以后的所有数，同上
4. 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;
}