poj 2796 Feel Good

單調棧。

其實,單調隊列和單調棧是類似的,在我看來,這兩個東西只是名字不一樣 - - !

比較容易想的一道題啦!

首先,這題的兩個關鍵點:

1、區間的和。這個簡單,地球人都知道!

2、區間的最小值。 簡單的暴力每個長度的區間的話,是O(n^2),毫無疑問會TLE。再仔細考慮下這道題,對於每個值,把這值當做最小值的區間當然是越長越好,這樣很快地將問題轉換成求一個數能伸展到的最大區間,這個區間的最小值是這個數。此時可以自己YY到單調棧的一個很好的性質:一個數進棧,那麼他前面那個數就是向左第一個小於它的數,一個數出棧,那麼使它出棧的那個數就是向右第一個小於它的數。而到最後留在棧中的元素向右沒有比他小的數咯~ 

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <math.h>
#include <stdlib.h>
#include <algorithm>
#include <vector>
#include <limits.h>
#include <queue>
#include <stack>
using namespace std;
struct S
{
    int num,id;
}s[100010];
struct C
{
    int num,b,e;
}c[100010];
long long dp[100010];
int main()
{
    int a[100010],i,j,n,top;
    long long ans = -1,b,e,sum;
    while(scanf("%d",&n)!=EOF)
    {
        ans = -1;
        dp[0] = 0;
        for(i = 1;i <= n;i ++)
        {scanf("%d",&a[i]);c[i].num = a[i];dp[i] = dp[i-1]+a[i];}
        s[0].num = -1;s[0].id = 0;
        top = 0;
        for(i = 1;i <= n;i ++)
        {
            for(j = top;j >= 0;j --)
            {
                if(a[i] > s[j].num) break;
                else
                {
                    c[s[j].id].e = i-1;
                }
            }
            s[j+1].num = a[i];s[j+1].id = i;
            top = j+1;c[i].b = s[top-1].id+1;
            //for(j = 0;j <= top;j ++) cout<<s[j].num<<" ";cout<<endl;
        }
        for(j = 0;j <= top;j ++) c[s[j].id].e = n;
        for(i = 1;i <= n;i ++)
        {
            //cout<<c[i].num<<" "<<c[i].b<<" "<<c[i].e<<endl;
            sum = dp[c[i].e] - dp[c[i].b-1];
            sum *= c[i].num;
            if(sum > ans) {ans =sum;b = c[i].b;e = c[i].e;}

        }
        printf("%I64d\n%I64d %I64d\n",ans,b,e);
    }
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章