自学内容网 自学内容网

【算法】增减序列(贪心,差分)

题目

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

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

输入格式

第一行输入正整数 n。

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

输出格式

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

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

数据范围

0 < n ≤ 1e5
0 ≤ ai < 2147483648

输入样例:

4
1
1
2
2

输出样例:

1
2

思路

假设我们有一个序列:9 8 7 10 11 12 4 5 ,第一步我们先求出差分数组,然后使得差分数组的值为0(第一位不需要变)

 sum1 = abs(所有负数之和)

sum2  = 所有正数之和 

sum1 与 sum2 均与差分数组第一位无关 

将差分数组除第一位全部变为0需要操作 max(sum1,sum2)次

差分数组的第一位的取值个数ans = max(sum1,sum2) - min(sum1,sum2) + 1;

代码 

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N = 1e5 + 10;
int n;
int a[N],b[N];
int p,q;

int32_t main()
{
    cin >> n;
    for(int i = 1; i <= n; i ++) cin >> a[i];
    for(int i = 1; i <= n; i ++) b[i] = a[i] - a[i - 1];
    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;
    return 0;
}

题目来自: 100. 增减序列 - AcWing题库

难度:中等
时/空限制:1s / 64MB
总通过数:16399
总尝试数:35129
来源:《算法竞赛进阶指南》
算法标签

题目来自:100. 增减序列 - AcWing题库


原文地址:https://blog.csdn.net/littlegengjie/article/details/135504691

免责声明:本站文章内容转载自网络资源,如本站内容侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!