增减序列问题

目录

题目描述

样例

题目解析 

代码

---

题目链接：100. 增减序列 - AcWing题库

题目描述

给定一个长度为 nn 的数列 a1,a2,…,an，每次可以选择一个区间 [l,r]，使下标在这个区间内的数都加一或者都减一。

求至少需要多少次操作才能使数列中的所有数都一样，并求出在保证最少次数的前提下，最终得到的数列可能有多少种。

输入格式

第一行输入正整数 n。

接下来 n 行，每行输入一个整数，第 i+1 行的整数代表 ai。

输出格式

第一行输出最少操作次数。

第二行输出最终能得到多少种结果。

数据范围

0<n≤105,
0≤ai<2147483648

样例

输入样例：

4
1
1
2
2

输出样例：

1
2

题目解析 

题目中说对一段区间加减，我们首先选用差分。在查分数组中，最终结果为b[1]可以取任意数，b[2]到b[n]都为0，b[n+1]为任意数，这样就能保证a序列中所有的数全部相同。在任意区间上的操作可以分为以下四类：

 我们可以从上面四种情况中可以发现，我们不需要第四种操作。我们先统计查分数组从2到n中所有正数加起来的大小，在计算所有负数加起来的大小，先通过一操作使得一方消为零，在通过操作2或3使得另一方也为0，这样b中2到n上就都为0了。所以最少次数为max(|q|,|p|)，最终数列的可能种类为min(|q|，|p|)+||p|-|q||=max(|q|，|p|)。

代码

#include<iostream>
#include<algorithm>
#include<cstring>

using namespace std;

typedef long long LL;

const int N = 1e5+10;
int a[N],b[N];
int n;

void insert(int l,int r,int x)
{
    b[l]+=x,b[r+1]-=x;
}

int main()
{
    cin>>n;
    for(int i=1;i<=n;i++) cin>>a[i];
    for(int i=1;i<=n;i++) insert(i,i,a[i]);
    
    LL q=0,p=0;
    for(int i=2;i<=n;i++)
        if(b[i]>0) q+=b[i];
        else if(b[i]<0) p-=b[i];
    cout<<max(p,q)<<endl<<abs(p-q)+1<<endl;
    
    return 0;
}