欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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;
}
相关标签: 前缀差分