AcWing 100. IncDec序列
程序员文章站
2022-07-12 17:38:02
...
给定一个长度为 nn 的数列 a1,a2,…,ana1,a2,…,an,每次可以选择一个区间 [l,r],使下标在这个区间内的数都加一或者都减一。
求至少需要多少次操作才能使数列中的所有数都一样,并求出在保证最少次数的前提下,最终得到的数列可能有多少种。
输入格式
第一行输入正整数nn。
接下来nn行,每行输入一个整数,第i+1行的整数代表aiai。
输出格式
第一行输出最少操作次数。
第二行输出最终能得到多少种结果。
数据范围
0<n≤1050<n≤105,
0≤ai<21474836480≤ai<2147483648
输入样例:
4
1
1
2
2
输出样例:
1
2
解法:开两个数组a[n],b[n],a[n]用于存放,b[n]为a的差分,所以只需要使b2,b3,bn变成0即可等价为a1,a2,an全相等,同时操作可转化为bi+1,bj-1
或则bi+1,bj-0和bi+0,bj-1三种操作,所以可以得出操作的次数为:
ans = max(b中正数和,b中负数和);而不同的结果数只与后两种操作有关,则 num = abs(b中正数和 - b中负数和) + 1;
#include<bits/stdc++.h>
using namespace std;
int main()
{
int n;
cin >> n ;
long long a[n+1],b[n+1];
long long f = 0 , z = 0 ;
for(int i=0;i<n;i++)
{
cin >> a[i] ;
if(i == 0) b[i] = a[i];
else
{
b [i] = a[i] - a[i-1];
if(b[i] > 0) z = z + b[i];
else if(b[i]<0) f = f + b[i];
}
}
f = abs(f);
cout << max(z,f) << endl << abs(f-z) + 1;
}