Extreme Subtraction

题目链接: Extreme Subtraction

大致题意

给定一个长度为n的序列, 你可以进行任意次下列操作:
​ ①选取[1, k]区间, 使得区间内所有数字-1
​ ②选取[k, n]区间, 使得区间内所有数字-1

问: 是否可以将整个序列变为全0.

解题思路

不得不说, 这个题的解法真的太巧妙了. 想明白了之后, 脑子里都是 妙啊!

假设原序列为a, 则我们需要构造原序列的差分序列b.

对于最终结果: 如果a序列可以变为全0, 则其差分序列必然也是全0序列.
==> 我们可以将对原序列的操作变为对于差分序列的操作, 即: 让差分序列变为全0序列.

那么这样有什么好处?

首先对于操作1: 相当于b[1] -= 1; b[k + 1] += 1; 称为新操作1
其次对于操作2: 相当于b[k] -= 1; b[n + 1] += 1; 称为新操作2

新操作1的实质是, 将b序列任意位置+1, 而代价是b[1]会-1;
新操作2的实质是: 将b序列的任意位置-1, 没有代价.(因为b[n + 1]的位置超出了序列长度, +1对整个序列无意义)

那么此时我们的思路就十分明显了:
​ 如果此时b[i]是负数, 那么我需要执行|b[i]|次新操作1, 将b[i]变为0.
​ 如果此时b[i]是正数, 那么我需要执行b[i]次新操作2, 将b[i]变为0.

因为新操作1至多执行b[i]次, 最后只需要判断新操作1是否需要执行超过b[i]次即可.

AC代码

#include <bits/stdc++.h>
#define rep(i, n) for (int i = 1; i <= (n); ++i)
using namespace std;
typedef long long ll;
const int N = 3E4 + 10;
int a[N], b[N];
int main()
{
    int t; cin >> t;
    while (t--) {
        int n; scanf("%d", &n);
        
        int sum = 0;
        rep(i, n) {
            scanf("%d", &a[i]);
            b[i] = a[i] - a[i - 1];
            if (b[i] < 0) sum += -b[i];
        }
        
        if (sum <= a[1]) printf("YES\n");
        else printf("NO\n");
    }
    return 0;
}

妙呀!

END