题意:
给了你个大小为a的数列,你现在可以执行三种操作:
1. 选择一个i,1~i的所有数都-1。
2. 选择一个i,i~n的所有数都-1。
3. 所有的数都+1。
问你让所有的数都变为0所需要的步骤。
思路:
看到了这样的操作就因该想到差分了。让所有的数都为0,其实就是让差分数组全为0即可。那么上面的操作就是这样的(假设原数组为a,差分数组为s):
1.s[1] - 1, s[i + 1] + 1。
2.s[i] - 1。
3.s[1] + 1。
那么当s[i] < 0的时候,我们采取的操作是操作1,但是s[1]我们需要+s[i]。
当s[i] > 0的时候,我们需要采取操作2。
最后我们在单独对s[1]进行操作,这个时候我们只需要加上abs(s[1])即可,因为无论s[1]的大小是正是负,我们都是通过操作1或者操作2来完成的。
AC代码
#include <bits/stdc++.h>