方法一:
由组合数公式
进行递推
其实就是求出杨辉三角
1
1 1
1 2 1
1 3 3 1
1 4 6 4 1
1 5 10 10 5 1
1 6 15 20 15 6 1...
这个两重循环就能求,当枚举的是每一行的最左边元素,就c[i][0] = 1
否则c[i][j] = c[i - 1][j] + c[i - 1][j - 1]
题目链接:计算组合数
#include <bits/stdc++.h>
using namespace std;
int c[25][25];
int main() {
int n, k;
cin >> n >> k;
for (int i = 0; i <= n; i++) {
for (int j = 0; j <= i; j++) {
if (j == 0) c[i][j] = 1;
else c[i][j] = c[i - 1][j] + c[i - 1][j - 1];
}
}
cout << c[n][k] << endl;
return 0;
}
时间复杂度:O()
方法二:
由组合数公式
直接计算
因为式子里有阶乘,又有除法取余,所以要先预处理阶乘和阶乘的逆元
题目链接:求组合数
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
ll f[100005], invf[100005], mod = 1e9 + 7;
ll pow(ll a, ll b, ll mod) {
ll res = 1;
while (b) {
if (b & 1) {
res = res * a % mod;
}
a = a * a % mod;
b >>= 1;
}
return res % mod;
}
void init() {
f[0] = 1, invf[0] = 1;
for (ll i = 1; i <= 100000; i++) {
f[i] = f[i - 1] * i % mod;
invf[i] = pow(f[i], mod - 2, mod);
}
}
ll C(ll a, ll b, ll mod) {
return f[a] * invf[a - b] % mod * invf[b] % mod;
}
int main() {
init();
ll n, m;
cin >> n >> m;
cout << C(n, m, mod) << endl;
return 0;
}
时间复杂度:O()