ACM学习历程—HDU 5446 Unknown Treasure(数论)（2015长春网赛1010题）

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <set>
#include <queue>
#include <vector>
#define LL long long
#define UNIT 10

using namespace std;

const int maxK = 15;
LL n, m, s[maxK], prime[maxK];
int k;

struct Bignum
{
    int val[105];
    int len;

    Bignum()
    {
        memset(val, 0, sizeof(val));
        len = 1;
    }

    Bignum operator=(const LL &a)
    {
        LL t, p = a;
        len = 0;
        while (p >= UNIT)
        {
            t = p - (p/UNIT)*UNIT;
            p = p / UNIT;
            val[len++] = t;
        }
        val[len++] = p;
        return *this;
    }

    Bignum operator*(const Bignum &a) const
    {
        Bignum x;
        int i, j, up;
        int x1, x2;
        for (i = 0; i < len; i++)
        {
            up = 0;
            for (j = 0; j < a.len; j++)
            {
                x1 = val[i]*a.val[j] + x.val[i+j] + up;
                if (x1 >= UNIT)
                {
                    x2 = x1 - x1/UNIT*UNIT;
                    up = x1 / UNIT;
                    x.val[i+j] = x2;
                }
                else
                {
                    up = 0;
                    x.val[i+j] = x1;
                }
            }
            if (up != 0)
                x.val[i+j] = up;
        }
        x.len = i + j;
        while (x.val[x.len-1] == 0 && x.len > 1)
            x.len--;
        return x;
    }

    LL operator%(const LL &a) const
    {
        LL x = 0;
        for (int i = len-1; i >= 0; --i)
            x = ((x*UNIT)%a+val[i]) % a;
        return x;
    }
};

LL mulMod(LL x, LL y, LL p)
{
    LL ans = 0;
    Bignum xx, yy;
    xx = x;
    yy = y;
    xx = xx*yy;
    ans = xx%p;
    return ans;
}

LL quickMod(LL a, LL b, LL p)
{
    LL ans = 1;
    a %= p;
    while (b)
    {
        if (b&1)
        {
            ans = ans*a%p;
            b--;
        }
        b >>= 1;
        a = a*a%p;
    }
    return ans;
}

LL C(LL n, LL m, LL p)
{
    if (m > n)
        return 0;
    LL ans = 1;
    for(int i = 1; i <= m; i++)
    {
        LL a = (n+i-m)%p;
        LL b = i%p;
        ans = ans*(a*quickMod(b, p-2, p)%p)%p;
    }
    return ans;
}

LL Lucas(LL x, LL y, LL p)
{
    if (y == 0)
        return 1;
    return C(x%p, y%p, p)*Lucas(x/p, y/p, p)%p;
}

//EXGCD
//求解方程ax+by=d,即ax=d mod(b)
//扩展可求逆元
//O(logn)
void exgcd(LL a, LL b, LL &x, LL &y, LL &d)
{
    if (b == 0)
    {
        x = 1;
        y = 0;
        d = a;
    }
    else
    {
        exgcd(b, a%b, y, x, d);
        y -= a/b*x;
    }
}

//中国剩余定理（互质）
//其中a为除数数组,n为模数数组
LL CRT(LL *a, LL *n, int len)
{
    LL N = 1, ans = 0;
    for (int i = 0; i < len; i++)
    {
        N *= n[i];
    }
    for (int i = 0; i < len; i++)
    {
        LL m = N / n[i];
        LL x, y, d;
        exgcd(m, n[i], x, y, d);
        x = (x%n[i] + n[i]) % n[i];
        //ans = (ans + m*a[i]*x%N) % N;
        ans = (ans + mulMod(mulMod(m, a[i], N), x, N)) % N;
    }
    return ans;
}

void input()
{
    scanf("%I64d%I64d%d", &n, &m, &k);
    for (int i = 0; i < k; ++i)
        scanf("%I64d", &prime[i]);
}

void work()
{
    for (int i = 0; i < k; ++i)
        s[i] = Lucas(n, m, prime[i]);
    LL ans = CRT(s, prime, k);
    printf("%I64d\n", ans);
}

int main()
{
    //freopen("test.in", "r", stdin);
    int T;
    scanf("%d", &T);
    for (int times = 0; times < T; ++times)
    {
        input();
        work();
    }
    return 0;
}