类欧几里得算法学习笔记

这个东西看起来好恐怖啊QAQ.jpg搞了几天，QWQ

---

其实类欧几里得算法，就是和欧几里得算法类似（欧几里得算法就是gcd那一堆啦），但是其实只有时间复杂度的证明是和gcd一样的，其它的都完全不同,emmmm。

---

先前置两道例题：【luoug-T36097 整点与质数】【[COCI2009-2010#1] ALADIN
】【讲解by hdxrie】

其实这个就是一个十分简单的类欧算法的应用
以前做到的时候还不知道，现在学了才知道，QAQ

---

我们可以从一个简单的式子来入门，并对其套路有一定的了解。

$T\le 10^5$组询问，每组对于给定的$n\le 10^9$求下面式子在$\mathrm{m}\mathrm{o}\mathrm{d}998244353$意义下的值$(a,b\ge 0,c>0)$：

$$\sum _{i=0}^n\lfloor \frac{ai+b}{c}\rfloor$$

其实前面的$\lfloor \frac{ip}{q}\rfloor$就是这个式子的简化版，$b=0$。

显然这个不能直接$O(n)$算，或者除法分块，所以我们要另寻它路。

首先我们可以分成两步考虑：

1. $a\ge c$或者$b\ge c$时

此时，我们显然可以将原式写成：

$$\sum _{i=0}^n\left(\lfloor \frac{a}{c}\rfloor i+\lfloor \frac{b}{c}\rfloor +\lfloor \frac{(a\%c)i+(b\%c)}{c}\rfloor \right)$$

将其拆开可以得到：

$$\lfloor \frac{a}{c}\rfloor \sum _{i=0}^{n}i+\sum _{i=0}^n\lfloor \frac{b}{c}\rfloor +\sum _{i=0}^n\lfloor \frac{(a\%c)i+(b\%c)}{c}\rfloor$$

前面两个可以用公式$O(1)$的求，而后面那个又是一个类似的子问题，递归求解即可。

所以我们令$f(a,b,c,n)={\sum }_{i=0}^n\lfloor \frac{ai+b}{c}\rfloor$，那么原式可以写成：

$$f(a,b,c,n)=\lfloor \frac{a}{c}\rfloor \frac{n\times (n+1)}{2}+(n+1)\lfloor \frac{b}{c}\rfloor +f(a\%c,b\%c,c,n)$$

此时我们就把问题从$a\ge c,b\ge c$转换$a,b<c$的情况了。

2. 那么显然另外一种情况就是$a<c$且$b<c$的时候了。

其实，如果看了$\mathrm{h}\mathrm{d}\mathrm{x}\mathrm{r}\mathrm{i}\mathrm{e}$的关于$[\mathrm{C}\mathrm{O}\mathrm{C}\mathrm{I}2010]\mathrm{A}\mathrm{L}\mathrm{A}\mathrm{D}\mathrm{I}\mathrm{N}$的那个题的讲解，这个也就是相当于求一条直线下整点个数，具体讲解可以看这篇论文【洪华敦-类欧几里得算法】。

那么我们可以枚举整点的坐标，原式就可以转化成（这里$m=\lfloor \frac{an+b}{c}\rfloor -1$）：

$$f(a,b,c,n)=\sum _{i=0}^n\sum _{j=1}^{m+1}[j⩽\lfloor \frac{ai+b}{c}\rfloor ]$$

这里$[a=1]$这种是逻辑判断符，如果里面语句为真，那么值为1，否则为0。

其实这里就是相当于一条$y=\frac{ax+b}{c}$的直线，所以我们枚举$i$就是枚举的$x$，$j$就是$y$，如果$j$小于等于的话$(i,j)$就是在这个直线下的整点。

然后我们先交换枚举顺序可以得到：

由于$a⩽\lfloor b\rfloor$时也满足$a⩽b$，所以变换时可以把向下取整符号去掉。

$$\begin{array}{rl}f(a,b,c,n)=& \sum _{j=1}^{m+1}\sum _{i=0}^n[j⩽\lfloor \frac{ai+b}{c}\rfloor ]\\ =& \sum _{j=0}^m\sum _{i=0}^n[j+1⩽\lfloor \frac{ai+b}{c}\rfloor ]\\ =& \sum _{j=0}^m\sum _{i=0}^n[j+1⩽\frac{ai+b}{c}]\\ =& \sum _{j=0}^m\sum _{i=0}^n[jc+c⩽ai+b]\\ =& \sum _{j=0}^m\sum _{i=0}^n[ai⩾jc+c-b]\\ =& \sum _{j=0}^m\sum _{i=0}^n[ai>jc+c-b-1]\\ =& \sum _{j=0}^m\sum _{i=0}^n[i>\lfloor \frac{jc+c-b-1}{a}\rfloor ]\end{array}$$

由于在$i\in [0,n]$中的$i>\lfloor \frac{jc+c-b-1}{a}\rfloor$的只有$\lfloor \frac{jc+c-b-1}{a}\rfloor \sim n$，所以进而化简得：

$$\begin{array}{rl}f(a,b,c,n)=& \sum _{j=0}^m\left(n-\lfloor \frac{jc+c-b-1}{a}\rfloor \right)\\ =& \sum _{j=0}^{m}n-\sum _{j=0}^m\lfloor \frac{jc+c-b-1}{a}\rfloor \\ =& n\times (m+1)-f(c,c-b-1,a,m)\end{array}$$

那么这样又可以将$f(a,b,c,n)\Rightarrow f(c,c-b-1,a,\lfloor \frac{an+b}{c}\rfloor -1)$，又将问题变小了。

那么边界就是当：

• $n=0$时，直接返回$\lfloor \frac{b}{c}\rfloor$
• $a=0$时，就是$(n+1)\times \lfloor \frac{b}{c}\rfloor$

f总结

$$f(a,b,c,n)=\{\begin{array}{ll}\lfloor \frac{b}{c}\rfloor & n=0\\ (n+1)\times \lfloor \frac{b}{c}\rfloor & a=0\\ \lfloor \frac{a}{c}\rfloor \frac{n\times (n+1)}{2}+(n+1)\lfloor \frac{b}{c}\rfloor +f(a\%c,b\%c,c,n)& a⩾corb⩾c\\ n\times \lfloor \frac{an+b}{c}\rfloor -f(c,c-b-1,a,m)& a<candb<c\end{array}$$

这时就已经可以写出一份十分简单的递归的代码了：

int S1(int n){return n*(n+1)/2;}
int f(int a,int b,int c,int n){
	if(!n) return b/c;
	if(!a) return (n+1)*(b/c);
	if(a>=c||b>=c)return S1(n)*(a/c)+n*(b/c)+f(a%c,b%c,c,n);
	return n*((a*n+b)/c)-f(c,c-b-1,a,(a*n+b)/c-1);
}

关于复杂度，首先我们看$a\ge c,b\ge c$的时候只需$O(1)$就转移到了$a<c,b<c$的情况了，而这时$f(a,b,c,n)\Rightarrow f(c,c-b-1,a,\lfloor \frac{an+b}{c}\rfloor -1)$这个就十分类似于欧几里得算法的更相减损术，直到$a=0$或$n=0$时就停止了，那么复杂度就类似于欧几里得算法，是$O(logn)$的。

---

有了初步的了解，我们就可以看看稍微难一点的：

同样的条件，再求下面两个式子的值：

$$\sum _{i=0}^n{\left(\lfloor \frac{ai+b}{c}\rfloor \right)}^2$$

和

$$\sum _{i=0}^{n}i\lfloor \frac{ai+b}{c}\rfloor$$

---

之所以放在一起说，是因为这两个在求取的过程中会有联系，而且还要用到前面的$f$。

首先我们令$h(a,b,c,n)={\sum }_{i=0}^n{\left(\lfloor \frac{ai+b}{c}\rfloor \right)}^2,g(a,b,c,n)={\sum }_{i=0}^{n}i\lfloor \frac{ai+b}{c}\rfloor$

---

那么我们先来看$h$的求法，和$f$类似（其实类欧的套路都差不多是这样），我们分两块考虑：

1. 当$a\ge c$或者$b\ge c$时：

显然同理有：

$$h(a,b,c,n)=\sum _{i=0}^n{\left(\lfloor \frac{a}{c}\rfloor i+\lfloor \frac{b}{c}\rfloor +\lfloor \frac{(a\%c)i+(b\%c)}{c}\rfloor \right)}^2$$

我们把那个恐怖的括号打开，可以得到（$(a+b+c{)}^2=a^2+b^2+c^2+2ab+2bc+2ac$）：

$$\begin{gathered} h(a,b,c,n)=\sum _{i=0}^n{\left(\lfloor \frac{a}{c}\rfloor i\right)}^2+\sum _{i=0}^n{\left(\lfloor \frac{b}{c}\rfloor \right)}^2+\sum _{i=0}^n{\left(\lfloor \frac{(a\%c)i+(b\%c)}{c}\rfloor \right)}^2 \\ +\sum _{i=0}^{n}2\cdot \lfloor \frac{a}{c}\rfloor i\cdot \lfloor \frac{b}{c}\rfloor +\sum _{i=0}^{n}2\cdot \lfloor \frac{b}{c}\rfloor \cdot \lfloor \frac{(a\%c)i+(b\%c)}{c}\rfloor +\sum _{i=0}^{n}2\cdot \lfloor \frac{a}{c}\rfloor i\cdot \lfloor \frac{(a\%c)i+(b\%c)}{c}\rfloor \end{gathered}$$

我们把这一大坨，整理化简一下，可以得到：

由于${\sum }_{i=0}^n{i}^2=\frac{n\times (n+1)\times (2n+1)}{6}$，${\sum }_{i=0}^{n}i=\frac{n\times (n+1)}{2}$

$$\begin{gathered} h(a,b,c,n)={\left(\lfloor \frac{a}{c}\rfloor \right)}^2\frac{n\times (n+1)\times (2n+1)}{6}+(n+1)\times {\left(\lfloor \frac{b}{c}\rfloor \right)}^2+h(a\%c,b\%c,c,n) \\ +2\cdot \lfloor \frac{a}{c}\rfloor \cdot \lfloor \frac{b}{c}\rfloor \cdot \frac{n\times (n+1)}{2}+2\cdot \lfloor \frac{b}{c}\rfloor \cdot f(a\%c,b\%c,c,n)+2\cdot \lfloor \frac{a}{c}\rfloor \cdot g(a\%c,b\%c,c,n) \end{gathered}$$

递归处理就可以了（假设我们现在知道$g$如何计算）。

2. 我们继续考虑当$a<c,b<c$的时候的情况：

通过$f$的经验，我们照着之前的变换一下看（$m$的意义同前面）：

$$h(a,b,c,n)=\sum _{i=0}^n{\left(\sum _{j=1}^{m+1}[j⩽\lfloor \frac{ai+b}{c}\rfloor ]\right)}^2$$

我们将后面拆成两个式子，然后按照$f$的方式变换：

$$\begin{array}{rl}h(a,b,c,n)=& \sum _{i=0}^n\left(\sum _{j=1}^{m+1}[j⩽\lfloor \frac{ai+b}{c}\rfloor ]\right)\times \left(\sum _{k=1}^{m+1}[k⩽\lfloor \frac{ai+b}{c}\rfloor ]\right)\\ =& \sum _{i=0}^n\left(\sum _{j=0}^m[j+1⩽\frac{ai+b}{c}]\right)\times \left(\sum _{k=0}^m[k+1⩽\frac{ai+b}{c}]\right)\\ =& \sum _{i=0}^n\left(\sum _{j=0}^m[jc+c⩽ai+b]\right)\times \left(\sum _{k=0}^m[kc+c⩽ai+b]\right)\\ =& \sum _{i=0}^n\left(\sum _{j=0}^m[ai⩾jc+c-b]\right)\times \left(\sum _{k=0}^m[ai⩾kc+c-b]\right)\\ =& \sum _{i=0}^n\left(\sum _{j=0}^m[ai>jc+c-b-1]\right)\times \left(\sum _{k=0}^m[ai>kc+c-b-1]\right)\end{array}$$

然后我们同理，交换枚举顺序，可以得到：

$$\begin{array}{rl}h(a,b,c,n)=& \sum _{j=0}^m\sum _{k=0}^m\sum _{i=0}^n[ai>\max (jc-b-1,kc-b-1)]\\ =& \sum _{j=0}^m\sum _{k=0}^m\sum _{i=0}^n[i>\max (\lfloor \frac{jc+c-b-1}{a}\rfloor ,\lfloor \frac{kc+c-b-1}{a}\rfloor )]\end{array}$$

同样，$i\in [0,n]$的$i>\max (\lfloor \frac{jc+c-b-1}{a}\rfloor ,\lfloor \frac{kc+c-b-1}{a}\rfloor )$的只有$>\max (\lfloor \frac{jc+c-b-1}{a}\rfloor ,\lfloor \frac{kc+c-b-1}{a}\rfloor )\sim n$这一部分，所以原式还可以写成：

$$\begin{array}{rl}h(a,b,c,n)=& \sum _{j=0}^m\sum _{k=0}^m\left(n-\max (\lfloor \frac{jc+c-b-1}{a}\rfloor ,\lfloor \frac{kc+c-b-1}{a}\rfloor )\right)\\ =& n(m+1{)}^2-\sum _{j=0}^m\sum _{k=0}^m\max (\lfloor \frac{jc+c-b-1}{a}\rfloor ,\lfloor \frac{kc+c-b-1}{a}\rfloor )\end{array}$$

这里的$\max$不好处理，但是我们可以通过枚举当前这个是$\max$的贡献，首先$j⩾k$的部分为：

$$\sum _{j=0}^m(j+1)\lfloor \frac{jc+c-b-1}{a}\rfloor$$

因为对于每个$j$，都有$0\sim j$也就是$j+1$个$k$比它小，所以每个的贡献次数是$j+1$次。

那么同理对于$j⩽k$的部分就是：
$$\sum _{k=0}^m(k+1)\lfloor \frac{kc+c-b-1}{a}\rfloor$$

但是这两个加起来，会把$j=k$的部分多加一次，所以我们减去多加了的这个部分：

$$\sum _{j=0}^m\lfloor \frac{jc+c-b-1}{a}\rfloor$$

那么原式就可以变成：

$$\begin{array}{rl}h(a,b,c,n)=& n(m+1{)}^2-\sum _{j=0}^m\sum _{k=0}^m\max (\lfloor \frac{jc+c-b-1}{a}\rfloor ,\lfloor \frac{kc+c-b-1}{a}\rfloor )\\ =& n(m+1{)}^2-\left(\sum _{j=0}^m(j+1)\lfloor \frac{jc+c-b-1}{a}\rfloor +\sum _{k=0}^m(k+1)\lfloor \frac{kc+c-b-1}{a}\rfloor -\sum _{j=0}^m\lfloor \frac{jc+c-b-1}{a}\rfloor \right)\\ =& n(m+1{)}^2-\left(2\cdot \sum _{j=0}^m(j+1)\lfloor \frac{jc+c-b-1}{a}\rfloor -\sum _{j=0}^m\lfloor \frac{jc+c-b-1}{a}\rfloor \right)\end{array}$$

我们把$(j+1)$打开，然后整理可以得到：

$$\begin{array}{rl}h(a,b,c,n)=& n(m+1{)}^2-\left(2\cdot \sum _{j=0}^{m}j\lfloor \frac{jc+c-b-1}{a}\rfloor +2\cdot \sum _{j=0}^m\lfloor \frac{jc+c-b-1}{a}\rfloor -\sum _{j=0}^m\lfloor \frac{jc+c-b-1}{a}\rfloor \right)\\ =& n(m+1{)}^2-\left(2\cdot \sum _{j=0}^{m}j\lfloor \frac{jc+c-b-1}{a}\rfloor +\sum _{j=0}^m\lfloor \frac{jc+c-b-1}{a}\rfloor \right)\\ =& n(m+1{)}^2-2\cdot g(c,c-b-1,a,m)-f(c,c-b-1,a,m)\end{array}$$

那么我们同样可以将其转移到更小的问题上，递归计算即可。

边界同理：

• 当$n=0$时，就是${\left(\frac{b}{c}\right)}^2$
• 当$a=0$时，就是$(n+1)\times {\left(\frac{b}{c}\right)}^2$

h总结

$$h(a,b,c,n)=\{\begin{array}{ll}{\left(\frac{b}{c}\right)}^2& n=0\\ (n+1)\times {\left(\frac{b}{c}\right)}^2& a=0\\ {\left(\lfloor \frac{a}{c}\rfloor \right)}^2\frac{n(n+1)(2n+1)}{6}+(n+1){\left(\lfloor \frac{b}{c}\rfloor \right)}^2+h(a\%c,b\%c,c,n)+2\lfloor \frac{a}{c}\rfloor \lfloor \frac{b}{c}\rfloor \frac{n(n+1)}{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)& a⩾corb⩾c\\ n(m+1{)}^2-2\cdot g(c,c-b-1,a,m)-f(c,c-b-1,a,m)& a<candb<c\end{array}$$

代码同样十分好写，仿照上面的即可，这里就不写了。

---

接下来只剩下对于$g$的求法了。

$g$相对$h$来说就很好变换了，直接跟着套路走：

1. 当$a⩾c,b⩾c$时：

$$\begin{array}{rl}g(a,b,c,n)=& \sum _{i=0}^{n}i\lfloor \frac{ai+b}{c}\rfloor \\ =& \sum _{i=0}^n\lfloor \frac{a}{c}\rfloor i^2+\lfloor \frac{b}{c}\rfloor i+i\lfloor \frac{(a\%c)i+(b\%c)}{c}\rfloor \\ =& \lfloor \frac{a}{c}\rfloor \frac{n(n+1)(2n+1)}{6}+\lfloor \frac{b}{c}\rfloor \frac{n(n+1)}{2}+g(a\%c,b\%c,c,n)\end{array}$$

2. 当$a<c,b<c$

根据套路，我们转换成这个形式：

$$\begin{array}{rl}g(a,b,c,n)=& \sum _{j=0}^m\sum _{i=0}^{n}i[i>\lfloor \frac{jc+c-b-1}{a}\rfloor ]\\ =& \sum _{j=0}^m\sum _{i=\lfloor \frac{jc+c-b-1}{a}\rfloor +1}^{n}i\\ =& \sum _{j=0}^m\left(\sum _{i=0}^{n}i-\sum _{i=0}^{\lfloor \frac{jc+c-b-1}{a}\rfloor }i\right)\\ =& \sum _{j=0}^m\frac{n(n+1)}{2}-\frac{\lfloor \frac{jc+c-b-1}{a}\rfloor \cdot \left(\lfloor \frac{jc+c-b-1}{a}\rfloor +1\right)}{2}\\ =& \frac{1}{2}\left(n(n+1)(m+1)-\sum _{j=0}^m{\left(\lfloor \frac{jc+c-b-1}{a}\rfloor \right)}^2-\lfloor \frac{jc+c-b-1}{a}\rfloor \right)\\ =& \frac{1}{2}\left(n(n+1)(m+1)-h(c,c-b-1,a,m)-f(c,c-b-1,a,m)\right)\end{array}$$

由此我们可以直接判断后递归计算即可。

边界条件还是一样的：

• 当$n=0$时，为$0$。
• 当$a=0$时，为$\lfloor \frac{b}{c}\rfloor (\frac{n(n+1)}{2})$

g总结

$$g(a,b,c,n)=\{\begin{array}{ll}0& n=0\\ \lfloor \frac{b}{c}\rfloor (\frac{n(n+1)}{2})& a=0\\ \lfloor \frac{a}{c}\rfloor \frac{n(n+1)(2n+1)}{6}+\lfloor \frac{b}{c}\rfloor \frac{n(n+1)}{2}+g(a\%c,b\%c,c,n)& a⩾corb⩾c\\ \frac{1}{2}\left(n(n+1)(m+1)-h(c,c-b-1,a,m)-f(c,c-b-1,a,m)\right)& a<candb<c\end{array}$$

代码同理。

---

技巧：为了方便求取$g,h$，通常我们可以在一个递归里面写，可以降低复杂度，并且写起了也挺简单的。

• 【模板-IN-luogu】

这个算是个简单的模板了，求的就是上面讲的$f,h,g$函数：

//代码中的h,g的意义对换了一下，懒得改了
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const ll Mod=998244353;
const ll inv2=499122177,inv6=166374059;
ll S1(ll x){if(x>=Mod)x%=Mod;return (x*(x+1)%Mod)*inv2%Mod;}
ll S2(ll x){if(x>=Mod)x%=Mod;return (x*(x+1)%Mod*(x+x+1)%Mod)*inv6%Mod;}
ll Sqr(ll x){return x*x%Mod;}
struct node{
	ll f,g,h;
	void clear(){f=g=h=0;}
	node(){}
	node(ll a,ll b,ll c):f(a),g(b),h(c){}
	void out(){printf("%lld %lld %lld\n",f,g,h);}
};
node calc(ll a,ll b,ll c,ll n){
	node ans,res;ans.clear();
	ll m,t1,t2,s1,s2;
	if(!n){ans.f=b/c;ans.g=Sqr(b/c);return ans;}
	if(!a){
		t1=b/c;
		ans.f=(n+1ll)*t1%Mod;
		ans.g=(n+1ll)*Sqr(t1)%Mod;
		ans.h=S1(n)*t1%Mod;
		return ans;
	}
	if(a>=c||b>=c){
		t1=a/c;t2=b/c;
		res=calc(a%c,b%c,c,n);
		s1=S1(n);s2=S2(n);
		ans.f=(((s1*t1%Mod)+(n+1ll)*t2%Mod)%Mod+res.f)%Mod;
		ans.g=(((Sqr(t1)*s2%Mod+(n+1ll)*Sqr(t2)%Mod)%Mod)+((t1*t2%Mod)*2ll*s1%Mod+(t1*2ll*res.h%Mod))%Mod+(res.g+t2*2ll*res.f%Mod)%Mod)%Mod;
		ans.h=((s2*t1%Mod+s1*t2%Mod)+res.h)%Mod;
		return ans;
	}
	m=(n*a+b)/c-1;
	res=calc(c,c-b-1,a,m);
	ll w1=n*(m+1)%Mod,w2=n*(n+1)%Mod,w3=m+1;
	if(w3>=Mod)w3%=Mod;
	ans.f=(w1-res.f)%Mod;if(ans.f<0)ans.f+=Mod;
	ans.g=((w1*w3)%Mod-((res.h*2ll%Mod+res.f)%Mod))%Mod;if(ans.g<0)ans.g+=Mod;
	ans.h=((w2*w3)%Mod-(res.f+res.g)%Mod)%Mod*inv2%Mod;if(ans.h<0)ans.h+=Mod;
	return ans;
}
int a,b,c,n,T;
int main(){
	for(scanf("%d",&T);T--;){scanf("%d%d%d%d",&n,&a,&b,&c);calc(a,b,c,n).out();}
	return 0;
}

---

我们其实可以求取更加一般化的式子，如这道模板题【IN-loj】

题意就是多组询问，给定$k_1,k_2,a,b,c,n$求下面式子在$\mathrm{m}\mathrm{o}\mathrm{d}998244353$意义下的值。

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

数据范围见原题面。

• 当$k_1=0,k_2=1$时，就是$f$函数。
• 当$k_1=0,k_2=2$时，就是$h$函数。
• 当$k_1=1,k_2=1$时，就是$g$函数。

那么当$k_1,k_2$不为上面的那些情况的时候应该怎么办呢？

我们还是按照套路来分析，我们令$F(k_1,k_2,a,b,c,n)={\sum }_{i=0}^n{i}^{k_1}{\lfloor \frac{ai+b}{c}\rfloor }^{k_2}$：

• 当$a\ge c,b\ge c$时：

$$F(k_1,k_2,a,b,c,n)=\sum _{i=0}^n{i}^{k_1}{\left(\lfloor \frac{a}{c}\rfloor i+\lfloor \frac{b}{c}\rfloor +\lfloor \frac{(a\%c)i+(b\%c)}{c}\rfloor \right)}^{k_2}$$

用二项式定理，我们可以将后面的括号展开得到：

用二项式定理推导一下可以得到
$(a+b+c{)}^n={\sum }_{i+j+k=n}{C}_n^i{C}_{n-i}^j{a}^i{b}^j{c}^k$

$$\begin{gathered} F(k_1,k_2,a,b,c,n)=\sum _{I=0}^n\sum _{i+j+k=k_2}{C}_{k_2}^i{C}_{k_2-i}^j{\lfloor \frac{a}{c}\rfloor }^i{\lfloor \frac{b}{c}\rfloor }^j\left(I^{k_1+i}\right){\lfloor \frac{(a\%c)I+(b\%c)}{c}\rfloor }^k \\ =\sum _{i+j+k=k_2}{C}_{k_2}^i{C}_{k_2-i}^j{\lfloor \frac{a}{c}\rfloor }^i{\lfloor \frac{b}{c}\rfloor }^j\sum _{I=0}^n\left(I^{k_1+i}\right){\lfloor \frac{(a\%c)I+(b\%c)}{c}\rfloor }^k \\ 0\le i,j,k \end{gathered}$$

我们可以发现，最后面那堆等于$F(k_1+i,k,a\%c,b\%c,c,n)$，所以递归处理即可。

• 当$a<c,b<c$的时候：

我们知道$x^k={\sum }_{i=0}^{x-1}(i+1{)}^k-i^k$

因为其等于：$x^k-(x-1{)}^k+(x-1{)}^k-(x-2{)}^k+\cdots -0^k=x^k-0=x^k$

那么我们用其来变换原式得到：

$$F(k_1,k_2,a,b,c,n)=\sum _{i=0}^n{i}^{k_1}\sum _{j=0}^{\lfloor \frac{ai+b}{c}\rfloor -1}(j+1{)}^{k_2}-j^{k_2}$$

我们交换枚举顺序可以得到（$m=\lfloor \frac{an+b}{c}\rfloor -1$）：

$$\begin{array}{rl}F(k_1,k_2,a,b,c,n)=& \sum _{j=0}^m\left((j+1{)}^{k_2}-j^{k_2}\right)\sum _{i=0}^n{i}^{k_1}[j\le \lfloor \frac{ai+b}{c}\rfloor ]\\ =& \sum _{j=0}^m\left((j+1{)}^{k_2}-j^{k_2}\right)\sum _{i=0}^n{i}^{k_1}[i<\lfloor \frac{jc+c-b-1}{a}\rfloor ]\\ =& \sum _{j=0}^m\left((j+1{)}^{k_2}-j^{k_2}\right)\left(\sum _{i=0}^n{i}^{k_1}-\sum _{i=0}^{\lfloor \frac{jc+c-b-1}{a}\rfloor }{i}^{k_1}\right)\end{array}$$

我们接下来将前面那一堆（也就是相当于$(m+1{)}^{k_2}$）乘进后面，得到：

$$\begin{array}{rl}F(k_1,k_2,a,b,c,n)=& \sum _{j=0}^m\left((j+1{)}^{k_2}-j^{k_2}\right)\sum _{i=0}^n{i}^{k_1}-\sum _{j=0}^m\left((j+1{)}^{k_2}-j^{k_2}\right)\sum _{i=0}^{\lfloor \frac{jc+c-b-1}{a}\rfloor }{i}^{k_1}\\ =& (m+1{)}^{k_2}\sum _{i=0}^n{i}^{k_1}-\sum _{j=0}^m\left((j+1{)}^{k_2}-j^{k_2}\right)\sum _{i=0}^{\lfloor \frac{jc+c-b-1}{a}\rfloor }{i}^{k_1}\end{array}$$

而我们知道，关于自然数幂的前缀和是一个关于幂次+1的多项式，所以我们可以用拉格朗日插值（或者高斯消元）把其系数求出来，然后就可以带入到后面那一堆，再用二项式定理展开，得到：

假设${\sum }_{i=0}^{k+1}{a}_{k,i}{x}^i={\sum }_{i=0}^x{i}^k$

$$\begin{array}{rl}F(k_1,k_2,a,b,c,n)=& (m+1{)}^{k_2}\sum _{i=0}^n{i}^{k_1}-\sum _{j=0}^m\left((j+1{)}^{k_2}-j^{k_2}\right)\sum _{i=0}^{\lfloor \frac{jc+c-b-1}{a}\rfloor }{i}^{k_1}\\ =& (m+1{)}^{k_2}\sum _{i=0}^n{i}^{k_1}-\sum _{k=0}^{k_2-1}\left(\sum _{j=0}^m{C}_{k_2}^k{j}^k\cdot \sum _{i=0}^{\lfloor \frac{jc+c-b-1}{a}\rfloor }{i}^{k_1}\right)\\ =& (m+1{)}^{k_2}\sum _{i=0}^n{i}^{k_1}-\sum _{k=0}^{k_2-1}\left(\sum _{j=0}^m{C}_{k_2}^k{j}^k\cdot \sum _{i=0}^{k_1+1}{a}_{k_1,i}{\left(\lfloor \frac{jc+c-b-1}{a}\rfloor \right)}^i\right)\\ =& (m+1{)}^{k_2}\sum _{i=0}^n{i}^{k_1}-\sum _{k=0}^{k_2-1}\sum _{i=0}^{k_1+1}{C}_{k_2}^k{a}_{k_1,i}\cdot \sum _{j=0}^m{j}^k{\left(\lfloor \frac{jc+c-b-1}{a}\rfloor \right)}^i\\ =& (m+1{)}^{k_2}\sum _{i=0}^n{i}^{k_1}-\sum _{k=0}^{k_2-1}\sum _{i=0}^{k_1+1}{C}_{k_2}^k{a}_{k_1,i}\cdot F(k,i,c,c-b-1,a,m)\end{array}$$

其中，有一步是这样变换的：
$$\begin{gathered} \sum _{i=0}^{x-1}(i+1{)}^k-i^k=\sum _{i=0}^{x-1}\sum _{j=0}^k{C}_k^j{i}^j-i^k \\ =\sum _{i=0}^{x-1}\sum _{j=0}^{k-1}{C}_k^j{i}^j \\ =\sum _{j=0}^{k-1}\left(\sum _{i=0}^{x-1}{C}_k^j{i}^j\right) \end{gathered}$$

那么，这个和前面的其实一样，递归计算就好啦，只不过要提前预处理一下系数和组合数。

$k_1,k_2$比较小，所以每次算的时候暴力枚举一下（或者可以用$FFT$之类的优化一下）就行了。

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const int M=13;
const ll Mod=1e9+7;
ll y[M],C[M][M],F[M][M][44];
ll fpow(ll a,ll b=Mod-2){
	a%=Mod;if(a<0)a+=Mod;ll res=1;for(;b;b>>=1,a=(a*a)%Mod)if(b&1)res=(res*a)%Mod;
	return res;
}
struct Ply{
	ll x[M];
	void clear(){memset(x,0,sizeof(x));}
	Ply(){memset(x,0,sizeof(x));}
	Ply operator +(const Ply &a)const{
		Ply ans;for(int i=0;i<M;i++)ans.x[i]=(x[i]+a.x[i])%Mod;
		return ans;	
	}
	Ply operator *(const Ply &a)const{
		Ply ans;
		for(int i=0;i<M;i++)for(int j=0;i+j<M;j++)
		ans.x[i+j]=(ans.x[i+j]+x[i]*a.x[j]%Mod)%Mod;
		return ans;
	}
	ll operator ()(ll a,int b){
		ll ans=x[b+1];for(int i=b;i>=0;i--)ans=(ans*a%Mod+x[i])%Mod;
		return ans;
	}
}A[M],B[M];
void calc(Ply &a,int n){
	Ply xx,yy;
	for(int i=0;i<=n;i++){
		xx.x[0]=y[i];
		for(int j=0;j<=n;j++)if(i^j)xx.x[0]=xx.x[0]*fpow(i-j)%Mod;
		for(int j=0;j<=n;j++)if(j^i){yy.x[0]=Mod-j;yy.x[1]=1;xx=xx*yy;}
		a=a+xx;xx.clear();
	}
}
ll f(int k1,int k2,ll a,ll b,ll c,ll n,int d=0){
	if(~F[k1][k2][d]) return F[k1][k2][d];
	ll ans=0;
	if(!a) ans=B[k1](n,k1)*fpow(b/c,k2)%Mod;
	else if(!k2) ans=B[k1](n,k1);
	else if(a>=c||b>=c){
		int x=a/c,y=b/c;a%=c;b%=c;
		for(ll i=0,X=1;i<=k2;i++,X=(X*x)%Mod){
			for(ll j=0,Y=X;i+j<=k2;j++,Y=Y*y%Mod){
				ans=(ans+f(k1+i,k2-i-j,a,b,c,n,d+1)*C[k2][i]%Mod*C[k2-i][j]%Mod*Y%Mod)%Mod;
			}
		}
	}else{
		int m=(a*n+b)/c;--m;
		ans=A[k2](m,k2);
		ans=ans*B[k1](n,k1)%Mod;
		for(int j=0;j<k2;j++)for(int i=0;i<=k1+1;i++){
			ans=(ans-f(j,i,c,c-b-1,a,m,d+1)*C[k2][j]%Mod*B[k1].x[i]%Mod)%Mod;
		}
	}
	if(ans<0)(ans%=Mod)+=Mod;ans%=Mod;
	return F[k1][k2][d]=ans;
}
void init(){
	C[0][0]=1;
	for(int i=1;i<=10;i++){
		C[i][0]=C[i][i]=1;
		for(int j=1;j<i;j++){
			C[i][j]=(C[i-1][j]+C[i-1][j-1])%Mod;
		}
	}
	for(int i=0;i<=10;i++){
		if(i)y[0]=1;
		for(int j=1;j<=i+1;j++){
			y[j]=(y[j-1]+fpow(j+1,i)-fpow(j,i))%Mod;
			if(y[j]<0)y[j]+=Mod;
		}
		calc(A[i],i+1);
		if(!i)y[0]=1;else y[0]=0;
		for(int j=1;j<=i+1;j++)y[j]=(y[j-1]+fpow(j,i))%Mod;
		calc(B[i],i+1);
	}
}
int T,a,b,c,n,k1,k2;
int main(){
	init();
	for(scanf("%d",&T);T--;){
		scanf("%d%d%d%d%d%d",&n,&a,&b,&c,&k1,&k2);
		memset(F,-1,sizeof(F));
		printf("%d\n",f(k1,k2,a,b,c,n));
	}
	return 0;
} 

---

题目：

• 【BZOJ3817-Sum】
• 【大佬-Blog】

LOJ模板题的另外两个讲解：

• zzq大佬的博客Orz
• 大佬博客Orz

---

End