题解 P5170 【【模板】类欧几里得算法】

ix35 · 2020-03-21 08:21:58 · 题解

最近在学万能欧几里得的套路，就记录一下。

目前的十篇题解中只有一篇是用万能欧几里得的，我感觉讲的也不是非常清楚，就自己写一遍吧。

P5170 【模板】类欧几里得算法

解决什么？

很多类欧几里得问题都可以归结为以下这个模型：

考虑笛卡尔坐标系中一条直线 y=\dfrac{Px+R}{Q}，并且做出所有形如 x=k\ (k\in Z) 的竖线以及 y=h\ (h\in Z) 的横线。

假想从左往右扫描这条直线，每遇到这条直线与一个横线的交点，则执行 U 操作；每遇到这条直线与一个竖线的交点，则执行 R 操作，规定同时遇到一条横线和竖线（即到达一个整点）时先执行 U 后执行 R。

这样一来，我们可以将一条直线转译成一个由 U,R 组成的操作序列。

例如，考虑 y=\dfrac{3x+2}{5}，其图像如下：

在 x\in (0,6] 的范围内，其对应的操作序列就是：URRURRURUR。

同时，操作要满足可合并性。例如上面的操作序列，我们设 P=UR，可以进一步转化为 PRPRPP。

操作的内容可能是对于一些变量做修改，具体的情况我们最后再说。而万能欧几里得的套路可以帮我们快速计算进行这样的操作序列后的某些信息。

我们将两个操作序列 S 和 T 的合并序列称为 S+T 或 ST，这个合并过程不一定满足交换律。

如何解决？

下面这段里面有作为整数的 R 和作为操作的 R，但好像并不会出现什么混乱，这里就不改了。

方便起见，我们用一个“结点”（Node）来刻画一个操作序列。

解决问题的过程是递归的，我们设 solve(P,Q,R,L,U,R) 表示处理 y=\dfrac{Px+R}{Q}\ (x\in (0,L]) 这个线段，U,R 分别为遇到横线和竖线后执行的操作序列。

既然这个东西叫万能欧几里得，自然就和欧几里得算法的迭代有关，这分为两个部分。

第一部分：P\ge Q 时，我们尝试将这个问题分解成规模为 P\bmod Q 的子问题。

我们发现，如果 P\ge Q，那么只要令 R'=U^{\frac{P}{Q}}R，然后直接递归 solve(P\bmod Q,Q,R,L,U,R') 即可，这是因为每一个 R 的前面必然有至少 \dfrac{P}{Q} 个连续的 U，它们可以和这个 R 合到一起。

第二部分：P<Q 时，交换 P,Q 地位，再进入第一部分的迭代。

容易发现，第 a 个 R 之前会有 \lfloor\dfrac{Pa+R}{Q}\rfloor 个 U，我们设第 a 个 R 在第 b 个 U 之前，那么：

因此，第 b 个 U 之前会有 \lfloor\dfrac{Qb-R-1}{P}\rfloor 个 R，这里就是一种转换思想，将 U,R 操作进行的转换。

迭代的子问题应该与原问题形式相同，所以开头的一段（第一个 U 之前）操作应该被截掉，对第一个 U 之后的部分进行递归。

假设现在去掉了第一个 U 以及它之前的 R，看看剩余的第 b 个 U 与第 b-1 个 U 之间剩余的 R 的个数：

这个形式就与原问题相同了，只需递归 solve(Q,P,Q-R-1,CntU-1,R,U) 即可，其中 CntU 为当前这一层 U 的总数，就等于 \lfloor\dfrac{PL+R}{Q}\rfloor，要 -1 是因为我们截去了第一个 U 之前的部分。

再考虑首位的零碎部分，首端（也就是第一个 U 之前）有 \lfloor\dfrac{Q-R-1}{P}\rfloor 个 R，所以前面要补的操作序列是 R^{\lfloor\frac{Q-R-1}{P}\rfloor}U，另外操作序列最后还有若干个 R，个数为 L-\lfloor\dfrac{Q\times CntU-R-1}{P}\rfloor，这一段也要拼到最后。

我们来看看这个过程的时间复杂度，假设一次合并操作的复杂度为 O(c)，那么形如 A^{b} 这样的式子可以使用快速幂优化到 O(c\log b)。

考虑算法过程中每层迭代用到了一次 R^{\lfloor\frac{Q-R-1}{P}\rfloor}，时间复杂度是 O(c \log \dfrac{Q}{P}) 的，也就是 O(c(\log Q-\log P))。

而我们又发现相邻两层的 P,Q 地位交换，所以下一层的 Q 就是这个 P，下一层的 \log Q 也就和这一层的 -\log P 抵消了。

所以总的复杂度就是 O(c\log \max(P,Q))。

Node solve (ll p,ll q,ll r,ll l,Node a,Node b) {
    if (!l) {return Node();}
    if (p>=q) {return solve(p%q,q,r,l,a,qpow(a,p/q)+b);}
    ll m=div(l,p,r,q);
    if (!m) {return qpow(b,l);}
    ll cnt=l-div(q,m,-r-1,p);
    return qpow(b,(q-r-1)/p)+a+solve(q,p,(q-r-1)%p,m-1,b,a)+qpow(b,cnt);
}

这题怎么做？

我们只需要讨论这题中 U 和 R 分别是什么，即可套用上面的做法。

首先考虑每个状态维护的值，我们设 s=\dfrac{Px+R}{Q}，我们要求的是 s 的和，s 的平方和以及 s\times x 的和。

一般这样的题目中，还需要额外维护的是这一段操作序列中 U 和 R 的个数，此外，为了计算 s\times x 的和，我们还需要记录 x 的和。

因此我们记录六个量：U 的个数 cntu，R 的个数 cntr，x 的和 sumi，s 的和 sums，s 的平方和 sqrs，s\times x 的和 prod。

遇到 U 时，我们需要执行：cntu\leftarrow cntu+1

遇到 R 时，我们需要执行：cntr\leftarrow cntr+1,\ \ sumi\leftarrow sumi+cntr,\ \ sums\leftarrow sums+cntu，