数论--求组合数

下面四个标题题面都是一样的,只是数据范围不一样,所以用到的代码也不一样

一

十万组数据         a,b小于2000

代码:

#include<iostream>

using namespace std;

const int N = 2010,mod = 1e9+7;

int c[N][N];

void init()
{
    for(int i=0;i<N;i++)
        for(int j = 0;j<=i;j++)
            if(!j) c[i][j] = 1;
            else c[i][j] = (c[i-1][j-1]+c[i-1][j]) % mod;
}

int main(void)
{
    init();
    
    int n;
    
    cin >> n;
    
    while(n--)
    {
        int a,b;
        cin >> a >> b;
        cout << c[a][b] << endl;
    }
    
    return 0;
}

二

一万组数据         a,b小于100000

代码:

#include<iostream>

using namespace std;

typedef long long LL;
const int N = 100010,mod = 1e9+7;

int fact[N],infact[N]; //fact是阶乘再取模 infact是逆元

int qmi(int a,int k, int p)
{
    int res = 1;
    while(k)
    {
        if(k & 1) res = (LL)res*a % p;
        a = (LL) a * a % p;
        k>>=1;
    }
    return res;
}

int main(void)
{
    fact[0] = infact[0] = 1;
    for(int i=1;i<N;i++)
    {
        fact[i] = (LL)fact[i-1] * i % mod;
        infact[i] = (LL)infact[i-1] * qmi(i,mod-2,mod) % mod;
    }
    
    int n;
    scanf("%d",&n);
    
    while(n--)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        
        cout << (LL)fact[a] * infact[b] % mod * infact[a-b] % mod << endl;
    }
    
    return 0;
}

三

20组数据         a,b<=10^18,p<=10^5(p是需要对于取模的数)

#include<iostream>

using namespace std;

typedef long long LL;

int p;

int qmi(int a,int k)
{
    int res = 1;
    while(k)
    {
        if(k & 1)res = (LL)res * a % p;
        a = (LL)a * a % p;
        k>>=1;
    } 
    return res;
}

int C(int a,int b)
{
    int res = 1;
    for(int i=1,j = a;i<=b;i++,j--)
    {
        res = (LL)res * j %p ;
        res = (LL)res *qmi(i,p-2) % p;
    }
    return res;
}

int lucas(LL a,LL b)
{
    if(a < p && b < p) return C(a,b);
    return (LL)C(a%p,b%p) * lucas(a/p,b/p) % p;
}

int main(void)
{
    int n;
    cin >> n;
    
    while(n--)
    {
        LL a,b;
        
        cin >> a >> b >> p;
        
        cout << lucas(a,b) << endl;
    }
    
    return 0;
}

四

题面:

代码:

#include<iostream>
#include<algorithm>
#include<vector>

using namespace std;

const int N = 5010;

int primes[N],cnt;
int sum[N];
bool st[N];

void get_primes(int n)
{
    for(int i = 2;i<=n ;i++)
    {
        if(!st[i]) primes[cnt++] = i;
        for(int j = 0;primes[j] <= n/i;j++)
        {
            st[primes[j] * i] = true;
            if(i%primes[j] == 0) break;
        }
    }
}

int get(int n,int p)
{
    int res = 0;
    while(n)
    {
        res+=n / p;
        n /= p;
    }
    return res;
}

vector<int> mul(vector<int>a,int b)
{
    vector<int> c;
    int t=0;
    for(int i=0;i<a.size();i++)
    {
        t+=a[i]*b;
        c.push_back(t%10);
        t/=10;
    }
    while(t)
    {
        c.push_back(t%10);
        t/=10;
    }
    
    return c;
}

int main(void)
{
    int a,b;
    cin >> a >> b;
    
    get_primes(a);
    
    for(int i=0;i<cnt;i++)
    {
        int p = primes[i];
        sum[i] = get(a,p) - get(b,p) - get(a-b,p);
    }
    
    vector<int> res;
    res.push_back(1);
    
    for(int i=0;i<cnt;i++)
    {
        for(int j=0;j<sum[i];j++)
        {
            res = mul(res,primes[i]);
        }
    }
    
    for(int i=res.size() - 1;i>=0;i--) cout << res[i];
    cout << endl;
    
    return 0;
}