C++类欧几里得算法

（本文用 $绿色$标注一切函数，用 $紫色$标注部分可以 $O(1)$计算的常量式，便于观察）（不知道为什么加了颜色过后括号不能自动变大了，将就看吧= =）

概述

类欧几里得算法用于求形如$f(a,b,c,n)=\sum _{i=0}^n\lfloor \frac{ai+b}{c}\rfloor$的式子。这个式子有个几何意义，表示求下图中阴影部分中的整点个数（直线为$y=ax+b$）：

算法

令$a^{\prime }=a\%c,b^{\prime }=b\%c$，由定义式得$$\begin{array}{rl}f(a,b,c,n)& =\sum _{i=0}^n\left(\lfloor \frac{a^{\prime }i+b^{\prime }}{c}\rfloor +\lfloor \frac{a}{c}\rfloor i+\lfloor \frac{b}{c}\rfloor \right)\\ & =f(a^{\prime },b^{\prime },c,n)+\lfloor \frac{a}{c}\rfloor \sum _{i=0}^{n}i+(n+1)\lfloor \frac{b}{c}\rfloor \end{array}$$于是只需要处理$a,b<c$的情况。

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

现在要想办法把里面变成$i$为主元。
如下图所示，可以看出$0\le j<5$，结合图示可以得到$$j<\lfloor \frac{ai+b}{c}\rfloor \iff (j+1)c\le ai+b$$
即$(j+1)c<ai+b+1$，即$$i>\lfloor \frac{cj-b+c-1}{a}\rfloor$$于是可以继续化简了$$\begin{array}{rl}f(a,b,c,n)& =\sum _{j=0}^{m-1}\sum _{i=0}^n[i>\lfloor \genfrac{}{}{0ex}{}{}{a}\\ & \\ & \\ & \end{array}$$