【入门】前缀最大值c++

描述

求一个数列的所有前缀最大值之和。
即：给出长度为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。

来源

动态规划

代码

#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;
}