CF 1700C Helping the Nature

题目链接

https://codeforces.com/problemset/problem/1700/C

题目描述

给出一个长度为 $n$ 的序列 $a$，每次可以进行三种操作中的一种：

• 选择 $i$，将 $a_1,a_2,\cdots ,a_i$ 减 $1$。
• 选择 $i$，将 $a_i,a_{i+1},\cdots ,a_n$ 减 $1$。
• 将所有 $a_i$ 加 $1$。

求最少需要多少次操作将所有 $a_i$ 变为 $0$。

$1\le T\le 2\times 10^4$，$1\le n\le 2\times 10^5$，$-10^9\le a_i\le 10^9$，$\sum n\le 2\times 10^5$。

输入输出样例

输入样例

4
3
-2 -2 -2
3
10 4 7
4
4 -4 4 -4
5
1 -2 3 -4 5

输出样例

2
13
36
33

解题思路

根据题目中的限制条件可知，作减法运算时对两端的区间起效果，加法运算是对所有数起效果的，我们可以使用操作1或操作2把序列变为单调序列，变为单调序列后再累加就方便多了。
算法步骤：

1. 使用减法构造一个单调递增的数列；
2. 将单调递增序列变为正数序列，以$a_1$的值为准；如果$a_1$的值为正，序列不需要累加
3. 将正数序列变为0序列，操作次数以$a_n$为准。

操作次数 = 减的次数+将序列变为正数序列次数+将正数序列变为0的次数

以数据4 -4 4 -4为例：

做题过程中，根据样例给出的数据模拟一下，很快就可以发现规律了。

参考代码

#include <iostream>
#include <cstdio>
using namespace std;
const int MAXN=200005;
int a[MAXN];
int main(){
	int t;
	//freopen("1700c.txt","r",stdin);
	scanf("%d",&t);
	while ( t-- ) {
		int n;
		scanf("%d",&n);
		for(int i = 1; i <= n; i++){
			scanf("%d", &a[i]);
		}
		long long sum = 0; //累积减掉的数值 
		for(int i = n; i >= 2;i--)
			if( a[i-1] > a[i]) sum += a[i-1] - a[i];
		//printf("%d\n",sum); 
		if( a[1]-sum>=0 ) printf("%lld\n", sum + a[n]);
		else printf("%lld\n",sum+(-1)*(a[1]-sum)*2 + a[n]);
	}
	return 0;
}