Codeforces Round #681 D - Extreme Subtraction（差分）

传送门

---

题目大意

给出一个序列，现在可以进行两种操作：

• 对于任何位置$i$，将其所有的前缀都减一
• 对于任何位置$i$，将其所有的后缀都减一

问最后能否将所有的数都减为零。

解题思路

首先我们不难发现，对于左半部分，如果是单调不增的，那么最后一定能都减为零，那么最后一定能通过操作前缀都减为零；同理对于右半部分，如果是单调不减的，那么最后一定也能通过操作后缀都减为零。但是如果出现了不满足单调的若干个数，如下图所示：

对于相邻于左边单调线的数，显然我们应该尽可能让其满足左边的单调性；对于相邻于右边单调线的数，显然我们应该尽可能让其满足右边的单调性。我们考虑使用两个指针维护这个序列。

• 对于$a[l+1]>a[l]$的数，需要右边所有的数都减去二者差值绝对值（至少减这么多，为了减小影响就减这么多），即$[l+1,n]$的数都减二者差值绝对值，这样不影响右边已经单调的数。而其他不满足右边单调性的数和单调线的关系不会改变。
• 对于$a[r-1]>a[r]$的数，需要左边所有的数都减去二者差值绝对值，即$[1,r-1]$的数都减二者差值绝对值，这样不影响左边已经单调的数。而其他不满足左边单调性的数和单调线的关系不会改变。

那么使用差分每次区间减法，最后差分数组还原所有数看是否有小于零的数即可。

//
// Created by Happig on 2020/11/5
//
#include <bits/stdc++.h>
#include <unordered_map>
#include <unordered_set>

using namespace std;
#define fi first
#define se second
#define pb push_back
#define ins insert
#define Vector Point
#define ENDL "\n"
#define lowbit(x) (x&(-x))
#define mkp(x, y) make_pair(x,y)
#define mem(a, x) memset(a,x,sizeof a);
typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
typedef pair<double, double> pdd;
const double eps = 1e-8;
const double pi = acos(-1.0);
const int inf = 0x3f3f3f3f;
const double dinf = 1e300;
const ll INF = 1e18;
const int Mod = 1e9 + 7;
const int maxn = 2e5 + 10;

int T, n;
int a[maxn], d[maxn];

void add(int l, int r, int x) {
    d[l] += x, d[r + 1] -= x;
}

int main() {
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    ios_base::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    cin >> T;
    while (T--) {
        cin >> n;
        for (int i = 1; i <= n; i++) {
            cin >> a[i];
            d[i] = a[i] - a[i - 1];
        }
        int l = 1, r = n;
        while (l <= r) {
            while (l + 1 <= r && a[l + 1] <= a[l]) l++;
            if (a[l] < a[l + 1]) add(l + 1, n, a[l] - a[l + 1]);
            while (r - 1 >= l && a[r - 1] <= a[r]) r--;
            if (a[r] < a[r - 1]) add(1, r - 1, a[r] - a[r - 1]);
            l++, r--;
        }
        for (int i = 1; i <= n; i++) {
            d[i] += d[i - 1];
            if (d[i] < 0) goto fail;
        }
        cout << "YES" << ENDL;
        continue;
        fail:
        cout << "NO" << ENDL;
    }
    return 0;
}

看其他博客发现有更牛皮的解法，原序列全为0意味着差分数组也全为0。利用差分的性质，如果操作前缀那么可以将差分数组的负数都变为0，此时减去的都是$d[1]$的值；如果操作后缀那么可以将差分数组的正数都变为0，但是累加到了$d[n+1]$，这对序列不产生影响。实际上问题变成了差分数组中所有的负数相加的绝对值是否大于$d[1]$，orz！

//
// Created by Happig on 2020/11/6
//
#include <bits/stdc++.h>
#include <unordered_map>
#include <unordered_set>

using namespace std;
#define fi first
#define se second
#define pb push_back
#define ins insert
#define Vector Point
#define ENDL "\n"
#define lowbit(x) (x&(-x))
#define mkp(x, y) make_pair(x,y)
#define mem(a, x) memset(a,x,sizeof a);
typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
typedef pair<double, double> pdd;
const double eps = 1e-8;
const double pi = acos(-1.0);
const int inf = 0x3f3f3f3f;
const double dinf = 1e300;
const ll INF = 1e18;
const int Mod = 1e9 + 7;
const int maxn = 1e6 + 10;

int a[maxn], d[maxn];

int main() {
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    ios_base::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    int T, n;
    cin >> T;
    while (T--) {
        cin >> n;
        ll sum = 0;
        for (int i = 1; i <= n; i++) {
            cin >> a[i];
            d[i] = a[i] - a[i - 1];
            if (d[i] < 0) sum += d[i];
        }
        if (d[1] < abs(sum)) cout << "NO" << ENDL;
        else cout << "YES" << ENDL;
    }
    return 0;
}