Codeforces Round #618 (Div. 1) C. Water Balance（思维+单调栈）

题目链接


题意：给出n个数，每次可以选任意区间，把这个区间的所有数都变成平均数，问怎么变能使得整个数组的字典序最小。。
思路：很明显，我们要先字典序最小的话首先要想办法把前面大的数给降下来，就是通过后面小的数一平均就可以降下来了，我们可以从后往前遍历，每次把变成平均数的那一段看成一个整体，维护一下就行。size存的是从当前为开始作为一整体的区间下标，意思也就是比如i=3和j=6是一个整体，那么size【i】和size【j】的下标都是6，也是为了优化时间复杂度。。。

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+10;
int n;
double a[N],size[N];
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;++i) scanf("%lf",&a[i]),size[i]=i;
	a[n+1]=N;
	for(int i=n;i>=1;--i)
	{
		double sum=a[i],cnt=1;
		int j=i+1;
		while(a[j]<a[i]){
			sum+=a[j]*(size[j]-j+1);
			cnt+=size[j]-j+1;
			a[i]=sum/cnt;
			size[i]=size[j];
			j=size[j]+1;
		}
	}
	int j=1;
	while(j<=n)
	{
		for(int i=j;i<=size[j];++i) a[i]=a[j];
		j=size[j]+1;
	}
	for(int i=1;i<=n;++i) printf("%.10lf\n",a[i]);
}