codevs 1288 埃及分数

1288 埃及分数

迭代加深搜索+剪枝

#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define N 1000000+15
#define LL long long
LL n,x,y;
LL ans[N],num[N];
bool if_;

inline void read(LL &now)
{
    char ch=getchar(); now=0;
    while(ch>'9'||ch<'0') ch=getchar();
    while(ch>='0'&&ch<='9') now=now*10+ch-'0',ch=getchar();
}

int gcd(LL a,LL b) { return !b?a:gcd(b,a%b); }

void dfs(LL step,LL from,LL m,LL d)
{
    if(step<0)return;
    if(d==1){
        if(step==1&&from>=m)
        {
            if(from<num[1])
            {
                if_=true;
                num[1]=from;
                for(int i=n;i>=2;i--)
                    num[i]=ans[i];
            }
        }
        return ;
    }
    for(LL i=m; ;i++)
    {
        ans[d]=i;
        LL g=gcd(step*i-from,from*i);
        dfs((step*i-from)/g,from*i/g,i+1,d-1);
        if(step*i>from*d) break;
    }
}

int main()
{
    read(x); read(y);
    num[1]=0x7f7f7f7f;
    for(int i=1;;i++)
    {
        n=i,dfs(x,y,1,i);
        if(if_)
        {
            for(int i=n;i;i--) printf("%lld ",num[i]);
            return 0;
        }
    }
    return 0;
}