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

POJ2796 Feel Good(单调栈)

时间:2016-02-28 19:57:26      阅读:177      评论:0      收藏:0      [点我收藏+]

标签:

题意:给一个非负整数序列,求哪一段区间的权值最大,区间的权值=区间所有数的和×区间最小的数。

用单调非递减栈在O(n)计算出序列每个数作为最小值能向左和向右延伸到的位置,然后O(n)枚举每个数利用前缀和O(1)计算出以这个数为最小值能得到的最大的区间权。

以前写的单调栈,三个if分支,写得繁繁杂杂的;现在重写了一下,感觉代码简洁了不少:

 1 #include<cstdio>
 2 using namespace std;
 3 #define MAXN 111111
 4 int stack[MAXN],top;
 5 int l[MAXN],r[MAXN];
 6 long long a[MAXN],s[MAXN];
 7 int main(){
 8     int n;
 9     while(~scanf("%d",&n)){
10         for(int i=1; i<=n; ++i) scanf("%d",a+i),s[i]=s[i-1]+a[i];
11         a[++n]=-1;
12         for(int i=1; i<=n; ++i){
13             l[i]=r[i]=i;
14             while(top && a[stack[top]]>a[i]){
15                 l[i]=l[stack[top]];
16                 r[stack[top]]=i-1;
17                 --top;
18             }
19             if(top && a[stack[top]]==a[i]) l[i]=l[stack[top]];
20             stack[++top]=i;
21         }
22         int x,y;
23         long long res=-1;
24         for(int i=1; i<n; ++i){
25             if(res<(s[r[i]]-s[l[i]-1])*a[i]){
26                 res=(s[r[i]]-s[l[i]-1])*a[i];
27                 x=l[i]; y=r[i];
28             }
29         }
30         printf("%lld\n%d %d\n",res,x,y);
31     }
32     return 0;
33 }

 

POJ2796 Feel Good(单调栈)

标签:

原文地址:http://www.cnblogs.com/WABoss/p/5225379.html

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