【ACWing】1120. 埃及分数

题目地址：

https://www.acwing.com/problem/content/1122/

在古埃及，人们使用单位分数的和（形如$\frac{1}{a}$的，$a$是自然数）表示一切有理数。如：$\frac{2}{3}=\frac{1}{2}+\frac{1}{6}$，但不允许 $\frac{2}{3}=\frac{1}{3}+\frac{1}{3}$，因为加数中有相同的。对于一个分数$\frac{a}{b}$，表示方法有很多种，但是哪种最好呢？首先，加数少的比加数多的好，其次，加数个数相同的，最小的分数越大越好。如：$$\begin{gathered} \frac{19}{45}=\frac{1}{3}+\frac{1}{12}+\frac{1}{180} \\ \frac{19}{45}=\frac{1}{3}+\frac{1}{15}+\frac{1}{45} \\ \frac{19}{45}=\frac{1}{3}+\frac{1}{18}+\frac{1}{30} \\ \frac{19}{45}=\frac{1}{4}+\frac{1}{6}+\frac{1}{180} \\ \frac{19}{45}=\frac{1}{5}+\frac{1}{6}+\frac{1}{18} \end{gathered}$$最好的是最后一种，因为 $\frac{1}{18}$比$\frac{1}{180}$，$\frac{1}{45}$，$\frac{1}{30}$，$\frac{1}{180}$都大。注意，可能有多个最优解。如：$$\begin{gathered} \frac{59}{211}=\frac{1}{4}+\frac{1}{36}+\frac{1}{633}+\frac{1}{3798} \\ \frac{59}{211}=\frac{1}{6}+\frac{1}{9}+\frac{1}{633}+\frac{1}{3798} \end{gathered}$$由于方法一与方法二中，最小的分数相同，因此二者均是最优解。给出$a,b$，编程计算最好的表达方式。保证最优解满足：最小的分数$\ge \frac{1}{10^7}$。

输入格式：
一行两个整数，分别为$a$和$b$的值。

输出格式：
输出若干个数，自小到大排列，依次是单位分数的分母。

数据范围：
$0<a<b<1000$

思路是迭代加深。其实就是直接枚举每个单位分数的分母，找到最优答案即可。在分解$\frac{a}{b}$的时候，如果不限制深度，容易走到非常深的位置还不一定能找到最优解，所以最好的办法还是迭代加深。对于DFS每一层，设当前深度是$d$，就枚举第$d$个数的分母可以取哪些数。我们可以逐层按照分母从小到大枚举，这样第$d$个数的分母首先要严格大于上一层枚举的分母。其次，设当前层还需要凑出的分数是$\frac{a}{b}$，先对其约分，如果$a=1$了，那么就找到了一个解，查看其是否更优（即其最大分母是不是小于之前找到的解的最大分母），如果更优则覆盖之前的解（当然如果之前没找到解，那么当前解直接覆盖）；否则开始枚举一个分数，设当前枚举的数是$\frac{1}{x}$，首先有$\frac{1}{x}<\frac{a}{b}$，所以有$x>\frac{b}{a}$，由于$a\ne 1$，所以其等价于$x\ge \lfloor \frac{b}{a}\rfloor +1$；此外，由于是按照分母从小到大枚举的，设当前DFS设定的最大深度是$D$，那么还需要枚举$D-d+1$个数（包括本层枚举的数），所以有$(D-d+1)\frac{1}{x}>\frac{a}{b}$，即$x\le \lfloor \frac{b}{a}(D-d+1)\rfloor$。所以只需枚举这个范围内的数即可。代码如下：

#include <iostream>
#include <cstring>
using namespace std;

const int N = 1e7;
int res[N], tmp[N];

// 求a和b的最大公约数，用于约分
long gcd(long a, long b) {
    return b ? gcd(b, a % b) : a;
}

// u表示本层应该从哪里开始枚举分母
bool dfs(int u, long a, long b, int depth, int max_depth) {
    if (depth == max_depth) {
        if (a == 1) {
        	// 如果分母大于了10^7，则与题目给出的数据范围不符，直接返回false
            if (b > 1e7) return false;

            tmp[depth - 1] = (int) b;
            // 如果是第一次算出解，或者已经算出的解不够最优，则更新答案
            if (!res[0] || b < res[depth - 1]) {
                memcpy(res, tmp, depth * (sizeof(int)));
            }

            return true;
        }

        return false;
    }

    bool found = false;
    // 枚举当前层的分数的分母
    for (int i = max(u, (int) (b / a) + 1); i <= b / a * (max_depth - depth + 1); i++) {
    	// 用long以防溢出
        long nx = a * i - b, ny = b * i;
        // 做约分
        long g = gcd(nx, ny);
        nx /= g, ny /= g;
        // 存储一下当前选择
        tmp[depth - 1] = i;
        // 当前选的分母是i，下一层至少要从i + 1开始枚举分母
        if (dfs(i + 1, nx, ny, depth + 1, max_depth)) found = true;
    }

    return found;
}

int main() {
    int a, b;
    scanf("%d%d", &a, &b);

    int g = (int) gcd(a, b);
    a /= g, b /= g;

    int max_depth = 1;
    while (!dfs(2, a, b, 1, max_depth))
        max_depth++;

    for (int i = 0; i < max_depth; i++)
        printf("%d ", res[i]);
    printf("\n");

    return 0;
}

时间复杂度指数级，空间取决于具体输入的搜索深度。