BZOJ2800: [Poi2012]Leveling Ground

题目大意：给定n个数和a,b每次可以选择一段区间+a,-a,+b或-b，问最少操作几次能把他们都变成0

首先应该能想到差分，这样就相当于在n+1个数每次选两个数，一个+a一个-a（或b），最后把他们都变成0

首先我们对于每个差分之后的B[i]先用exgcd构造出一组用a,b表示的解，即a*x[i]+b*y[i]=B[i]

最后这个点的解一定形如X[i]=x[i]+k*b,Y[i]=y[i]-k*a

这样最终答案就可以用(∑|X[i]|+∑|Y[i]|)/2来表示

然后我们可以考虑贪心求出每个点的最优的k，也就是让这一步可以用三分完成

我们这时得到的一定是一个非常小的答案，但是他不一定符合题意，也就是说其中的+a数并不一定等于-a数，所以我们要进行一些调整

我们可以把所有的数全部推进堆里，每次挑选一个修改之后代价更小的来修改，直到+a数等于-a数，这时+b数一定也等于-b数，所以这一定是最小合法答案

具体实现可以看代码

 

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#include<vector>
#define N 100010
using namespace std;
long long gcd(long long a,long long b)
{
	if(!a) return b;
	return gcd(b%a,a);
}
long long A[N],B[N],X[N],Y[N];
void exgcd(long long a,long long b,long long &x,long long &y)
{
	if(a==0){x=0;y=1;return;}
	exgcd(b%a,a,x,y);
	long long t=x;
	x=y-b/a*x;
	y=t;
}
struct ppp{long long w,v;};
bool operator <(const ppp &x,const ppp &y){return x.v>y.v;}
priority_queue<ppp>P;
int main()
{
	long long n,a,b;
	scanf("%lld%lld%lld",&n,&a,&b);
	long long p=gcd(a,b);
	a/=p;b/=p;
	long long i,j,x,y;
	for(i=1;i<=n;i++)
	scanf("%lld",&A[i]);
	n++;
	exgcd(a,b,X[0],Y[0]);
	for(i=1;i<=n;i++)
	{
		B[i]=A[i]-A[i-1];
		if(B[i]%p!=0) {puts("-1");return 0;}
		B[i]/=p;
		X[i]=B[i]*X[0];
		Y[i]=B[i]*Y[0];
	}
	if(a==b)
	{
		long long ans=0;
		for(i=1;i<=n;i++)
		ans+=abs(B[i]/a);
		printf("%lld",ans/2);
		return 0;
	}
	for(i=1;i<=n;i++)
	{
		long long l=-abs(B[i]),r=abs(B[i]),m1,m2,v1,v2;
		while(r-l>=3)
		{
			m1=l+(r-l)/3;
			m2=r-(r-l)/3;
			v1=abs(X[i]+m1*b)+abs(Y[i]-m1*a);
			v2=abs(X[i]+m2*b)+abs(Y[i]-m2*a);
			if(v1<v2) r=m2;
			else l=m1;
		}
		long long minn=1e18,minb=0;
		for(j=l;j<=r;j++)
		{
			v1=abs(X[i]+j*b)+abs(Y[i]-j*a);
			if(v1<minn)
			{
				minn=v1;
				minb=j;
			}
		}
		X[i]+=minb*b;Y[i]-=minb*a;
	}
	long long tmp=0;
	for(i=1;i<=n;i++)
	tmp+=X[i];
	if(tmp<0)
	{
		for(i=1;i<=n;i++)
		swap(X[i],Y[i]);
		swap(a,b);
		tmp=-tmp/a*b;
	}
	tmp=tmp/b;
	for(i=1;i<=n;i++)
	P.push((ppp){i,(abs(X[i]-b)+abs(Y[i]+a))-(abs(X[i])+abs(Y[i]))});
	while(tmp--)
	{
		i=P.top().w;
		P.pop();
		X[i]-=b;Y[i]+=a;
		P.push((ppp){i,(abs(X[i]-b)+abs(Y[i]+a))-(abs(X[i])+abs(Y[i]))});
	}
	long long ans=0;
	for(i=1;i<=n;i++)
	ans+=abs(X[i])+abs(Y[i]);
	printf("%lld",ans/2);
}