题意:给定一个分数,问用分子为1的分数加和来构成这个分数有多少种方式。要求每种情况分数的个数不超过n,分母乘积不超过a。
思路:搜索。一开始做犯了一个错误导致一直TLE,就是把当前分数和的分子和分母存为全局变量,导致每次dfs之后还要把当前加上的分数剪掉,浪费了太多时间。具体细节见注释!
#include <cstdio> #include <cstring> #define min(a,b) ((a)<(b)?(a):(b)) using namespace std; int p,q,a,n,res; int gcd(int x,int y){ if(!y) return x; return gcd(y, x%y); } int exd(int np,int nq){//判断是否超过了目标值 return q*np>p*nq; } void dfs(int d,int from,int mul,int fz,int fm){ //五个参数:1、搜索深度(最多还可以使用多少个分数); //2、从哪个数开始搜索当前分数; //3、前面选出的分数的分母乘积 //4、前面选出的分数之和的分子 //5、前面选出的分数之和的分母 int i,j,nfz,nfm,lcm; if(!d)//分母不能超过d个 return; for(i = from;i*mul<=a;i++){//从小到大搜索当前分数的分母 lcm = fm*i/gcd(fm,i);//求最小公倍数,作为分数和的分母 nfm = lcm; nfz = lcm/fm*fz+lcm/i;//分数和的分子 j = gcd(nfm,nfz);//化简分数 nfm /= j; nfz /= j; if(p*nfm>q*(lcm/fm*fz+lcm/i*d))//后面的分数都取最大的情况仍然达不到目标值,重要剪枝 return; if(nfz==p && nfm==q){//达到目标值,注意不是返回而是继续 res++; continue; } if(exd(nfz,nfm))//如果和已经超过了目标值,重要剪枝 continue; dfs(d-1, i, mul*i, nfz,nfm); } } int main(){ // freopen("a.txt","r",stdin); while(scanf("%d %d %d %d",&p,&q,&a,&n) && (p+q+a+n)){ int i; res=0; i = gcd(p,q); p /= i; q /= i; dfs(n,1,1,0,1); printf("%d\n",res); } return 0; }