寒假刷题14：Water Balance （贪心+单调栈）

题目链接：

water balance

题目解析：

最朴素的想法：对于每一个点，遍历它之后的每一个点，找到使  最小的点，将他们之间的点全赋值。

时间复杂度为O（n）,显然会超时。

很容易发现最终的序列是从前往后单调不减的，故可用单调栈维护。

 

AC代码：

#include <bits/stdc++.h>
using namespace std;
#define ll long long
struct str
{
    int l,r;
    ll sum;
}s[1000100];
int n, top=0, a[1000100];
double ans[1000100];
int main()
{
    scanf("%d", &n);
    for(int i = 1; i <= n; i++) scanf("%d", &a[i]);
    s[++top].l=n;
    s[top].r=n;
    s[top].sum=a[n];
    s[0].l = n + 1;
    for(int i=n-1;i>0;i--)
    {
        int tot = 1;
        ll temp=a[i];
        while(top && s[top].sum * tot <= temp * (s[top].r - s[top].l + 1))
        {
            temp += s[top].sum;
            tot += s[top].r - s[top].l + 1;
            top--;
        }
        s[++top].r = s[top - 1].l - 1;
        s[top].l = i;
        s[top].sum = temp;
    }
    for(int i = 1; i <= top; i++)
        for(int j = s[i].l; j <= s[i].r; j++) 
            ans[j] = 1.0 * s[i].sum / (s[i].r - s[i].l + 1);
    for(int i = 1; i <= n; i++) printf("%.10lf\n", ans[i]);
    return 0;
}