本文参考刘汝佳《算法竞赛入门经典》(第2版)
【理解迭代加深搜】
迭代加深搜,本身是dfs算法,不过使得dfs不再是盲目地一搜到底,使得dfs在一定的搜索深度内结束掉,核心思想就是枚举搜索的深度(如果能枚举到的话,否则不能迭代深搜),而不是枚举整个解答树空间。
针对Egyptian Fractions (HARD version) UVA - 12558这个题目,整个解答树似乎是无穷大的,用bfs显然不行,此题正是用迭代加深搜,枚举深度,对于枚举到的某个深度d,对于分数a/b,展开式子右边分数的分母, 如果分母从e枚举,且枚举到了第i层,那么如果i*(1/e)<=a/b,无解,剪枝,然后分母枚举到e=e+1,改变a/b=a/b-1/e,则如果i*(1/e)<=a/b,无解,剪枝,如此迭代深搜下去,一定会得到结果。这里需要说明下,当然是深度最小的在满足其他条件的情况下是最优解,但我们怎么保证在枚举每个深度的时候,即在某个深度的前提下,怎么保证得到的是该深度下的最优解呢?下面代码中,用到了tmp来更新ans,来保证这一点。
个人有疑问的地方是,对于本身就是埃及分数的,为什么不能这样展开:1/3=1/3,个人觉得题目的叙述有待完善。暂且按照书上的意思,把课本上的代码修改一下,可以AC掉这个题目如下:
#include<iostream>
#include<algorithm>
#include<cstring>
#include<set>
using namespace std;
typedef long long int ll;
const ll maxn=1000000;
ll max_depth;//枚举到的某个深度
ll tmp[maxn];//存放中间过程求出的右式分数分母
ll ans[maxn];//存放最终右式分数分母
bool ok;//标记是否搜索到结果
set<ll>se;//用于判断哪些元素不能使用(已经存在集合中)
//得到第一个i,使得1/i<a/b(本题目说的不清楚,这里默认右式必定有两项以上,不会这样:1/3=1/3)
ll get_first(ll a,ll b){return b/a+1;}
ll gcd(ll a,ll b){return b?gcd(b,a%b):a;}
//判断tmp是否比ans更优
bool better(ll cur){
for(int i=cur;i>=0;i--){
if(ans[i]!=tmp[i]) return ans[i]==-1 || ans[i]>tmp[i];
}
return false;
}
//当前深度是cur,右式分数从分母是from开始枚举,当前要处理的分数是a/b
void dfs(ll cur,ll from,ll a,ll b){
if(cur==max_depth){
if(a!=1 || se.count(b)) return;
tmp[cur]=b;
if(better(cur)) memcpy(ans,tmp,sizeof(ll)*(cur+1));//确保ans存放的是最优结果
ok=true;
return;
}
from=max(from,get_first(a,b));//右边式子分数从分母是from开始枚举
for(int i=from;;i++){
//如果剩下的maxd-cur+1个分数分母都是i,还不能达到a/b,无解,剪枝
if(b*(max_depth-cur+1)<=a*i) return;
if(se.count(i)) continue;
tmp[cur]=i;
//计算a/b-1/i结果,进行迭代深搜
ll bb=b*i;
ll aa=a*i-b;
ll g=gcd(aa,bb);//用于下面约分
dfs(cur+1,i+1,aa/g,bb/g);
}
}
int main()
{
ll T;
ll a,b,k,x,kase=0;
cin>>T;
while(T--){
cin>>a>>b>>k;
se.clear();
for(int i=0;i<k;i++){cin>>x; se.insert(x);}
ok=false;
for(max_depth=1;;max_depth++){
memset(ans,-1,sizeof(ans));
dfs(0,get_first(a,b),a,b);
if(ok) break;
}
cout<<"Case "<<++kase<<": "<<a<<"/"<<b<<"="<<1<<"/"<<ans[0];
for(int i=1;i<=max_depth;i++){
cout<<"+"<<1<<"/"<<ans[i];
}
cout<<endl;
}
return 0;
}