题目链接:增减序列
代码很短,但是分析过程很重要,我这个菜鸡搞了一上午才搞明白
思路分析简述:
b1 = a1 b2 = a2-a1 b3 = a3-a2 以此类推.. bn = an - an-1
如果让数组a中的每个每个数都相等,等价于第一个数b1任选 其他的数b2=b3 ... =bn=0
因此原问题可转化为:
- 至少操作多少次,可使得b2~bn都为0
- b1有多少种选择
操作的类型可以分为三大类:
1、 2≤L≤R≤n-1 在b2~bn中令某个数加1,另一个数减1
2 、L=1 ,R≤n-1 b1加1或减1 b2~bn的某个数加一或减1
3 、2≤L ,R=n b2~bn的某个数加一或减1,b(n+1)加1或减1
设b2,b3…bn中正数总和为p,负数总和的绝对值为q。首先以正负数匹配的方式尽量执行1类操作,可执行min(p,q)次。剩余|p - q|个为匹对,每个可以选与b1或b(n + 1)匹配,即执行2 或 3 类操作,共需|p - q|次
综上所诉,最少操作次数为min(p,q) + |p - q|。根据|p - q|次第2、3类操作的选择情况,能产生|p - q| + 1中不同的b1的值,即最终得到的序列a可能有|p - q| + 1 种
AC代码:
#include<iostream>
#include<algorithm>
#include<cstdio>
using namespace std;
const int N = 100010;
typedef long long LL;
int n;
int a[N];//原数组
int b[N];//差分数组
int main()
{
scanf("%d", &n);
for (int i = 1; i <= n; i++) scanf("%d", &a[i]);
for (int i = 1; i <= n; i++) b[i] = a[i] - a[i - 1];
LL p = 0;//记录差分数组a中正数之和
LL q = 0;//记录差分数组b中负数之和
for (int i = 2; i <= n; i++) {
if (b[i] > 0) p += b[i];
else {
q -= b[i];
}
}
cout << max(p, q) << endl;
cout << abs(p + q) + 1 << endl;
system("pause");
return 0;
}