码迷,mamicode.com
首页 > 其他好文 > 详细

POJ 2796 Feel Good(单调栈)

时间:2020-04-17 23:19:06      阅读:62      评论:0      收藏:0      [点我收藏+]

标签:second   维护   mat   https   --   开区   它的   net   void   

题目链接
??这个题首先想象一下朴素的算法怎么做。想要知道一个区间的最大值,那么对于每一个以\(a_i\)为最小的元素的区间,它的左右端点都是比它小的数(为了方便我用开区间来描述)。所以只要以\(a_i\)的起点向左右扩展就行了,这样的复杂度是\(O(n^2)\)
??那么如何来优化呢?计算区间的和用前缀和就行了,区间的端点呢?可以用单调栈来解决。我们用单调栈来维护区间最小值,那么对当前元素来说,在它之前的一个最小值和在它之后的一个最小值就是区间的两个端点。所以我们用一个单调递增栈从前到后面扫一遍,就求出了\(a_i\)左边比\(a_i\)小的一个元素,同样倒着扫一遍就能求出另一边的元素。

const int maxn = 1e5+10;
ll arr[maxn], pre[maxn];
int n, l[maxn], r[maxn];
stack<int> sk;
int main(void) {
    scanf("%d", &n);
    for (int i = 1; i<=n; ++i) {
        scanf("%lld", &arr[i]);
        pre[i] += pre[i-1]+arr[i];
    }
    for (int i = 1; i<=n; ++i) {
        while(!sk.empty() && arr[sk.top()]>=arr[i]) sk.pop();
        l[i] = sk.empty() ? 1 : sk.top()+1;
        sk.push(i);
    }
    while(!sk.empty()) sk.pop();
    for (int i = n; i>=1; --i) {
        while(!sk.empty() && arr[sk.top()]>=arr[i]) sk.pop();
        r[i] = sk.empty() ? n : sk.top()-1;
        sk.push(i);
    }
    ll maxx = -1; P ans;
    for (int i = 1; i<=n; ++i)
        if (maxx < (ll)arr[i]*(pre[r[i]]-pre[l[i]-1])) {
            maxx = (ll)arr[i]*(pre[r[i]]-pre[l[i]-1]);
            ans = P(l[i], r[i]);
        }
    cout << maxx << endl << ans.first << ‘ ‘ << ans.second << endl; 
    return 0;
}

POJ 2796 Feel Good(单调栈)

标签:second   维护   mat   https   --   开区   它的   net   void   

原文地址:https://www.cnblogs.com/shuitiangong/p/12723156.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!