upc组队赛1 小C的数学问题【单调栈】(POJ2796)

时间：2022-12-30 20:53:45

小C的数学问题

题目描述

小C是个云南中医学院的大一新生，在某个星期二，他的高数老师扔给了他一个问题。

让他在1天的时间内给出答案。

但是小C不会这问题，现在他来请教你。

请你帮他解决这个问题。

有n个数，每个数有权值。

数学老师定义了区间价值为区间和乘上区间内的最小值。

现在要你找出有最大区间价值的区间是什么，并输出区间价值。

输入

每个输入文件只包含单组数据。

第一行一个整数n。(1 <= n <= 100000)

第二行n个整数a_1,a_2,...,a_n。(0 <= a_i <= 1000000)

输出

第一行输出一个整数，表示最大的区间价值。

第二行输出两个整数，表示区间的起点和终点。

保证答案唯一。

样例输入

6

10 1 9 4 5 9

样例输出

108

3 6

题解

原题POJ2796 用单调栈维护区间的最小值，复杂度就优化到了O(N)

代码

#include<iostream>

#include<cstdio>     //EOF,NULL

#include<cstring>    //memset

#include<cstdlib>    //rand,srand,system,itoa(int),atoi(char[]),atof(),malloc

#include<cmath>           //ceil,floor,exp,log(e),log10(10),hypot(sqrt(x^2+y^2)),cbrt(sqrt(x^2+y^2+z^2))

#include<algorithm>  //fill,reverse,next_permutation,__gcd,

#include<string>

#include<vector>

#include<queue>

#include<stack>

#include<utility>

#include<iterator>

#include<iomanip>             //setw(set_min_width),setfill(char),setprecision(n),fixed,

#include<functional>

#include<map>

#include<set>

#include<limits.h>     //INT_MAX

#include<cmath> // bitset<?> n

using namespace std;

#define rep(i,a,n) for(int i=a;i<n;i++)

#define per(i,a,n) for(int i=n-1;i>=a;i--)

#define fori(x) for(int i=0;i<x;i++)

#define forj(x) for(int j=0;j<x;j++)

#define memset(x,y) memset(x,y,sizeof(x))

#define memcpy(x,y) memcpy(x,y,sizeof(y))

#define all(x) x.begin(),x.end()

#define readc(x) scanf("%c",&x)

#define read(x) scanf("%d",&x)

#define read2(x,y) scanf("%d%d",&x,&y)

#define read3(x,y,z) scanf("%d%d%d",&x,&y,&z)

#define print(x) printf("%d\n",x)

#define lowbit(x) x&-x

#define lson(x) x<<1

#define rson(x) x<<1|1

#define pb push_back

#define mp make_pair

#define N 100001

typedef pair<int,int> P;

typedef long long LL;

typedef long long ll;

const double eps=1e-8;

const double PI = acos(1.0);

const int INF = 0x3f3f3f3f;

const int inf = 0x3f3f3f3f;

const int mod = 1e9+7;

const int MAXN = 1e6+7;

const int maxm = 1;

const int maxn = 100000 + 10;

const ll MOD = 998244353;

int a[maxn];

int pos[maxn],st[maxn];

ll sum[maxn];

ll ans,tmp;

int n;

int main()

{

    read(n);

    memset(sum, 0);

    for (int i = 0; i < n; i++){

        read(a[i]);

        sum[i] = sum[i - 1] + a[i]; // 前缀和 方便求区间的总和

    }

    a[n] = -1;

    ans = -1;

    int top = 0;

    int l, r;

    for (int i = 0; i <= n; i++){

        if (top == 0 || a[i] > a[st[top - 1] ]){ //如果比栈顶大 就入栈

            st[top++] = i;//单调栈的顶部

            pos[i] = i; //储存当前栈最小值延续到的左边界

        }

        else{

          while (top >= 1 && a[i] <= a[st[top - 1]]){ //如果ai比栈顶小,弹出所有比ai大的数

              --top;

              tmp = a[ st[top] ] * (sum[i - 1] - sum[pos[st[top]]-1]);

           //总价值 =  最小值（栈顶元素）  *  区间和（前缀和相减）

              if (ans < tmp){

                  l = pos[st [top] ] + 1;

                  r = i;

                  ans = tmp;

              }

          }

          pos[i] = pos[st[top]];  // 以ai为最小值的区间左边界是他自己，从自己开始

          st[top++] = i; //入栈

        }

    }

    printf("%lld\n", ans);

    printf("%d %d\n", l, r);

    return 0;

}

相关文章