python

7.5 KiB

Raw Blame History Unescape Escape

This file contains ambiguous Unicode characters that may be confused with others in your current locale. If your use case is intentional and legitimate, you can safely ignore this warning. Use the Escape button to highlight these characters.

前缀和和差分洛谷题单总结

参考文献

一、公式

前缀和的公式

一维：s[i] = a[i] + s[i-1]

二维：s[i][j] = a[i][j] + s[i-1] [j] + s[ i] [j-1] - s[i-1][j-1]

差分的公式

一维：b[i] =s[i] - s[i-1]

二维：b[i] = s[i][j] - s[i-1][j]-s[i][j-1]+s[i-1][j-1]

二、题单

`P1115` 最大子段和

分析先求每个位置的前缀和（某个区间求和前缀和可以说是最快的），然后去找该位置前前缀和的最小值，如果要求一段和最大，就要用这段和减去前面最小的值。

#include <bits/stdc++.h>
using namespace std;
const int N = 1e6 + 10;
const int INF = 0x3f3f3f3f;
int n, a[N], s[N], ans[N];

int main() {
    cin >> n;
    for (int i = 1; i <= n; i++)
        cin >> a[i], s[i] = s[i - 1] + a[i]; // 前缀和

    int mi = 0;
    for (int i = 1; i <= n; i++) {
        ans[i] = s[i] - mi;
        mi = min(mi, s[i]);
    }
    int res = -INF;
    for (int i = 1; i <= n; i++) res = max(res, ans[i]);
    cout << res << endl;
    return 0;
}

`P3397` 地毯

分析看到这里的时候，我就想到了一个矩阵的某个子矩阵进行加减，瞬间想到二维差分和二位前缀和,二位差分的公式为：

由差分算的二位前缀和公式：

#include <bits/stdc++.h>
using namespace std;

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

int main() {
    cin >> n >> m;
    while (m--) {
        // 从0开始构建差分数组
        int x1, y1, x2, y2;
        cin >> x1 >> y1 >> x2 >> y2;
        b[x1][y1] += 1; // 进行子矩阵的加减，差分
        b[x2 + 1][y1] -= 1;
        b[x1][y2 + 1] -= 1;
        b[x2 + 1][y2 + 1] += 1;
    }
    // 还原为原始数组
    for (int i = 1; i <= n; i++) {
        for (int j = 1; j <= n; j++) {
            s[i][j] = b[i][j] + s[i - 1][j] + s[i][j - 1] - s[i - 1][j - 1]; // 把之前的加减结果进行求和
            printf("%d ", s[i][j]);                                          // 注意输出格式，每个数带一个空格
        }
        printf("\n"); // 结束一行的输出输出一个换行符号
    }
    return 0;
}

`P1083` [`NOIP2012` 提高组] 借教室

暴力

还是先看暴力怎么做吧，对于m次借教室，我们可以每次把区间s\sim t的空教室数r-=d，当有一次r<0时，则当前这个人无法被满足，直接输出-1和当前这个人的号数，然后直接结束程序。如果m次借教室都操作完成后依然没有房间数r<0，则说明所有人都可以被满足，则输出0。

综合上述做法，得分60。

#include <bits/stdc++.h>
using namespace std;
int n, m;
const int N = 1000010;
int r[N];
int main() {
    cin >> n >> m;
    // 每一天可租借教室数
    for (int i = 1; i <= n; i++) cin >> r[i];

    // 从哪天到哪天，借多少个
    for (int i = 1; i <= m; i++) {
        int d, s, t;
        cin >> d >> s >> t;
        // 从开始天到结束天
        for (int j = s; j <= t; j++) {
            r[j] -= d;      // 减去借走的教室数
            if (r[j] < 0) { // 小于0了！
                cout << -1 << endl
                     << i << endl;
                return 0;
            }
        }
    }

    cout << 0 << endl;
    return 0;
}

显然，这样做法的时间复杂度时O(N*M)的，无法通过此题，从而我们可以推知该题正确的时间复杂度应该是log级的。

正解

既然时间复杂度时log级的，于是想到了二分。

再看到每个人借教室的时间可以看成一个区间，且该区间只会对其他在该区间要借教室的人产生影响，对于区间之外的借教室的人是不会产生影响的，于是又想到了差分。

差分序列：（可用于区间增减）记录相邻两个量的变化量，所以当在一段区间[l,r]上增加a时，只需要在l处加a，在r+1处-a即可。

对于为什么可以二分：如果一个人无法被满足，则他后面的人全都不能被满足；如果一个人可以被满足，则他前面的人都可以被满足,这恰恰吻合了我们二分的性质。

#include <bits/stdc++.h>
using namespace std;
const int N = 1000010;
#define int long long
#define endl "\n"
int n, m;                   // 天数和订单的数量
int r[N];                   // 第i天学校有r[i]个教室可借用
int d[N], s[N], t[N];       // 借的教室数目、从第s天借到t天
int b[N];                   // 差分数组
bool check(int x) {         // 判断能不能通过x个人
    memset(b, 0, sizeof b); // 每次判断都要先初始化差分数组
    int sum = 0;            // 记录需要借的教室数
    for (int i = 1; i <= x; i++) {
        b[s[i]] += d[i];     // 因为只会对在s~l之间要借用教室的人产生影响，所以可以差分
        b[t[i] + 1] -= d[i]; // 差分,注意:是t[i]+1，因为要包含t[i]这个点
    }
    for (int i = 1; i <= n; i++) {
        sum += b[i];                  // 因为cf是差分数组，所以sum就是在第i天的借教室的总数
        if (sum > r[i]) return false; // 不可行，如果要借的教室多于空的教室
    }
    return true; // 可行
}
signed main() {
    cin >> n >> m;
    for (int i = 1; i <= n; i++) cin >> r[i];
    for (int i = 1; i <= m; i++) cin >> d[i] >> s[i] >> t[i];
    if (check(m)) {        // 如果全部满足
        cout << 0 << endl; // 输出0
        exit(0);           // 直接结束程序
    }
    int l = 1, r = m; // 二分左右区间
    while (l < r) {
        int mid = l + r >> 1;
        if (check(mid))  // 如果可行
            l = mid + 1; // 增多满足人数
        else             // 否则
            r = mid;     // 减少满足人数
    }
    cout << "-1" << endl
         << l; // 输出-1和需要修改的人
}

`P3406` 海底高铁

分析 ① 每一段的最小费用加起来则总体费用最小。

② 这里的区间是线段而不是一个具体的数，所以我们需要以一个统一的标准进行区间段的区分：于是我们想到了以每个区间的左端点值进行整个线段的记录。

③ 节约时间可以对某段区间做同样的加减数的方法：想到的就是差分（当然有差分就会有由差分求前缀和）。

④ 最后用得到的线段数比较两种购买方案。

注意当然代码中还有很多需要记录的细节！！！例如：线段数是站点数-1，同时差分和前缀和的循环最好是从1开始（涉及边界问题）

7.5 KiB Raw Blame History Unescape Escape