Faray 数列问题

首先,Farey 数列 \(F_n\) 表示分母不超过 \(n\) 的所有既约真分数按大小顺序排列的集合,形式化来说

\[F_n = \left\{\frac{p}{q} \bigg\vert 0 < p < q \le n, \, \gcd(p, q) = 1\right\} \]

这个数列的渐进大小为 \(\sum\limits_{i = 1}^{n} \varphi(n) \sim \frac{3}{\pi^2}n^2 + o(n \log n)\)。

关于 Farey 数列,有两个经典结论:

  • 若 \(\frac{a}{b}\) 和 \(\frac{c}{d}\) 是 \(F_n\) 中的两个连续元素,那么 \(\frac{a + c}{b + d}\) 也是一个合法的 Faray 数。

  • 若 \(\frac{a}{b}\) 和 \(\frac{c}{d}\) 是 \(F_n\) 中的两个连续元素,那么它们的下一个元素 \(\frac{p}{q}\) 满足

    \[\begin{cases}p = \left\lfloor\frac{n + b}{d}\right\rfloor c - a \\ q = \left\lfloor\frac{n + b}{d}\right\rfloor d - b\end{cases} \]

我们只对上述定理二进行证明,对 \(n\) 阶 Farey 数列,相邻两项 \(\frac{a}{b} < \frac{c}{d}\) 满足 \(bc - ad = 1\),根据相邻两项 \(\frac{a}{b}, \, \frac{c}{d}\) 均在 Stern-Brocot 树上,我们有下一项 \(\frac{p}{q}\) 满足 \(\frac{c}{d} = \frac{a + p}{b + q}\),也就是 \((p + a)d = (q + b)c\),因此存在 \(k\) 使得

\[\begin{cases} kc = a + p \\ kd = b + q \end{cases} \]

为了使 \(p, \, q\) 精度高,那么 \(k\) 的值应该尽可能大,所以对上界 \(n\) 来说,\(p, \, q\) 由下式定义:

\[\begin{cases}p = kc - a \le n \\ q = kd - b \le n\end{cases} \]

显然由真分数定义知 \(q > p\),所以最大 \(k = \left\lfloor\frac{n + b}{d}\right\rfloor\),因此

\[\begin{cases}p = \left\lfloor\frac{n + b}{d}\right\rfloor c - a \\ q = \left\lfloor\frac{n + b}{d}\right\rfloor d - b\end{cases} \]

生成这个数列有两种计算方式,一是种基于第一个定理的 Stern-Brocot 树,这个很好实现。

cpp 复制代码
int build(int a, int b, int c, int d, int n) {
    if (b + d > n) return 0;
    return 1 + build(a, b, a + c, b + d, n) + build(a + c, b + d, c, d, n);
}

另一种是基于第二个定理的递推,也比较好写。

cpp 复制代码
PII nxt_fraction(PII fac1, PII fac2) {
    auto [a, b] = fac1;
    auto [c, d] = fac2;
    return {(n + b) / d * c - a, (n + b) / d * d - b};
}

很显然,他们都是 \(O(n^2)\) 的时间复杂度进行 Faray 数列特定元素的计算。

为了方便我们解决母问题,我们给出这样两个问题:

给定 \(n, \, k\),求 \(F_n\) 的第 \(k\) 项。
给定 \(n\) 和既约真分数 \(\frac{p}{q}(p \le n)\),判断其在 \(F_n\) 中的排名。

我们先考虑第一个问题,我们发现第一个问题可以通过如下步骤进行:

  • 二分 \(\frac{j}{n}\) 中的 \(j\),计算 \(\text{rank}(\frac{j}{n})\)

    • 记 \(r = \text{rank}(\frac{j}{n})\),如果 \(r < k\),向上搜索,否则向下搜索。
  • 找到一个区间 \(\left[\frac{j}{n}, \, \frac{j + 1}{n}\right)\) 使得目标分数在这个范围以内

  • 统计这个区间内的分数,找到合法的分数使得其排名为所给 \(k\)

    • 注意到这个区间内不同分母的分数至多一个,因为 \(\frac{1}{n}\) 是最小间距了

    • 对于一个分母为 \(q\) 的分数,唯一可能的分数的分母只能是 \(\left\lfloor\frac{(j + 1)q - 1}{n}\right\rfloor\)

  • 找到严格大于 \(\frac{j}{n}\) 的最小分数 \(\frac{p}{q}\),利用结论二进行递推,直到给定分数为 \(k\) 排名

我们可以看出,问题一可以转化为问题二实现,因此我们现在来考虑问题二如何完成。

更平凡的,问题二可以规约为如下问题:

给定实数 \(x\),求分母 \(q \le n\) 的既约真分数 \(\frac{p}{q} \le x\) 的个数。

记 \(A_q\) 表示,以 \(q\) 为分母满足上述条件的既约真分数的个数,显然的,可以表示原问题为求这样一个集合:

\[S = \left\{\frac{p}{q} \bigg\vert \frac{p}{q} \le x, \, q \le n, \, \gcd(p, \, q) = 1\right\} \]

也就是求 \(\sum\limits_{i = 1}^{n}A_i = |S|\),而我们知道

\[\left\lfloor x \cdot q \right\rfloor = \sum\limits_{d \mid q}A_d \]

这告诉我们

\[A_q = \left\lfloor x \cdot q \right\rfloor - \sum\limits_{d \mid q, \, d < q}A_d \]

按顺序从小到大递推即可,每次计算复杂度 \(O(n\log{n})\),那么问题一的复杂度就是 \(O(n\log^2{n})\),有没有更优的做法呢?

考虑 \(A_q\) 的本质形态,利用莫比乌斯反演,我们可以得到

\[A_q = \sum\limits_{i = 1}^{\lfloor x \cdot q \rfloor}[\gcd(i, q) = 1] = \sum\limits_{i = 1}^{\lfloor x \cdot q \rfloor}\sum\limits_{d \mid \gcd(i, q)}\mu(d) \]

如果我们对所有 \(A_q\) 求和,那么就有

\[\begin{aligned} |S| &= \sum_{i = 1}^{n}A_i \\ &= \sum_{i = 1}^{n}\sum_{j = 1}^{\lfloor x \cdot i \rfloor}[\gcd(i, \, j) = 1] \\ &= \sum_{i = 1}^{n}\sum_{j = 1}^{\lfloor x \cdot i \rfloor}\sum_{d \mid \gcd(i, j)}\mu(d) \\ &= \sum_{d \mid i, \, d \mid j}\sum_{i = 1}^{n}\sum_{j = 1}^{\lfloor x \cdot i \rfloor}\mu(d) \\ &= \sum_{d = 1}^{n}\sum_{i = 1}^{\left\lfloor\frac{n}{d}\right\rfloor}\sum_{j = 1}^{\lfloor x \cdot i \rfloor}\mu(d) \\ &= \sum_{d = 1}^{n}\mu(d)\sum_{i = 1}^{\left\lfloor\frac{n}{d}\right\rfloor}\lfloor x \cdot i \rfloor \end{aligned}\]

考虑设 \(S(n) = \sum\limits_{i = 1}^{n}\lfloor x \cdot i \rfloor\),那么我们相当于要求的东西也就是

\[\sum_{d = 1}^{n}\mu(d)S\left(\left\lfloor\frac{n}{d}\right\rfloor\right) \]

对 \(\lfloor x \cdot i \rfloor\) 求前缀和,预处理莫比乌斯函数,我们可以在 \(O(n)\) 时间复杂度解决这个问题。

如此,我们在解决 Farey 数列问题上我们有:

  • 预处理:\(O(n)\)

  • 给 \(\text{rank}\) 求分数:\(O(n\log{n})\)

  • 给分数求 \(\text{rank}\):\(O(n)\)

  • 求 \(F_n\) 中比某实数小的数的个数 \(O(n)\)

  • 正实数的有理逼近 \(O(n\log{n})\)

我们给出一份模版代码供参考

cpp 复制代码
#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define PII pair<ll, ll>
const int N = 2e6 + 10;
ll n;
ll primes[N], st[N], mu[N], cnt;

// precalculate Mu
void init_mu() {
    mu[1] = 1;
    for (int i = 2; i < N; i ++ ) {
        if (!st[i]) primes[cnt ++ ] = i, mu[i] = -1;
        for (int j = 0; i * primes[j] < N; j ++ ) {
            st[i * primes[j]] = 1;
            if (i % primes[j] == 0) break;
            mu[i * primes[j]] = -mu[i];
        }
    }
}

// O(nlogn) calculate next one by one
PII nxt_fraction(PII fac) {
    ll l = 0, r = n;
    while (l < r) {
        ll mid = l + r + 1 >> 1;
        if (mid * fac.second <= n * fac.first) l = mid;
        else r = mid - 1;
    }
    PII res = {r + 1, n};
    for (int i = 1; i <= n; i ++ ) {
        PII fac_i = {((r + 1) * i - 1) / n, i};
        if (fac_i.first * fac.second <= fac_i.second * fac.first) continue;
        if (fac_i.first * res.second < fac_i.second * res.first) res = fac_i;
    }
    ll d = __gcd(res.first, res.second);
    return {res.first / d, res.second / d};
}

// O(1) calculate next one by two nearly
PII nxt_fraction(PII fac1, PII fac2) {
    auto [a, b] = fac1;
    auto [c, d] = fac2;
    return {(n + b) / d * c - a, (n + b) / d * d - b};
}

ll fraction_to_rank(PII fac) {
    static ll A[N];
    for (int i = 1; i <= n; i ++ ) A[i] = fac.first * i / fac.second + A[i - 1];
    ll res = 0;
    for (int i = 1; i <= n; i ++ ) res += mu[i] * A[n / i];
    return res;
}

PII rank_to_fraction(ll k) {
    ll l = 0, r = n;
    while (l < r) {
        ll mid = l + r + 1 >> 1;
        if (fraction_to_rank({mid, n}) <= k) l = mid;
        else r = mid - 1;
    }
    k -= fraction_to_rank({r, n});
    PII a = {r / __gcd(r, n), n / __gcd(r, n)}, b = nxt_fraction(a);
    if (!k) return a;
    while (k -- ) a = nxt_fraction(a, b), swap(a, b);
    return a;
}

void solve() {
    cin >> n;
    PII now = {1, n};
    while (now.first != now.second) {
        ll rk = fraction_to_rank(now);
        PII fac = rank_to_fraction(rk);
        cout << rk << " : " << fac.first << " " << fac.second << " | " << now.first << " " << now.second << "\n";
        now = nxt_fraction(now);
    }
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(0), cout.tie(0);
    init_mu();
    int T = 1;
    // cin >> T;
    while (T -- ) solve();
    return 0;
}

Expansion

容易发现,我们实际问题中很难遇到 \(x\) 是无理数的情况,那么我们很容易想到,不妨设 \(x = \frac{p}{q}\),有

\[\sum_{d = 1}^{n}\mu(d)\sum_{i = 1}^{\left\lfloor\frac{n}{d}\right\rfloor}\lfloor x \cdot i \rfloor = \sum_{d = 1}^{n}\mu(d)\sum_{i = 1}^{\left\lfloor\frac{n}{d}\right\rfloor}\left\lfloor \frac{pi}{q} \right\rfloor \]

容易看出,后者是我们类欧几里得算法的标准形式,利用数论分块,可以做到 \(O(\sqrt{n}\log^2{n})\) 每个询问。

更进一步,你可以使用杜教筛对前面的莫比乌斯函数求和,预处理时间复杂度降为 \(O(n^{\frac{2}{3}})\)。

如果你仍然对此法感觉不优,你可以利用狄利克雷前缀和分块处理,总体时间复杂度可以达到惊人的 \(O(n^{\frac{2}{3}} + \sqrt{n}\log^{\frac{3}{2}}{n})\)。

cpp 复制代码
#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define PII pair<ll, ll>
const int N = 1e5;
ll n, k;
ll primes[N], st[N], mu[N], cnt;
unordered_map<ll, ll> summu;

// precalculate Mu
void init_mu() {
    mu[1] = 1;
    for (int i = 2; i < N; i ++ ) {
        if (!st[i]) primes[cnt ++ ] = i, mu[i] = -1;
        for (int j = 0; i * primes[j] < N; j ++ ) {
            st[i * primes[j]] = 1;
            if (i % primes[j] == 0) break;
            mu[i * primes[j]] = -mu[i];
        }
    }
    for (int i = 2; i < N; i ++ ) mu[i] += mu[i - 1];
}

ll get_mu(ll n) {
    if (n < N) return mu[n];
    if (summu.count(n)) return summu[n];
    ll res = 1;
    for (ll l = 2, r; l <= n; l = r + 1) {
        r = n / (n / l);
        res -= (r - l + 1) * get_mu(n / l);
    }
    return summu[n] = res;
}

ll f(ll a, ll b, ll c, ll n) {
	if (!a) return b / c * (n + 1);
	if (a < c && b < c) {
		ll m = (a * n + b) / c;
		if (!m) return 0;
		return n * m - f(c, c - b - 1, a, m - 1);
	}
	return f(a % c, b % c, c, n) + (n + 1) * n / 2 * (a / c) + (n + 1) * (b / c);
}

// O(n + logn) calculate next one by one
PII nxt_fraction(ll n, PII fac) {
    ll l = 0, r = n;
    while (l < r) {
        ll mid = l + r + 1 >> 1;
        if (mid * fac.second <= n * fac.first) l = mid;
        else r = mid - 1;
    }
    PII res = {r + 1, n};
    for (int i = 1; i <= n; i ++ ) {
        PII fac_i = {((r + 1) * i - 1) / n, i};
        if (fac_i.first * fac.second <= fac_i.second * fac.first) continue;
        if (fac_i.first * res.second < fac_i.second * res.first) res = fac_i;
    }
    ll d = __gcd(res.first, res.second);
    return {res.first / d, res.second / d};
}

// O(1) calculate next one by two nearly
PII nxt_fraction(ll n, PII fac1, PII fac2) {
    auto [a, b] = fac1;
    auto [c, d] = fac2;
    return {(n + b) / d * c - a, (n + b) / d * d - b};
}

// O(sqrt(n)logn) calculate fraction's rank
ll fraction_to_rank(ll n, PII fac) {
    ll res = 0;
    for (ll l = 1, r; l <= n; l = r + 1) {
        r = n / (n / l);
        res += (get_mu(r) - get_mu(l - 1)) * f(fac.first, 0, fac.second, n / l);
    }
    return res;
}

// O(sqrt(n)log(n)^2) calculate the fraction of k-th rank
PII rank_to_fraction(ll n, ll k) {
    ll l = 0, r = n;
    while (l < r) {
        ll mid = l + r + 1 >> 1;
        if (fraction_to_rank(n, {mid, n}) <= k) l = mid;
        else r = mid - 1;
    }
    k -= fraction_to_rank(n, {r, n});
    PII a = {r / __gcd(r, n), n / __gcd(r, n)}, b = nxt_fraction(n, a);
    if (!k) return a;
    while (k -- ) a = nxt_fraction(n, a, b), swap(a, b);
    return a;
}

void solve() {
    cin >> n;
    PII now = {1, n};
    while (now.first != now.second) {
        ll rk = fraction_to_rank(n, now);
        PII fac = rank_to_fraction(n, rk);
        cout << rk << " : " << fac.first << " " << fac.second << " | " << now.first << " " << now.second << "\n";
        now = nxt_fraction(n, now);
    }
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(0), cout.tie(0);
    init_mu();
    int T = 1;
    // cin >> T;
    while (T -- ) solve();
    return 0;
}

恰好这里有一个练手题,试试看。