cf1299C-Water Balance

题目

题目链接

内容

有n个数,你可以进行一种操作,选一段区间,将这段区间所有元素的值替换为这段区间的平均值.输出这n个数最小的字典序

分析

比赛时,我写的是并查集加优先队列.
这里讲解一下码量更少,复杂度更优秀的单调栈.我们用单调栈维护区间的元素数量和平均值
我们从前往后加入这n个数.每加入一个数,我们更新单调栈,使栈中元素的平均值,从上往下依次减小.

!!! 有一个奇怪的情况:在cf上提交时 c++11会超时,c++14和17都是1.9秒.

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> pii;
typedef vector<int> vi; 
#define pb push_back
#define mp make_pair
#define fi first
#define se second
#define rep(i,a,b) for(int i=(a);i<=(b);i++) 
#define per(i,a,b) for(int i=(a);i>=(b);i--)
const int maxn=1e6+10;
const int mod=1e9+7;
const int inf=0x3f3f3f3f;
double a[maxn];
int num[maxn],tp=0;
void update() {
	while(tp>1) {
		if(a[tp]<a[tp-1]) {
			a[tp-1]=(a[tp]*num[tp]+a[tp-1]*num[tp-1])/(num[tp]+num[tp-1]);
			num[tp-1]+=num[tp];
			tp--;
		}
		else break;
	}
}
int n;
int main() {
	ios::sync_with_stdio(false);cin.tie(0);
	cin>>n;
	rep(i,1,n) {
		int x;
		cin>>x;
		a[++tp]=x;
		num[tp]=1;
		update();
	}	
	rep(i,1,tp){
		rep(j,1,num[i]) cout<<fixed<<setprecision(10)<<a[i]<<'\n';
	}
	return 0;
}