【算法讲28：类欧几里得算法求和】洛谷 P5170

溢流眼泪

于 2021-07-21 14:16:34 发布

阅读量287

点赞数 1

分类专栏：【算法/知识点浅谈】文章标签：算法

本文链接：https://blog.csdn.net/weixin_45775438/article/details/118963660

版权

【算法/知识点浅谈】专栏收录该内容

65 篇文章 10 订阅

订阅专栏

这篇博客介绍了如何利用类欧几里得算法解决一类求和问题，涉及f(a,b,c,n)、g(a,b,c,n)和h(a,b,c,n)三个函数的推导及优化。通过递归和边界条件处理，实现了复杂度的降低，但原始实现仍导致超时（TLE），最后提出了结构体优化方案，将时间复杂度降至O(Tlogn)。

摘要由CSDN通过智能技术生成

P5170 【模板】类欧几里得算法
 知乎：一类求和问题
 想看具体的推导与公式满天飞的可以看这里

引入

给定 $n, a, b, c$ ，求
$\sum_{i=0}^n\Big\lfloor \frac{ai+b}{c} \Big\rfloor \\ \sum_{i=0}^n\Big\lfloor \frac{ai+b}{c} \Big\rfloor^2 \\ \sum_{i=0}^n i\Big\lfloor \frac{ai+b}{c} \Big\rfloor \\$
样例数 $1\le T\le 10^5$
$0\le n,a,b,c\le 10^9$

记号

我们记一下这三个函数：
$f(a,b,c,n)=\sum_{i=0}^n\Big\lfloor \frac{ai+b}{c} \Big\rfloor\\ h(a,b,c,n)=\sum_{i=0}^n\Big\lfloor \frac{ai+b}{c} \Big\rfloor^2\\ g(a,b,c,n)=\sum_{i=0}^n i\Big\lfloor \frac{ai+b}{c} \Big\rfloor \\$

$f (a, b, c, n)$ 的推导

当 $a\ge c$ 或 $\ge c$ 时，我们有
$\begin{aligned} f(a,b,c,n)&=\sum_{i=0}^n \Big \lfloor \frac{(a - a \bmod c)*i + (a \bmod c) * i + b}{c} \Big \rfloor\\ &=f(a\%c,b\%c,c,n)+\frac{n(n+1)}{2} \lfloor\frac{a}{c}\rfloor + (n+1)\lfloor\frac{b}{c}\rfloor \end{aligned}$
当 $a < c$ 且 $b < c$ 时，设 $m=\lfloor\frac{an+b}{c}\rfloor$
$\begin{aligned} f(a,b,c,n)&=\sum_{i=0}^n\sum_{j=1}^m\Big[ \lfloor \frac{ai+b}{c} \rfloor \ge j\Big]\\ &=\sum_{i=0}^n\sum_{j=0}^{m-1}\Big[ \lfloor \frac{ai+b}{c} \rfloor \ge j+1\Big]\\ &=\sum_{i=0}^n\sum_{j=0}^{m-1}\Big[ i > \frac{jc+c-b-1}{a}\Big]\\ &=\sum_{j=0}^{m-1}n-\Big\lfloor \frac{jc+c-b-1}{a} \Big\rfloor\\ &=nm-f(c,c-b-1,a,m-1) \end{aligned}$
边界条件 $a = 0$

$g (a, b, c, n)$ 的式子

当 $a\ge c$ 或 $\ge c$ 时，有
$\begin{aligned} g(a,b,c,n)&=g(a\%c,b\%c,c,n)+n(n+1)(2n+1)/6 \lfloor\frac{a}{c}\rfloor\\ &+n(n+1)/2 \lfloor\frac{b}{c}\rfloor \end{aligned}$
当 $a < c$ 且 $b < c$ 时，有
$\begin{aligned} g(a,b,c,n)&=\frac{mn(n+1)-f(c,c-b-1,a,m-1)-h(c,c-b-a,a,m-1)}{2} \end{aligned}$

$h (a, b, c, n)$ 的式子

当 $a\ge c$ 或 $\ge c$ 时，得到：
$\begin{aligned} h(a,b,c,n)&=h(a\%c,b\%c,c,n)+\frac{n(n+1)(2n+1)}{6}\lfloor\frac{a}{c}\rfloor^2\\ &+2\lfloor\frac{b}{c}\rfloor f(a\%c,b\%c,c,n)\\ &+2\lfloor\frac{a}{c}\rfloor g(a\%c,b\%c,c,n)\\ &+\lfloor\frac{a}{c}\rfloor\lfloor\frac{b}{c}\rfloor n(n+1)\\ &+\lfloor\frac{b}{c}\rfloor^2(n+1) \end{aligned}$
当 $a < c$ 且 $b < c$ 时
$\begin{aligned} h(a,b,c,n)&=m(m+1)n-2g(c,c-b-1,a,m-1)\\ &-2f(c,c-b-1,a,m-1)\\ &-f(a,b,c,n) \end{aligned}$

回到题目上来

然后把式子一放，总能过了吧？

ll f(ll,ll,ll,ll);      //   floor[ (ai+b)/c ]
ll g(ll,ll,ll,ll);      // i floor[ (ai+b)/c ]
ll h(ll,ll,ll,ll);      // [ floor[ (ai+b)/c ] ]^2


ll f(ll a,ll b,ll c,ll n){
    if(a == 0)
        return (n + 1) * (b / c) % MOD;
    if(a >= c || b >= c){
        return (f(a%c,b%c,c,n) + n * (n + 1) % MOD * iv2 % MOD * (a / c) + (n + 1) * (b / c) % MOD) % MOD;
    }else{
        ll m = (a * n + b) / c;
        return (n * m % MOD - f(c,c-b-1,a,m-1) + MOD) % MOD;
    }
}


ll g(ll a,ll b,ll c,ll n){
    if(a == 0)
        return n * (n + 1) % MOD * iv2 % MOD * (b/c) % MOD;
    if(n == 0)return 0;
    if(a >= c || b >= c){
        return (g(a%c,b%c,c,n) + n * (n + 1) % MOD * (2*n+1) % MOD * iv6 % MOD * (a/c) % MOD + n * (n + 1) % MOD * iv2 % MOD * (b/c) % MOD ) % MOD;
    }else{
        ll m = (a * n + b) / c;
        return (n * m % MOD * (n + 1) % MOD - f(c,c-b-1,a,m-1) - h(c,c-b-1,a,m-1) + 2 * MOD) % MOD * iv2 % MOD;
    }
}

ll h(ll a,ll b,ll c,ll n){
    if(a == 0)
        return (n+1) * (b/c) % MOD * (b/c) % MOD;
    if(n == 0)return (b/c) * (b/c) % MOD;
    if(a >= c || b >= c){
        return (h(a%c,b%c,c,n) + n * (n + 1) % MOD * (2*n+1) % MOD * iv6 % MOD * (a/c) % MOD * (a/c) % MOD + 2 * (b / c) % MOD * f(a%c,b%c,c,n) % MOD + 2 * (a / c) % MOD * g(a%c,b%c,c,n) % MOD + (a/c) * (b/c) % MOD * n % MOD * (n+1) % MOD + (b/c) * (b/c) % MOD * (n + 1) % MOD ) % MOD;
    }else{
        ll m = (a * n + b) / c;
        return (n * m % MOD * (m + 1) % MOD - 2 * g(c,c-b-1,a,m-1) - 2 * f(c,c-b-1,a,m-1) - f(a,b,c,n)) % MOD;
    }
}

结果不大行，都 $\color{red}TLE$ 了，得优化一下
我们用一个结构体去存 $Query\{F,G,H\}$ 表示目前的 $f (a, b, c, n), g (a, b, c, n), h (a, b, c, n)$ 的值
然后就变成了这样：
时间复杂度： $O(T\log n)$

struct Query{
    ll f,g,h;
};
Query solve(ll a,ll b,ll c,ll n){
    Query ans,tmp;
    if(a == 0){
        ans.f = (n + 1) * (b / c) % MOD;
        ans.g = n * (n + 1) % MOD * iv2 % MOD * (b/c) % MOD;
        ans.h = (n+1) * (b/c) % MOD * (b/c) % MOD;
        return ans;
    }
    if(a >= c || b >= c){
        tmp = solve(a % c,b % c,c,n);
        ans.f = (tmp.f + (a / c) * n % MOD * (n + 1) % MOD * iv2 % MOD + (b / c) * (n + 1) % MOD) % MOD;
        ans.g = (tmp.g + (a / c) * n % MOD * (n + 1) % MOD * (2 * n + 1) % MOD * iv6 % MOD + (b / c) * n % MOD * (n + 1) % MOD * iv2 % MOD) % MOD;
        ans.h = ((a / c) * (a / c) % MOD * n % MOD * (n + 1) % MOD * (2 * n + 1) % MOD * iv6 % MOD +
                (b / c) * (b / c) % MOD * (n + 1) % MOD + (a / c) * (b / c) % MOD * n % MOD * (n + 1) % MOD +
                tmp.h + 2 * (a / c) % MOD * tmp.g % MOD + 2 * (b / c) % MOD * tmp.f % MOD) % MOD;
        return ans;
    }
    ll m = (a * n + b) / c;
    tmp = solve(c,c-b-1,a,m-1);
    ans.f = (n * (m % MOD) % MOD - tmp.f) % MOD;
    ans.g = (n * (n + 1) % MOD * (m % MOD) % MOD - tmp.f - tmp.h) % MOD * iv2 % MOD;
    ans.h = (n * (m % MOD) % MOD * ((m + 1) % MOD) % MOD - 2 * tmp.g - 2 * tmp.f - ans.f) % MOD;
    return ans;
}