[POJ 1980] Unit Fraction Partition Dfs

http://poj.org/problem?id=1980

题意：输入四个数 p, q, a, n 其中 p 是分子， q 是分母，将这个分数拆成不超过 n 个分子为 1 的分数相加，而且这些分子的乘积不能超过 a 。问拆分的总数 。

思路：Dfs，主要是要剪枝。

#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>

using namespace std;

const double e = 1e-8;

int a, n;
int counts;

int Dfs(int top, int pre, int mul, double sum) //枚举top个了，分母从pre开始，分母乘积为mul，剩下的和是 sum
{
    if(top > n) //枚举超出 n 个了
        return 0;
    if(fabs(sum) < e){ //sum == 0
        counts++;
        return 0;
    }
    double len = a/mul; //最大的分母
    if(1.0 / len > sum + e)  //剪枝，最小的分数也比剩下的多
        return 0;
    for(int i = pre; i <= len; i++){
        if(mul * i > a) //乘积超过 a 
            return 0;
        if((n-top)*1.0/i < sum - e) //剩下的最大的和 (n-top)*1.0/i 小于sum
            return 0;
        if(1.0/i > sum + e) //太大了
            continue;
        Dfs(top+1, i, mul*i, sum-(1.0/i));
    }
}

int main()
{
    double p, q;
    while(scanf("%lf%lf%d%d", &p, &q, &a, &n) && (p||q||a||n)){
        counts = 0;
        Dfs(0, ((int)q % (int)p == 0 ? q/p : q/p+1), 1, p / q);
        printf("%d\n", counts);
    }
    return 0;
}