题目链接:登录—专业IT笔试面试备考平台_牛客网
题意很简单,就是给你一个p和q,问能不能找到a和b使得p/q=a/b+b/a.
正解比较简单,鉴于不想打符号了,就把解题过程写在纸上了,大家可以参考下,一会我会讲解另一种做法 我们知道分数是有理数,所以只要p*p-4*q*q不是一个平方数那么就不可能会被表示,反之一定可以被表示,而且表示形式已经被求出,所以这种做法比较简单,下面附上这种做法的代码:
#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
int main()
{
int T;
cin>>T;
while(T--)
{
long long p,q;
scanf("%lld%lld",&p,&q);
if(p*p<4*q*q)
{
puts("0 0");
continue;
}
long long t=sqrt(p*p-4*q*q);
if(t*t!=p*p-4*q*q)
{
puts("0 0");
continue;
}
long long a=p+t,b=2*q;
printf("%lld %lld\n",a/__gcd(a,b),b/__gcd(a,b));
}
return 0;
}
下面来对本道题用另一种方式进行讲解,那就是用质因子爆搜的方式进行搜索,对于题目中所给的等式,有一点是可以确定的,那就是如果p和q是最简分数的话,那么q一定是a和b的最小公倍数,所以我们对q进行质因子分解,不妨设q=x1^y1*x2^y2+……+xn^y2,那么根据最小公倍数的性质可以知道,对于a和b中至少有一个数含有xi^yi,另一个数含有xi的次方可以是0~yi的任何一个数,但不能超过yi,这样的话我们就能先对q进行质因子分解,然后枚举a和b关于xi的次方数来进行搜索答案,因为每个质因子含有的次数不会很多,所以复杂度也不是很高,但还是要比第一种做法复杂的,对了,还需要注意的一个问题就是q等于1的情况,这样q是不含有任何质因子的,我们需要将p和q同时乘以2,人为的添加上一个质因子就可以对其进行正常处理了
下面是代码:
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<queue>
using namespace std;
typedef long long ll;
const int N = 100,M = 100010;
int prime[M],cnt;
bool st[M];
void primes(){
for(int i=2;i<M;i++){
if(!st[i]) prime[++cnt] = i;
for(int j=1;j<=cnt && prime[j]*i < M;j++){
st[i*prime[j]] = true;
if(i % prime[j] == 0) break;
}
}
}
int ans[N],o[N],p,q;;
bool flag;
ll qpow(ll a,ll b)
{
ll ans=1;
while(b)
{
if(b&1) ans*=a;
b>>=1;
a*=a;
}
return ans;
}
void dfs(int pos,ll a,ll b,int tt)
{
if(flag) return ;
if(pos>tt)
{
if(!flag&&a*b*p==a*a*q+b*b*q)
{
int t=__gcd(a,b);
if(a>b) swap(a,b);
a/=t;b/=t;
flag=true;
printf("%lld %lld\n",a,b);
}
return ;
}
ll ttt=qpow(ans[pos],o[pos]);
for(ll i=0,t=1;i<=o[pos];i++,t*=ans[pos])
dfs(pos+1,a*t,b*ttt,tt);
for(ll i=0,t=1;i<=o[pos];i++,t*=ans[pos])
dfs(pos+1,a*ttt,b*t,tt);
}
int main()
{
int T;
cin>>T;
primes();
while(T--)
{
scanf("%d%d",&p,&q);
int t = __gcd(p,q);p /= t;q /= t;
q*=2;p*=2;
int x = q,tt = 0;
memset(o,0,sizeof o);
for(int i=1;i<=cnt && prime[i]*prime[i] <= x;i++){
if(x % prime[i] == 0){
ans[++tt] = prime[i];
while(x % prime[i] == 0) x /= prime[i],o[tt] ++;
}
}
if(x > 1) ans[++tt] = x,o[tt] = 1;
flag=false;
dfs(1,1,1,tt);
if(!flag) printf("0 0\n");
}
return 0;
}