USACO 2012 Mar Haybale Restacking 重排乾草

假設對於所有位置都只從其左邊一個位置搬運乾草過去，則可設ab[i]爲i號位接收到的乾草數量。ab[i]=a[i-1]-b[i-1]+ab[i-1];

根據ab[i]排序。

這裏要採用快速排序，不然超時。

ps：這個程序用不到long long int。不知道爲啥當時也許是本博主腦抽了都寫了long long int。

#include<cstdio>
#include<cstdlib>
#include<ctime>
long long int n,a[100001],b[100001],ab[100001],ans,tmp;
void swap(long long int &a,long long int &b)
{
	long long int tmp=a;
	a=b;
	b=tmp;
}
void sort(long long int l,long long int r)
{
	long long int key=ab[rand()%(r-l+1)+l];
	long long int i=l,j=r;
	while(i<=j)
	{
		while(ab[i]<key)
			i++;
		while(ab[j]>key)
			j--;
		if(i<=j)
		{
			swap(ab[i],ab[j]);
			swap(a[i],a[j]);
			swap(b[i],b[j]);
			i++;
			j--;
		}
	}
	if(l<j)
		sort(l,j);
	if(i<r)
		sort(i,r);
}
int main()
{
	srand(time(NULL));
	freopen("restack.in","r",stdin);
	freopen("restack.out","w",stdout);
	scanf("%lld",&n);
	for(int i=1;i<=n;i++)
	{
		scanf("%lld%lld",&a[i],&b[i]);
		ab[i]=a[i-1]-b[i-1]+ab[i-1];
	}
	ab[1]=a[n]-b[n]+ab[n];
	sort(1,n);
	if(n%2==0)
		tmp=ab[n/2];
	else
		tmp=ab[n/2+1];
	for(int i=1;i<=n;i++)
	{
		ans+=abs(ab[i]-tmp);
	}
	printf("%lld\n",ans);
	fclose(stdin);
	fclose(stdout);
	return 0;
}