CF D. Minimize the Difference

原题链接:Problem - D - Codeforces

题意:给你长度为n的数组,可以无限次的让i位置的数-1,让i+1的位置的数+1。问最大值-最小值的最小值是多少?

思路:可以观察出,操作的真正意义是让i位置的数减少,i后面的数增加,如果p[i+1]<p[i]那么操作一定是优秀的,从感性的角度理解,就是让数组变得平均,但是直接模拟,会导致超时。观察可以知道,最后形成的数组,一定是单调的,后面大于前面,并且有很多相同的数,因为要处理这些相同的数,所以导致超时,那么就可以用栈来优化这个过程。栈存放的元素类型是pair,第一维放值,第二维放数量。对于每一个数,初始sum=p[i],cnt=1,如果sum/cnt<=栈顶的值,那么就可以减少栈顶的值,让当前的值增加,也就是sum+栈顶的值*数量,cnt+数量。这样也就是总和为sum,然后分到cnt个位置的问题,因为要尽可能的平均,那么就如果sum能整除cnt,那么就往栈里面放入sum/cnt和cnt,如果不能整除,那么就放入sum/cnt,cnt-sum%cnt和sum/cnt+1,sum%cnt。

为什么是这样,可以看下面的图片。

//冷静,冷静,冷静
//调不出来就重构
//#pragma GCC optimize(2)
//#pragma GCC optimize("O3")
#include<bits/stdc++.h>
#define endl '\n'
#define count2(x) __builtin_popcountll(x)
#define is2(x) __builtin_ffsll(x)
using namespace std;
typedef long long ll;
typedef long double ld;
typedef pair<ll,ll> pii;
const int N=1e6+10,mod=1000000007;
ll p[N];
void Jiuyuan()
{
	ll n;
	cin>>n;
	ll mx=0;
	for(int i=1;i<=n;i++)
	{
		cin>>p[i];
	}
	stack<pii> st;
	for(int i=1;i<=n;i++)
	{
		ll sum=p[i],cnt=1;
		while(st.size()&&st.top().first>=sum/cnt)
		{
			sum+=st.top().first*st.top().second;
			cnt+=st.top().second;
			st.pop();
		}
		st.push({sum/cnt,cnt-sum%cnt});
		if(sum%cnt)
		{
			st.push({sum/cnt+1,sum%cnt});
		}
	}
	ll max1=0,min1=1e18;
	while(st.size())
	{
		max1=max(max1,st.top().first);
		min1=min(min1,st.top().first);
		st.pop();
	}
	cout<<max1-min1<<endl;
}
int main()
{
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	ll T=1;
	cin>>T;
	while(T--)
	{
		Jiuyuan();
	}
    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值