#646. 【入门】前缀最大值

说明

求一个数列的所有前缀最大值之和。
即：给出长度为n的数列a[i],求出对于所有1<=i<=n，max(a[1],a[2],...,a[i])的和。
比如，有数列：666 304 692 188 596，前缀最大值为：666 666 692 692 692，和为3408。
对于每个位置的前缀最大值解释如下：对于第1个数666，只有一个数，一定最大；对于第2个数，求出前两个数的最大数，还是666；对于第3个数，求出前3个数的最大数是692……其余位置依次类推，最后求前缀最大值得和。

由于读入较大，数列由随机种子生成。
其中a[1]=x，a[i]=(379*a[i-1]+131)%997。

输入格式

一行两个正整数n,x，分别表示数列的长度和随机种子。(n<=100000,x<997)

输出格式

一行一个正整数表示该数列的前缀最大值之和。

样例

输入数据 1

5 666

输出数据 1

3408

提示

数列为{666,304,692,188,596}，前缀最大值为{666,666,692,692,692},和为3408。

这一道题相对这个单元算是简单的了

思路很容易出来：

1. 用 a[N] 数组存储数字（如果怕空间复杂度的话可以看我给的第二个版本），并且不断枚举a[i + 1]（a[i] = (379 * a[i - 1] + 131) % 997，题目里有）
2. 再开一个 sum[N] 数组存储每一次从 a[1]~a[i] 中的最大值（一样，可以去掉）
3. 最后用 ans 变量存答案，每一次遍历时更新 sum[i] 和 ans 间的最大值即可

最后是两个版本，时间复杂度都为 O(n)，空间复杂度后者更佳（老手看看，新人参考）

AC code

（顺便说一句，拿走代码前点个赞再走呗~）

方案一：

#include <bits/stdc++.h>

using namespace std;
const int mod = 997;
const int N = 100005;
int n, ans, a[N], sum[N];

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);

    cin >> n >> a[1];
    sum[1] = a[1];
    //或者在此处:ans += a[1];

    for (int i = 2; i <= n; i++)
    {
        a[i] = (379 * a[i - 1] + 131) % mod;
        sum[i] = max(sum[i - 1], a[i]);
        ans += sum[i];
    }

    cout << (ans + a[1]);

    return 0;
}

Copy

方案二：（空间复杂度优化，但有亿点麻烦）

#include <bits/stdc++.h>

using namespace std;
const int mod = 997;
int n, ans;
int a1; //相当于 a[i - 1]
int a2; //当前的 a[i]
int b1; //相当于 sum[i - 1]
int b2; //当前的 sum[i]

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);

    cin >> n >> a1;
    b1 = a1;
    ans += a1;

    for (int i = 2; i <= n; i++)
    {
        a2 = (379 * a1 + 131) % mod;
        b2 = max(b1, a2);
        ans += b2;

        //更新
        a1 = a2;
        b1 = b2;
    }

    cout << ans;

    return 0;
}

Copy

完结，撒花 ✿✿ヽ(°▽°)ノ✿~