【ACWing】885. 求组合数 I

题目地址:

https://www.acwing.com/problem/content/887/

给定 n n n组询问,每次询问给定两个正整数 a a a b b b,求 ( a b ) a\choose b (ba)。答案模 1 0 9 + 7 10^9+7 109+7返回。

数据范围:
1 ≤ n ≤ 10000 1\le n\le 10000 1n10000
1 ≤ b ≤ a ≤ 2000 1\le b\le a\le 2000 1ba2000

思路是动态规划。先预处理出来所有的 ( a b ) a\choose b (ba),然后回答每次询问。设 f [ i ] [ j ] = ( i j ) f[i][j]={i\choose j} f[i][j]=(ji),那么 f [ i ] [ 0 ] = 1 f[i][0]=1 f[i][0]=1,并且 f [ i ] [ j ] = f [ i − 1 ] [ j − 1 ] + f [ i − 1 ] [ j ] f[i][j]=f[i-1][j-1]+f[i-1][j] f[i][j]=f[i1][j1]+f[i1][j]代码如下:

#include <iostream>
using namespace std;

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

int f[N][N];

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

int main() {
    
    
    int n;
    cin >> n;

    init();

    while (n--) {
    
    
        int a, b;
        cin >> a >> b;
        cout << f[a][b] << endl;
    }

    return 0;
}

预处理时间复杂度 O ( N 2 ) O(N^2) O(N2),每次询问 O ( 1 ) O(1) O(1),空间 O ( N 2 ) O(N^2) O(N2) N N N是询问的参数范围。

猜你喜欢

转载自blog.csdn.net/qq_46105170/article/details/113967199