【BZOJ2987】Earthquake

Description

给定a,b,c,求满足方程Ax+By<=C的非负整数解

A,B<=10^9.C<=Min(A,B)*10^9

Input

Output

Sample Input

3 4 13

Sample Output

12

HINT

Source

其实题目的表述有异议.意思是求解的个数.

对于某一个值为0的可以直接算

显然答案应当是

$$\sum_{x=1}^{n} \lfloor \frac{c-ax}{b}\rfloor$$

以下就是类欧的主要思想:

我们可以轻易的把$-ax$ 换成 $+ax$且$a\lt b$
若$b\leq c$也可以将$c$转化为$c\lt b$

$$\sum_{x=1}^{n} \lfloor \frac{c+ax}{b}\rfloor$$
$$=\sum_{x=1}^{n} \sum_{k} [kb\leq c+ax]$$
$$=\sum_{k} n-\lceil \frac{kb-c}{a}\rceil +1$$
$$=\sum_{k} n-\lfloor \frac{kb-c+a-1}{a}\rfloor +1$$

可以发现这变成了一个规模更小的问题
就可以递归求解了
复杂度是$O(logn)$

直接抄了学姐的公式来写题解感觉良心谴责max

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define LL long long
using namespace std;
LL calc(LL n,LL a,LL b,LL c)
{
    LL ret=0;
    if (!n) return ret;
    if (c<0)
    {
        LL t=(a-c-1)/a;
        ret-=t*n;c+=t*a;
    }
    if (c/a>0||b/a>0)
    {
        ret+=c/a*n;ret+=b/a*n*(n+1)/2;
        c%=a;b%=a;
    }
    LL newn=(b*n+c)/a;
    ret+=newn*(n+1);ret-=calc(newn,b,a,b-c-1);
    return ret;
}
int main()
{
    LL a,b,c,ans=0;cin>>a>>b>>c;
    LL minn=(c-a)/b,tmp=(b+a-1)/a;
    ans-=tmp*(minn+1)*minn/2;ans+=c/a+c/b+1;
    cout<<ans+calc(minn,a,tmp*a-b,c);
}