UVa 10277 - Boastin' Red Socks (枚举)

题意：

已知你有至少2只不多于50000只袜子。有两种颜色，红色和黑色。

现在已知你从所有袜子中恰好选择2只红袜子的概率为p/q。

现在告诉你p和q，求有多少只红袜子和多少只黑袜子。

算法：

由于袜子总数最多为50000。

设袜子总数为sum，红袜子总数为red。

那么从所有袜子中恰好选择2只红袜子的概率为：

C(red,2)/C(sum,2) = red!/(2!*(red-2)!)  * （sum!/（2!*(sum-2)!) 

                              = red*(red-1)/(sum*(sum-1)) 

设p、q互素。

也就是说red*(red-1)为p的倍数，sum*(sum-1)为q的倍数。

且它们倍数关系相同。red*(red-1)/p = sum*(sum-1)/q。

这样，就得到算法：

先求p，q的最大公约数，将p/q化到最简。

然后在2到50000中枚举sum，判断sum*(sum-1)是否为q的倍数。

再在通过是多少倍，判断是否存在一个整数red，使得red*(red-1) = t*p，t = sum*(sum-1)/q。

当p为0的时候求gcd会使除数为0.故要特判。

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

using namespace std;

typedef long long ll;

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

int main()
{
    ll p,q;
    while(scanf("%lld%lld",&p,&q)!=EOF)
    {
        if(p==0 && q==0)
            break;
        if(p==0)
        {
            printf("0 2\n");
            continue;
        }
        int g = gcd(p,q);
        p/=g;
        q/=g;
        int sum = 0,red = 0;
        for(int i=2;i<=50000;i++)
        {
            ll tmp = (ll)i*(i-1);
            sum = i;
            if(tmp>=q && tmp%q==0)
            {
                ll b = tmp/q;
                ll x = p*b;
                ll t = sqrt(x+0.5);
                if(t*(t+1)==x)
                    red = t+1;
            }
            if(red && sum)
                break;
        }
        if(!red || !sum)
            printf("impossible\n");
        else printf("%d %d\n",red,sum-red);
    }
    return 0;
}