快速幂与组合数

1. 快速幂模板(a^b%p)

#include<iostream>
using namespace std;

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

int main()
{
	int a,b,p;
	cin>>a>>b>>p;
	cout<<qmi(a,b,p)<<endl;
	return 0;
}

2. 快速幂求逆元

费马定理
如果p是质数，且gcd(a,p)=1，那么a^(p-1)恒等于1（mod p）.

题目描述：
给定 n 组 ai,pi，其中 pi 是质数，求 ai 模 pi 的乘法逆元，若逆元不存在则输出 impossible。
乘法逆元的定义：（将除法问题转化为乘法问题）

测试样例：

输入：
3
4 3
8 5
6 3
输出：
1
2
impossible

代码如下：

#include<iostream>
using namespace std;

int qmi(int a,int k,int p)
{
    int res=1;
    while(k)
    {
        if(k&1)
            res=(long long)res*a%p;
        a=(long long)a*a%p;
        k>>=1;
    }
    return res;
}
int main()
{
    int n;
    scanf("%d",&n);
    while(n--)
    {
        int a,p;
        scanf("%d %d",&a,&p);
        int res=qmi(a,p-2,p);
        if(a%p)
            printf("%d\n",res);
        else
            printf("impossible\n");
    }
    return 0;
}

3. a*b%p

#include<iostream>
using namespace std;
typedef unsigned long long ULL;
int main()
{
    ULL a,b,p;
    cin>>a>>b>>p;
    ULL res=0;
    while(b)
    {
        if(b&1)
            res=(res+a)%p;
        b>>=1;
        a=a*2%p;
    }
    cout<<res<<endl;
    return 0;
}

4. 龟速乘模板

龟速乘的作用是修复快速幂的bug，当取模大于大于1e9时，两数相乘会爆long long，因此衍生出龟速乘的写法

long long slow_mul(long long x,long long y,long long mod)
{
    long long ans=0;
    while(y)
    {
        if(y&1) ans=(ans+x)%mod;
        x=(x+x)%mod;
        y=y>>1;
    }
    return ans;
}

long long qmi(long long x,long long y,long long mod)
{
    long long cnt=1;
    while(y)
    {
        if(y&1) cnt=slow_mul(cnt,x,mod)%mod;
        x=slow_mul(x,x,mod)%mod;
        y=y>>1;
    }
    return cnt;
}

5. 求组合数

题目描述：
给定 n 组询问，每组询问给定两个整数 a，b，请你输出 Cbamod(109+7) 的值。
测试样例：

输入：
3
3 1
5 3
2 2
输出：
3
10
1

代码1（1≤n≤10000,1≤b≤a≤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()
{
    init();
    int n,a,b;
    scanf("%d",&n);
    for(int i=0; i<n; i++)
    {
        scanf("%d %d",&a,&b);
        printf("%d\n",c[a][b]);
    }
    return 0;
}

代码2（1≤n≤10000,1≤b≤a≤10^5）:

#include<iostream>
using namespace std;

const int N=100010,mod=1e9+7;
int fact[N],infact[N];
int qmi(int a,int k,int p)
{
    int res=1;
    while(k)
    {
        if(k&1)
            res=(long long)res*a%p;
        a=(long long)a*a%p;
        k>>=1;
    }
    return res;
}
int main()
{
    fact[0]=infact[0]=1;
    for(int i=1; i<=N; i++)
    {
        fact[i]=(long long)fact[i-1]*i%mod;
        infact[i]=(long long)infact[i-1]*qmi(i,mod-2,mod)%mod;
    }
    int n;
    scanf("%d",&n);
    while(n--)
    {
        int a,b;
        scanf("%d %d",&a,&b);
        printf("%d\n",(long long)fact[a]*infact[b]%mod*infact[a-b]%mod);//排列组合公式
    }
    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()
{
    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(a - b, p) - get(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 -- ) printf("%d", res[i]);
    puts("");

    return 0;
}

6. 满足条件的01序列

题目描述：
给定 n 个 0 和 n 个 1，它们将按照某种顺序排成长度为 2n 的序列，求它们能排列成的所有序列中，能够满足任意前缀序列中 0 的个数都不少于 1 的个数的序列有多少个。输出的答案对 109+7 取模。
测试样例：

输入:
3
输出：
5

解题思路：

代码如下：

#include<iostream>
using namespace std;

const int mod=1e9+7;
int qmi(int a,int k,int p)
{
    int res=1;
    while(k)
    {
        if(k&1) res=(long long)res*a%p;
        a=(long long)a*a%p;
        k >>= 1;
    }
    return res;
}
int main()
{
    int n;
    scanf("%d",&n);
    int a=n*2,b=n;
    long long res=1;
    for(int i=a; i>a-b; i--) res=res*i%mod;
    for(int i=1; i<=b; i++) res=res*qmi(i,mod-2,mod)%mod;
    res=res*qmi(n+1,mod-2,mod)%mod;//除n+1
    cout<<res<<endl;
    return 0;
}

7. 问题 L: 组合数问题I

#include<iostream>
using namespace std;

const int N=2010;
int c[N][N],k,arr[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]+c[i-1][j-1])%k;
            if(!c[i][j])
                arr[i+1][j+1]=1;
        }
    }
    for(int i=1;i<N;i++)
    {
        for(int j=1;j<N;j++)
        {
            arr[i][j]=arr[i][j]+arr[i][j-1]+arr[i-1][j]-arr[i-1][j-1];
        }
    }
}
int main()
{
    int t;
    scanf("%d %d",&t,&k);
    init();
    while(t--)
    {
        int a,b;
        scanf("%d %d",&a,&b);
        printf("%d\n",arr[a+1][b+1]);
    }
    return 0;
}