array
by ysy
【题目描述】
给定一个长度为n的数列,每次你可以进行以下操作之一:
(1)将一个数+a;
(2)将一个数-a;
(3)将一个数+b;
(4)将一个数-b;
你需要将所有数全部变为0,求最小操作数。
【输入数据】
第一行三个整数n,a,b,第二行n个整数x1~xn表示数列。
【输出数据】
一行一个整数表示答案。无解输出-1。
【样例输入】
2 2 3
1 2
【样例输出】
3
【数据范围】
对于10%的数据,n,a,b,|xi|<=1000。
对于30%的数据,n,a,b<=1000。
对于另外10%的数据,a=1。
对于另外10%的数据,a=2,b=3。
对于100%的数据,1<=n<=105,1<=a,b<=109,|xi|<=109。
【题解思路】
很容易转化成数学模型:ax+by = c,使(|x|+|y|)min。
对于方程ax+by = c,我们可以用exgcd求出一组解。
当a,b互质时,保证ax+by = c有解。
设d = gcd(a,b).a/d*x+b/d*y = c/d;
此时可求出一组特解:x',y'。
则ax+by = c的通解可以表示为:x = c/d * x' + k * b/d,y = c/d * y' - k * a/d;
然后如何使(|x|+|y|)min。考虑到对于上述通解,我们可以打表或意念理解,这是个单峰函数。
即存在唯一且确定值k,使得|c/d*x' + k*b/d|+|c/d*y' - k* a/d|最小,尽可能使绝对值接近零,那么对于这两个数使得x取得最小的正数或最大的负数(绝对值尽量接近0)。
时间复杂度 O(nlog|xi|)
![](https://i-blog.csdnimg.cn/blog_migrate/8f900a89c6347c561fdf2122f13be562.gif)
![](https://i-blog.csdnimg.cn/blog_migrate/961ddebeb323a10fe0623af514929fc1.gif)
#include<bits/stdc++.h> using namespace std; #define ll long long #define ull unsigned long long #define rep(k,i,j) for(int k = i;k <= j; ++k) #define FOR(k,i,j) for(int k = i;k >= j; --k) inline int read(){ int x=0,f=1; char ch=getchar(); while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();} while(ch>='0'&&ch<='9'){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();} return x*f; } int n,a,b,k; inline void exgcd(int a,int b,int m,ll &x,ll &y){ if(!b) x = m/a,y = 0; else { exgcd(b,a%b,m,x,y); swap(x,y); y -= a/b*x; } } inline int gcd(int a,int b){return b ? gcd(b,a%b) : a;} ll x,y,p; int main(){ freopen("array.in","r",stdin); freopen("array.out","w",stdout); n = read(),a = read(),b = read(); k = gcd(a,b); a /= k,b /= k; if(a<b) swap(a,b); rep(i,1,n){ int j = read(); if(j%k) printf("-1\n"),exit(0); exgcd(a,b,j/k,x,y); if(y<0) { x -= b*((-y)/a+1); y += a*((-y)/a+1); } x += b*(y/a); y -= a*(y/a); p += min(abs(x)+abs(y),abs(x+b)+abs(y-a)); } printf("%lld\n",p); return 0; } /* 2 2 3 1 2 */