uva 10277 Boastin' Red Socks

概率题

给你p和q，表示概率p/q，当有n个红袜子，m个黑袜子时，取两次，两次都是红袜子的概率为p/q，输出n和m，如果不可能则输出impossible

设总袜子数为b，红袜子数为a，则a/b*(a-1)/(b-1)=p/q , 可见分子p是a*(a-1)的形式，分母q也是这种形式

所以我们用暴力来枚举i*(i-1)分母，并且当分子也能写成j*(j+1)的形式时就找到了解

其实这题意义不大也不是什么经典题，能AC也就行了不必太抠

#include <cstdio>
#include <cmath>

long long gcd(long long a , long long b)
{ return b==0 ? a : gcd(b,a%b); }

int main()
{
    long long p,q;
    while(scanf("%lld%lld",&p,&q))
    {
       if(!p && !q) break;
       if(p==q)  //概率为1
       { printf("2 0\n"); continue;}
       if(p==0)  //概率为0
       { printf("0 2\n"); continue;}
       long long g=gcd(p,q);
       p/=g; q/=g;
       long long i,j;
       for(i=2; i<=50000; i++)
           if(i*(i-1)%q==0)
           {
                long long n=i*(i-1)/q;
                long long m=n*p;
                j=(long long)sqrt(m+0.5);
                if(j*(j+1)==m && j+1>=2)
                    break;
           }
        if(i>50000) printf("impossible\n");
        else        printf("%lld %lld\n",j+1,i-j-1);
    }
    return 0;
}