POJ 2645 Boastin' Red Socks(组合数学)

Description
有很多只红色或蓝色的袜子,现从中任意选取两只都是红袜子的概率为p/q,问红色袜子和蓝色袜子的数量,如果有多重可行方案,输出总数最小的那个
Input
两个整数p和q(0< p<=q)
Output
如果存在可行方案则输出总数最小的那个,如果不存在可行方案则输impossible
Sample Input
1 2
6 8
12 2499550020
56 789
0 0
Sample Output
3 1
7 1
4 49992
impossible
Solution
p=0时是0 2
p=q时是2 0(不是1 1)
0< p< q时,首先通分p和q,之后枚举袜子总数i,设红色袜子有j只,由j*(j-1)/(i*(i-1))=p/q,即i*(i-1)p=j(j-1)q,因(p,q)=1,故i(i-1)%q=0,在枚举过程中遇到满足条件的i,就用上式求出j的值,即j=(int)(sqrt(temp+0.5))+1,其中temp=i*(i-1)/q*p,如果j*(j-1)=temp,那么当前为最优方案,如果所有的i都不满足条件则impossible
Code

#include<cstdio>
#include<iostream>
#include<cmath>
using namespace std;
#define maxn 50000
typedef long long ll;
ll gcd(ll a,ll b)
{
    if(b==0)return a;
    return gcd(b,a%b);
}
int main()
{
    ll p,q;
    while(scanf("%lld%lld",&p,&q),p||q)
    {
        if(p==0)printf("0 2\n");
        else if(p==q)printf("2 0\n");
        else
        {
            ll g=gcd(p,q),i,j,temp;
            p/=g,q/=g;
            for(i=2;i<=maxn;i++)
                if(i*(i-1)%q==0)
                {
                    temp=i*(i-1)/q*p;
                    j=(ll)sqrt(temp+0.5)+1;
                    if(j*(j-1)==temp&&j>=2)
                        break;
                }
            if(i>maxn)printf("impossible\n");
            else printf("%lld %lld\n",j,i-j);
        }
    }
    return 0;
}

你可能感兴趣的:(POJ 2645 Boastin' Red Socks(组合数学))