问题描述:
简述:可以从[0,i]中每个数减 1,也可以[i,n-1]中每个数减 1,问能否使数组中的元素全部为0
差分思想:
给定一个数组 : 11 7 9 6 8,它的差分数组为 11 -4 2 -3 2, 那么我们的问题转化为能不能通过操作 1 和操作 2使得查分数组变为 0 0 0 0 0
操作1:diff[1]-1 且diff[i+1]+1。那么我们就可以凭借操作1,将差分数组中的负数变成0,同时减小ans[1]。
操作2:diff[i+1]-1且diff[n+1]+1。那么我们就可以凭借操作2,将差分数组中的正数变成0,而ans[n+1]为多少和我们并没有关系。
解释一下上面两句话的意思,例如,我们先碰到 -4,那么这个位置要加上4个 1,相同的,第一个位置要减去4个1,然后我们遇到2,那我们进行操作2,即这个位置减去2个1,第n+1个位置加上2个1,但我们不关心第n+1个位置的值,因为这是我们假想出来的。 这样从左到右遍历后,第2 到第 n 个差分数组中的元素都会是0(可能n+1不会是,但不重要)。这些减去的数是a[0](也可以理解为diff[1])至少要减去的,因为我们尽可能用操作2去帮a[0]分担了(贪心),如果这必须要减去的数还要比a[0]大,那寄了。如果要小于等于,我们接下来的操作只需要减a[0]了。
#include<iostream>
using namespace std;
#include<vector>
#include<algorithm>
#include <numeric>
void solve() {
int n = 0;
cin >> n;
vector<int> a(n + 1, 0);
vector<int> diff(n + 1, 0);
for (int i = 1; i <= n; i++) {
cin >> a[i];
diff[i] = a[i] - a[i - 1];
}
int res = 0;
for (int i = 2; i <= n; i++) {
if (diff[i] < 0) res += (-1) * diff[i];
}
if (res <= a[1]) {
cout << "YES" << endl;
}
else cout << "NO" << endl;
}
int main() {
int N = 0;
cin >> N;
for (int i = 0; i < N; i++)
solve();
}