类欧几里得算法学习记录

原题链接1

题目描述相当简单，给你$T$组询问

求出

$$\sum _{x=0}^n{x}^{k_1}{\lfloor \frac{ax+b}{c}\rfloor }^{k_2}$$

对$1e9+7$取模的结果

先将问题给简化一下，设$$f(a,b,c,n)=\sum _{x=0}^n\lfloor \frac{ax+b}{c}\rfloor$$

如果$a\ge c$或者$b\ge c$，这样的话可以将$a$和$b$对$c$取模用来简化问题:

$$f(a,b,c,n)=\sum _{x=0}^n\lfloor \frac{ax+b}{c}\rfloor$$

$$=\sum _{x=0}^n\lfloor \frac{\left(\lfloor \frac{a}{c}\rfloor c+a\mathrm{m}\mathrm{o}\mathrm{d}c\right)x+\left(\lfloor \frac{b}{c}\rfloor c+b\mathrm{m}\mathrm{o}\mathrm{d}c\right)}{c}\rfloor$$

$$=\frac{n(n+1)}{2}\lfloor \frac{a}{c}\rfloor +(n+1)\lfloor \frac{b}{c}\rfloor +\sum _{x=0}^n\lfloor \frac{(a\mathrm{m}\mathrm{o}\mathrm{d}c)x+(b\mathrm{m}\mathrm{o}\mathrm{d}c)}{c}\rfloor$$

$$=\frac{n(n+1)}{2}\lfloor \frac{a}{c}\rfloor +(n+1)\lfloor \frac{b}{c}\rfloor +f(a\mathrm{m}\mathrm{o}\mathrm{d}c,b\mathrm{m}\mathrm{o}\mathrm{d}c,c,n)$$

接下来考虑$a<b$,$b<c$的情况。

观察式子你会发现，只存在$x$这个变量。所以要从$x$下手。

进行条件与贡献的转化。

例如这个式子：
$$f(a,b,c,n)=\sum _{x=0}^n\lfloor \frac{ax+b}{c}\rfloor$$

其中$0\le x\le n$是条件，$\lfloor \frac{ax+b}{c}\rfloor$则是总贡献

接下来就是贡献合并计算，但是这个式子的贡献相当难合并。

此时就是要进行一定的转化。

$$\sum _{x=0}^n\lfloor \frac{ax+b}{c}\rfloor =\sum _{x=0}^n\sum _{y=0}^{\lfloor \frac{ax+b}{c}\rfloor -1}1$$

这样的话就多了一个变量$y$，设法求出$y$的贡献。

再对这个公式进行一定变换的得到

$$=\sum _{y=0}^{\lfloor \frac{an+b}{c}\rfloor -1}\sum _{x=0}^n\left[y<\lfloor \frac{ax+b}{c}\rfloor \right]$$
原先$n$限制了$x$的上界，而$x$限制了$y$的上界。改变为了把$y$放到贡献的式子里。

这么做是为了让$y$不被$x$限制，如此使得$x$，$y$都被$n$限制。

接下来把向下取整的符号去掉

$$y<\lfloor \frac{ax+b}{c}\rfloor \leftrightarrow y+1\le \lfloor \frac{ax+b}{c}\rfloor \leftrightarrow y+1\le \frac{ax+b}{c}$$

再做一定的变换
$$y+1\le \frac{ax+b}{c}\leftrightarrow yc+c\le ax+b\leftrightarrow yc+c-b-1<ax$$

最后得到
$$yc+c-b-1<ax\leftrightarrow \lfloor \frac{yc+c-b-1}{a}\rfloor <x$$

此时$x$就被我们消掉了

设$m=\lfloor \frac{an+b}{c}\rfloor$

得到原式为$$f(a,b,c,n)=\sum _{y=0}^{m-1}\sum _{x=0}^n\left[x>\lfloor \frac{yc+c-b-1}{a}\rfloor \right]$$

$$=\sum _{y=0}^{m-1}n-\lfloor \frac{yc+c-b-1}{a}\rfloor$$

$$=nm-f(c,c-b-1,a,m-1)$$

会发现$a$,$c$分子分母互换了位置，再重复上述过程。取模，递归。辗转相除。

时间复杂度为$O(\log n)$

接下来考虑两个变形

$$1.g(a,b,c,n)=\sum _{x=0}^{n}x\lfloor \frac{ax+b}{c}\rfloor$$

$$2.h(a,b,c,n)=\sum _{x=0}^n{\lfloor \frac{ax+b}{c}\rfloor }^2$$

对于第一条，与之前类似，先取模。

$$g(a,b,c,n)=g(a\mathrm{m}\mathrm{o}\mathrm{d}c,b\mathrm{m}\mathrm{o}\mathrm{d}c,c,n)+\lfloor \frac{a}{c}\rfloor \frac{n(n+1)(2n+1)}{6}+\lfloor \frac{b}{c}\rfloor \frac{n(n+1)}{2}$$

接下来考虑$a<c$,$b<c$的情况，设$m=\lfloor \frac{an+b}{c}\rfloor$

$$g(a,b,c,n)=\sum _{x=0}^{n}x\lfloor \frac{ax+b}{c}\rfloor$$

$$=\sum _{y=0}^{m-1}\sum _{x=0}^n\left[y<\lfloor \frac{ax+b}{c}\rfloor \right]\cdot x$$

令$$t=\lfloor \frac{yc+c-b-1}{a}\rfloor$$

可以得到：
$$\sum _{y=0}^{m-1}\sum _{x=0}^n[x>t]\cdot x$$

$$=\sum _{y=0}^{m-1}\frac{1}{2}(t+n+1)(n-t)$$

$$=\frac{1}{2}\left[mn(n+1)-\sum _{y=0}^{m-1}{t}^2-\sum _{y=0}^{m-1}t\right]$$

$$=\frac{1}{2}\left[mn(n+1)-h(c,c-b-1,a,m-1)-f(c,c-b-1,a,m-1)\right]$$

对于$h$，同样，先取模。
$$h(a,b,c,.n)=h(a\mathrm{m}\mathrm{o}\mathrm{d}c,b\mathrm{m}\mathrm{o}\mathrm{d}c,c,n)$$

$$+2\lfloor \frac{b}{c}\rfloor f(a\mathrm{m}\mathrm{o}\mathrm{d}c,b\mathrm{m}\mathrm{o}\mathrm{d}c,c,n)$$

$$+2\lfloor \frac{a}{c}\rfloor g(a\mathrm{m}\mathrm{o}\mathrm{d}c,b\mathrm{m}\mathrm{o}\mathrm{d}c,c,n)$$

$$+{\lfloor \frac{a}{c}\rfloor }^2\frac{n(n+1)(2n+1)}{6}$$

$$+{\lfloor \frac{b}{c}\rfloor }^2(n+1)+\lfloor \frac{a}{c}\rfloor \lfloor \frac{b}{c}\rfloor n(n+1)$$

对于$a<c$,$b<c$,$m=\lfloor \frac{an+b}{c}\rfloor ,t=\lfloor \frac{yc+c-b-1}{a}\rfloor$

在对平方进行处理：

$$n^2=2\frac{n(n+1)}{2}-n=\left(2\sum _{x=0}^{n}x\right)-n$$

这样的话再添加$y$的时候就成了求和，不用处理$\sum \times \sum$的式子

$$h(a,b,c,n)=\sum _{x=0}^n{\lfloor \frac{ax+b}{c}\rfloor }^2=\sum _{x=0}^n\left[\left(2\sum _{y=1}^{\lfloor \frac{ax+b}{c}\rfloor }y\right)-\lfloor \frac{ax+b}{c}\rfloor \right]$$

$$=\left(2\sum _{x=0}^n\sum ^{\lfloor \frac{ax+b}{c}\rfloor }y\right)-f(a,b,c,n)$$

再化简前一部分：

$$\sum _{x=0}^n\sum _{y=1}^{\lfloor \frac{ax+b}{c}\rfloor }y$$

$$=\sum _{x=0}^n\sum _{y=0}^{\lfloor \frac{ax+b}{c}\rfloor -1}(y+1)$$

$$=\sum _{y=0}^{m-1}(y+1)\sum _{x=0}^n\left[y<\lfloor \frac{ax+b}{c}\rfloor \right]$$

$$=\sum _{y=0}^{m-1}(y+1)\sum _{x=0}^n[x>t]$$

$$=\sum _{y=0}^{m-1}(y+1)(n-t)$$

$$=\frac{1}{2}nm(m+1)-\sum _{y=0}^{m-1}(y+1)\lfloor \frac{yc+c-b-1}{a}\rfloor$$

$$=\frac{1}{2}nm(m+1)-g(c,c-b-1,a,m-1)-f(c,c-b-1,a,m-1)$$

所以：

$$h(a,b,c,n)=nm(m+1)-2g(c,c-b-1,a,m-1)-2f(c,c-b-1,a,m-1)-f(a,b,c,n)$$

然后呢就能A掉这题了

#include<bits/stdc++.h>
using namespace std;
#define f1(a,b,c) for(int c=a;c<=b;c++)
#define f2(a,b,c) for(int c=a;c>=b;c--)
#define f3(a,b,c) for(int c=a;c;c=b)
#define so1(a,n) sort(a+1,a+n+1,mycmp);
#define so2(a,n) sort(a+1,a+n+1);
#define ll long long
const ll twx=+100;
const ll MOD=998244353;
const ll i_2=499122177;
const ll i_6=166374059;
ll t;
ll n,a,b,c;
struct LV
{
    LV()
    {
        f=g=h=0;
    }
    ll f,g,h;//上文所说的三个函数
};
LV work(ll n,ll a,ll b,ll c)
{
    ll a_c=a/c;
    ll b_c=b/c;
    ll m=(a*n+b)/c;
    ll n_1=n+1;
    ll n_2_1=n*2+1;
    LV d;
    if(a==0)
    {
        d.f=b_c*n_1%MOD;
        d.g=b_c*n%MOD*n_1%MOD*i_2%MOD;
        d.h=b_c*b_c%MOD*n_1%MOD;
        return d;
    }
    if(a>=c||b>=c)
    {
        d.f=n*n_1%MOD*i_2%MOD*a_c%MOD+b_c*n_1%MOD;
        d.g=a_c*n%MOD*n_1%MOD*n_2_1%MOD*i_6%MOD+b_c*n%MOD*n_1%MOD*i_2%MOD;
        d.h=a_c*a_c%MOD*n%MOD*n_1%MOD*n_2_1%MOD*i_6%MOD+b_c*b_c%MOD*n_1%MOD+a_c*b_c%MOD*n%MOD*n_1%MOD;
        d.f%=MOD;
        d.g%=MOD;
        d.h%=MOD;
        LV e=work(n,a%c,b%c,c);
        d.h+=e.h+2*b_c%MOD*e.f%MOD+2*a_c%MOD*e.g%MOD;
        d.g+=e.g;
        d.f+=e.f;
        d.f%=MOD;
        d.g%=MOD;
        d.h%=MOD;
        return d;
    }
    LV e=work(m-1,c,c-b-1,a);
    d.f=n*m%MOD-e.f;
    d.f=(d.f%MOD+MOD)%MOD;
    d.g=m*n%MOD*n_1%MOD-e.h-e.f;
    d.g=(d.g*i_2%MOD+MOD)%MOD;
    d.h=n*m%MOD*(m+1)%MOD-2*e.g-2*e.f-d.f;
    d.h=(d.h%MOD+MOD)%MOD;
    return d;
}
ll read()
{
    ll sum=0;
    ll flag=1;
    char c=getchar();
    while(c<'0'||c>'9')
    {
        if(c=='-')
        {
            flag=-1;
        }
        c=getchar();
    }
    while(c>='0'&&c<='9')
    {
        sum=((sum*10)+c-'0');
        c=getchar();
    }
    return sum*flag;
}
void init()
{
	t=read();
    while(t--)
    {
        n=read();
        a=read();
        b=read();
        c=read();
        LV ans=work(n,a,b,c);
        printf("%lld %lld %lld\n",ans.f,ans.h,ans.g);
    }
}
int main()
{
	//freopen(".in","r",stdin);
	//freopen(".out","w",stdout);
    init();
	return 0;
}

现在回到最开始讨论的那个问题：

给你$T$组询问

求出

$$\sum _{x=0}^n{x}^{k_1}{\lfloor \frac{ax+b}{c}\rfloor }^{k_2}$$

对$1e9+7$取模的结果

同样的，我们设$$f(a,b,c,n,k1,k2)=\sum _{x=0}^n{x}^{k_1}{\lfloor \frac{ax+b}{c}\rfloor }^{k_2}$$

当$a=0$时，原式=$${\lfloor \frac{b}{c}\rfloor }^{k_2}\sum _{x=0}^n{x}^{k_1}$$

在这种情况下，这就是个自然数幂和
令$B$表示伯努利数,$$S_k(n)=\sum _{x=0}^n{x}^{k_1}$$

$$S_k(n)=\frac{1}{k+1}\sum _{y=0}^k(-1{)}^y(\genfrac{}{}{0ex}{}{k+1}{y})B_y{n}^{k+1-y}$$

令$P_{x,y}=(-1{)}^y(\genfrac{}{}{0ex}{}{x+1}{y})$

当$a\ge b$时，设$a=qc+r,0\le r<b$，则：
$$\lfloor \frac{yc-b-1}{a}\rfloor <x\le \lfloor \frac{yc+c-b-1}{a}\rfloor$$

设$m=\frac{an+b}{c}$

则：

$$\sum _{x=0}^n{x}^{k_1}{\lfloor \frac{ax+b}{c}\rfloor }^{k_2}=S_{k_1}(n)m^{k_2}+\sum _{y=1}^m((y-1{)}^{k_2}-y^{k_2})S_{k_1}\left(\lfloor \frac{yc-b-1}{a}\rfloor \right)$$

而

$$(y-1{)}^{k_2}-y^{k_2}=\sum _{i=0}^{k_2-1}(\genfrac{}{}{0ex}{}{k_2}{i})(-1{)}^{k_2-i}{y}^i$$

最后得到：

$$\sum _{x=0}^n{x}^{k_1}{\lfloor \frac{ax+b}{c}\rfloor }^{k_2}=S_{k_1}(n)m^{k_2}+\sum _{i=0}^{k_2-1}\sum _{h=0}^{k_1+1}\left(\left(\genfrac{}{}{0ex}{}{k_2}{i}\right)(-1{)}^{k_2-i}{P}_{k_1,h}\sum _{y=1}^m{y}^i{\lfloor \frac{yc-b-1}{c}\rfloor }^h\right)$$

$$这篇总结挂了$$