万能欧几里得

#6440. 万能欧几里得

计算 ${\sum }_{i=1}^L{A}^i{B}^{\lfloor \frac{Pi+R}{Q}\rfloor }$

• 万能欧几里得算法：我们做直线 $y=\frac{Pi+R}{Q}$ 从 $(0,R/Q)$ 开始走，维护一个三元组 $(a,b,c)$，碰到上边界则将 $b$ 左乘上 $B$，碰到左边界就把 $a$ 右乘上 $A$，并更新答案 $c\Leftarrow a\ast b$

• 发现这个操作可以构成一个序列，形如 { B . B , A , B , B , B , A , B , B , A , B , B , B , A } \{B.B,A,B,B,B,A,B,B,A,B,B,B,A\} {B.B,A,B,B,B,A,B,B,A,B,B,B,A}
  万能欧几里得算法可以将这个序列合并，压缩使得我们能快速得到答案
  令 $f(n,p,q,r,a,b)={\sum }_{i=1}^n{A}^i{B}^{\lfloor \frac{pi+r}{q}\rfloor }$，其中 $a,b$ 表示我们的一段序列

• 若 $p\ge q$，则发现一个 $A$ 前面一定有 $\lfloor \frac{p}{q}\rfloor$ 个 $B$
  于是我们可以令 $A^{\prime }=B\ast \lfloor \frac{p}{q}\rfloor +A$，实现了序列的压缩

• 若 $p<q$，我们可以通过变换使其变为 $p\ge q$ 的情况，这个变换就叫做坐标旋转
  我们考虑求出 $y=\lfloor \frac{px+r}{q}\rfloor$ 的反函数，即为解一个方程
  $$\begin{gathered} y\le \lfloor \frac{px+r}{q}\rfloor <y+1 \\ qy-r\le px<qy+q-r \\ \lfloor \frac{qy-r+p-1}{p}\rfloor \le x<\lfloor \frac{qy+q-r+p-1}{p}\rfloor \end{gathered}$$
  那么若实现坐标翻转，某一个特定的 $A$ 可以前可以有
  $\lfloor \frac{qy+q-r-1}{p}\rfloor -\lfloor \frac{qy-r-1}{p}\rfloor$ 个 $B$，对比前面一个特定的 $A$前有 $\lfloor \frac{px+p+r}{q}\rfloor -\lfloor \frac{px+r}{q}\rfloor$，那么新的直线可以写作 $x=\frac{qy-r-1}{p}$
  注意分母无论减去多少个分子都不会影响操作序列
  然后这里的定义域要映射成值域，$y=1$ 时不能取满所以我们单独计算，所以新的定义域变成了 $(1,、\lfloor \frac{pn+r}{q}\rfloor ]$，同时定义域后方还有一些点没有算到，同样单独计算
  然后我们就可以递归到 $f(\lfloor \frac{pn+r}{q}\rfloor -1,q,p,q-r-1,b,a)$ 来计算
  $Code$

• 既然它是万能的，那么我们可以顺手切掉一道 【模板】类欧几里得算法
  维护一个 5 元组 $(i,\lfloor \frac{ai+b}{c}\rfloor ,\sum \lfloor \frac{ai+b}{c}\rfloor ,\sum \lfloor \frac{ai+b}{c}{\rfloor }^2\sum i\lfloor \frac{ai+b}{c}\rfloor )$ 合并的时候拆项讨论就做完了
  比类欧几里得好写到哪里去了，不过脑子够笨还是码了好久
  $Code$