AcWing 1120. 埃及分数【IDA*】【迭代加深+剪枝】

Description:

题目来源：https://www.acwing.com/problem/content/description/1122/

Answer:

使用IDA*算法。

IDA*算法是基于迭代加深的A*算法，迭代加深只有在状态呈指数级增长时才有较好的效果，而A*就是为了防止状态呈指数级增长的。

IDA*算法其实是同时运用迭代加深与全局最优性剪枝。

如果要寻找长度最短且最小分数最大的最优方案，可以考虑搜索分数项最少的所有情况，然后找到最优解。这样得到的最优解也是全局最优解。

很容易想到使用dfs或者bfs，但是dfs深度是没有上限的，很有可能会爆栈，bfs宽度也没有上限，可能连第一层都扩展不完，于是我们要进行优化。

迭代加深： 因为我们要求的是分数项的个数最少，也就是深度较浅的时候，那么我们可以从小到达枚举搜索的深度$maxd(1-maxd)$，如果在某一个深度的搜索中找到了可行解，那么我们可以继续进行搜索了。这个叫做迭代加深搜索。

剪枝： 深度的问题解决了，接下来就该解决宽度的问题了。假设当前的分数项为$\frac{1}{x_k}$，当前深度为$d$，那么我们还剩下$maxd-d+1$的深度还没搜索。假设剩下的$maxd-d+1$项都为$\frac{1}{x_k}$，但是我们知道搜索的分数项的是一次比一次小的，并且互不相同，所以当满足$\frac{(maxd-d+1)}{x_k}<=\frac{a}{b}$的时候，那么这个解就永远找不到了，就没有必要继续搜索了。因为我们每一项都是以最大的那项假设的，如果这都达不到$\frac{a}{b}$，那么其他的就更不可能满足了，因为接下来搜索的分数项都比当前的小。

算法过程：

• 枚举搜索的最大深度$maxd$;
• 当搜索到某个$maxd$的时候，找到了可行解，那么这些可行解中的最优解就是全局最优解;

dfs过程：

• 递归的终点：深度达到$maxd$​;
  • 判断最后一项是否满足要求。
    • 如果最后一项$\frac{a_i}{b_i}$不是埃及分数，那么肯定不是可行解，直接退出；
    • 如果最后一项$\frac{a_i}{b_i}$​比前一项的还大，那么肯定也不是可行解，直接退出
  • 如果最后满足要求，说明是一个可行解，那么尝试更新ans。
• 枚举下一层的起始数，直到不可能在$maxd$深度完成为止。

AC代码如下：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
ll maxd;
//第一层存放在了0这个位置
ll ans[1000];//全局最优解
ll temp[1000];//某次搜索的可行解
//比较temp和ans哪个更优
bool cmp(int d){
    //如果ans[d-1]这个位置上为0 说明整个数组是空的 也就是第一次 直接返回true
    //因为进行比较一定是同一层的，解的长度一定相同。
    //因为我们只要在比较浅的层找到解，我们就不会继续往下搜索了。
    if(!ans[d-1]) return true;
    //temp最后一个分数项的分母比较小，所以temp更优
    return temp[d-1] < ans[d-1];
}
//求最大公约数
ll gcd(ll a, ll b){
    return a%b? gcd(b, a%b):b;
}
//获得小于a/b的最大的一个分数 的 分母
ll getFirst(ll a, ll b){
    return b/a+1;
}
//d 搜索的深度 last 当前最大的分数项，也就是最小的分母 目标分解的分数：a/b
bool dfs(ll d, ll last, ll a, ll b){
    if(d == maxd){//迭代的终点
        //无可行解 判断最后一个解是否符合要求
        if(b%a || b>1e7) return false;
        //最后一项的分母1/（a/b）= b/a
        temp[d-1] = b/a;
        //最后一项比前一项大 也就是最后一项的分母比前一项小 是不符合要求的。
        if(temp[d-1] <= temp[d-2]) return false;
        
        //找到可行解，更新最优解
        if(cmp(d)){
            memcpy(ans, temp, sizeof ans);
        }
        
        return true;
    }
    
    bool flag = false;
    //保证每层搜索的起点一定要>= a1/b1的最大分数项的分母
    //例如 19/45 不写这句话 搜索的结果为 3 12 180 
    //第一层试探到3 接下来进行第二层用的i+1=4，这个4必须满足是大于a1/b1的最大分数项的分母的，要不然不成立
    last = max(last, getFirst(a, b));
    for(ll i=last; ;i++){
        //剪枝 
        //假设剩下的层数(maxd-d+1)全部都是当前的分数项，如果这些项加起来比a/b小或者等于，那么一定是不符合要求的。
        if((maxd-d+1)*b <= i*a) break;
        
        //当前分数项的分母为i
        temp[d-1] = i;
        //进行下一层的搜索
        ll a1 = a*i-b;
        ll b1 = b*i;
        ll num = gcd(a1, b1);
        a1/=num;
        b1/=num;
        if(dfs(d+1, i+1, a1, b1)) flag = true;
    }
    return flag;
}

int main(){
    ll a, b;
    cin>>a>>b;
    int num  = gcd(a, b);
    a/=num;
    b/=num;
    
    for(maxd = 1; ; maxd++){
        //d=1, 从第一次层开始遍历
        if(dfs(1, getFirst(a, b), a, b)) break;
    }
    //输出全局最优解
    for(int i=0; i<maxd; i++){
        cout<<ans[i]<<" ";
    }
    cout<<endl;
    
    return 0;
}