Honk's pool

Honk’s pool

思路

每次只对数组中 i，j 下标的 pool 进行操作，实际上是对当下及之前的操作，下标所标记的只是显示作用，无需对之前的 pool 进行 +1 操作，以节省时间。

题解

双指针从首尾依次向中间移动就好，注意有时候靠后的指针使用的平均值需要加一。

#include <iostream>
#include <cstdio>
#include <string>
#include <stack>
#include <cstring>
#include <algorithm>
using namespace std;
#define ll long long
int a[500100];
int main()
{
    int n,k;
	while(scanf("%d%d",&n,&k)!=EOF)
	{
		ll sum=0;
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&a[i]);
			sum=sum+(ll)a[i];
		}
		int aven;
		int ave=sum/(ll)n;
		if(ave*n!=sum){
			aven=ave+1;
		}else{
			aven=ave;
		}
		sort(a+1,a+n+1);
		//kl可用加水次数，kr可用取水次数 
		int kl=k,kr=k;
		//每次对add个坑加或取水 
		int addl=1,addr=1;
		//记录到达位置 
		int i,j;
		for(i=1;a[i]<ave && kl>=addl;)
		{
			if(a[i]==a[i+1])
			{
				addl++;
				i++;
				continue;
			}
			a[i]++;
			kl-=addl;
		}
		for(j=n;a[j]>aven && kr>=addr;)
		{
			if(a[j]==a[j-1])
			{
				addr++;
				j--;
				continue;
			}
			a[j]--;
			kr-=addr;
		}
		//for(int i=1;i<=n;i++)
		//cout<<a[i]<<endl; 
		printf("%d\n",a[j]-a[i]);
	} 
	return 0;
}