BZOJ2987: Earthquake【类欧几里得】

2987: Earthquake

将式子移项得到$y\le \frac{-Ax+C}{B}$

答案就是${\sum }_{x=0}^n[\frac{-Ax+C}{B}+1]$

设$f(n,A,B,C)={\sum }_{x=0}^n[\frac{-Ax+C}{B}+1]$

当$B\le A$||$B\le C$

$f(n,A,B,C)=f(n,A\%B,B,C\%B)+[A/B]\frac{(n+1)n}{2}+[C/B](n+1)$

当$C\le 0$||$A\le 0$

$f(n,A,B,C)=f(n,A\%B+B,B,C\%B+B)+[A/B-1]\frac{(n+1)n}{2}+[C/B](n+1)$

#include<cstdio>
using namespace std;
typedef long long LL;
LL a,b,c,n;
LL Solve(LL n,LL A,LL B,LL C){
	if(!B) return 0;
	if(A<0||C<0) return (A/B-1)*n*(n+1)/2+(C/B-1)*(n+1)+Solve(n,A%B+B,B,C%B+B);
	if(A>=B||C>=B) return A/B*n*(n+1)/2+C/B*(n+1)+Solve(n,A%B,B,C%B);	
	int kmax=(A*n+C)/B;
	return n*kmax-Solve(kmax-1,B,A,B-C-1);
}
int main(){
	scanf("%lld%lld%lld",&a,&b,&c);
	printf("%lld\n",Solve(c/a,-a,b,c+b));
	return 0;
}