蓝桥杯ACwing习题

1.https://www.acwing.com/problem/content/description/308/

#include <iostream>
#include <cstring>
#include <algorithm>
#define x first
#define y second

using namespace std;

const int N = 1e6 + 10 , mod = 1e9 + 7;

typedef pair<int, int> PII;
typedef long long LL;

PII q[2010];
int n , m , k;
LL f[N];
LL fac[N] , infac[N];

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

LL C(int a ,int b)
{
    return (fac[a] * infac[b] % mod * infac[a - b]) % mod;
}

int main()
{
    cin >> n >> m >> k;
    
    for(int i = 0 ; i < k ; i ++) cin >> q[i].x >> q[i].y;
    
    q[k] = {n , m}; // 设置最后一个点为不能走的点 ，所以最后一个点必定会被枚举到 
    sort(q , q + k + 1); // sort 前面是从 第几个数开始排序，排序多少个

    // 求逆元，费马小定理,求 组合数的O(N) 方法
    fac[0] = 1 , infac[0] = 1;
    for(int i = 1 ; i < N ; i ++)
    {
        fac[i] = (fac[i - 1] * i) % mod;
        infac[i] = infac[i - 1] * qmi(i , mod - 2 , mod) % mod;
    }
    
    // 思路 f[i] 表示的是到当前的禁点不经过前面所有的禁点的路径数
    // 同理 f[j] 表示的就是到 j 点不经过前面禁点的数
    // 但由于 j 点是一个禁点 ，所以 从 j 到 i 的路径一定是不合法的
    // i 可以由 j 转移而来 所以从起点到终点一定不会经过禁点
    // 枚举所有的 i , i 由所有坐标在 i 之前的 j 转移而来 
    // 用从左上角到当前的点的所有方案数 - 所有不合法的方案数就是ans
    // f[i] = f[j] * C(x - a + y - b , x - a) 后面的方案数是由 j - i 的方案数
    
    f[0] = 1;
    for(int i = 0 ; i <= k ; i ++)
    {
        int x = q[i].x , y = q[i].y;
        f[i] = C(x + y - 2 , y - 1); // - 2 是因为第一个点已经固定了
        for(int j = 0 ; j < i ; j ++)
        {
            int a = q[j].x , b = q[j].y;
            if(x >= a && y >= b)
            f[i] = (f[i] - f[j] * C(x - a + y - b , y - b) % mod) % mod; 
        }
    }
    
    cout << (f[k] + mod) % mod;
    
    return 0;
}