zoukankan      html  css  js  c++  java
  • 【模板】组合数取模

    (N le 2000, M le 2000)
    直接利用递推式预处理即可。
    代码如下

    #include <bits/stdc++.h>
    
    using namespace std;
    
    const int mod = 1e9 + 7;
    
    int main() {
    	ios::sync_with_stdio(false);
    	cin.tie(0), cout.tie(0);
    	int n;
    	cin >> n;
    	static int f[2010][2010];
    	for (int i = 0; i <= 2000; i++) {
    		f[i][0] = 1;
    	}
    	for (int i = 1; i <= 2000; i++) {
    		for (int j = 1; j <= 2000; j++) {
    			f[i][j] = (f[i - 1][j] + f[i - 1][j - 1]) % mod;
    		}
    	}
    	while (n--) {
    		int a, b;
    		cin >> a >> b;
    		cout << f[a][b] << endl;
    	}
    	return 0;
    }
    

    (N, M le 1e5)
    预处理出阶乘和阶乘的逆元,利用组合数的定义直接回答。

    #include <bits/stdc++.h>
    
    using namespace std;
    
    typedef long long LL;
    
    const int mod = 1e9 + 7;
    const int maxn = 1e5 + 10;
    
    LL fpow(LL a, LL b) {
    	LL ret = 1 % mod;
    	for (; b; b >>= 1, a = a * a % mod) {
    		if (b & 1) {
    			ret = ret * a % mod;
    		}
    	}
    	return ret;
    }
    
    int main() {
    	ios::sync_with_stdio(false);
    	cin.tie(0), cout.tie(0);
    	int n;
    	cin >> n;
    	vector<LL> inv(maxn), fac(maxn);
    	auto prework = [&]() {
    		fac[0] = 1;
    		for (int i = 1; i <= 1e5; i++) {
    			fac[i] = fac[i - 1] * i % mod;
    		}
    		inv[1e5] = fpow(fac[1e5], mod - 2);
    		for (int i = 1e5 - 1; i >= 0; i--) {
    			inv[i] = inv[i + 1] * (i + 1) % mod;
    		}
    	};
    	prework();
    	auto get = [&](int a, int b) {
    		return fac[a] * inv[a - b] % mod * inv[b] % mod;
    	};
    	while (n--) {
    		int a, b;
    		cin >> a >> b;
    		cout << get(a, b) << endl;
    	}
    	return 0;
    }
    /*
    (a, b) = a! / ((a - b)! * b!)
    a! = a * (a - 1)!
    a!' * a = (a - 1)!'
    */
    

    (N, M le 1e18, p le 1e5,p in prime)
    预处理出 (1...p) 的阶乘和阶乘的逆元,用卢卡斯定理进行回答。
    代码如下

    #include <bits/stdc++.h>
    
    using namespace std;
    
    typedef long long LL;
    
    LL fpow(LL a, LL b, LL c) {
    	LL ret = 1 % c;
    	for (; b; b >>= 1, a = a * a % c) {
    		if (b & 1) {
    			ret = ret * a % c;
    		}
    	}
    	return ret;
    }
    
    int main() {
    	ios::sync_with_stdio(false);
    	cin.tie(0), cout.tie(0);
    	int n;
    	cin >> n;
    	while (n--) {
    		LL a, b, p;
    		cin >> a >> b >> p;
    		vector<LL> inv(p + 1), fac(p + 1);
    		fac[0] = 1;
    		for (int i = 1; i <= p; i++) {
    			fac[i] = fac[i - 1] * i % p;
    		}
    		inv[p - 1] = fpow(fac[p - 1], p - 2, p);
    		for (int i = p - 2; i >= 0; i--) {
    			inv[i] = inv[i + 1] * (i + 1) % p;
    		}
    		auto C = [&](LL x, LL y) -> LL {
    			if (x < y) {
    				return 0;
    			}
    			return fac[x] * inv[x - y] % p * inv[y] % p;
    		};
    		function<LL(LL, LL)> Lucas = [&](LL x, LL y) -> LL {
    			if (y == 0) {
    				return 1;
    			}
    			return C(x % p, y % p) * Lucas(x / p, y / p) % p;
    		};
    		cout << Lucas(a, b) << endl;
    	}
    	return 0;
    }
    
  • 相关阅读:
    [CF997E] Good SubSegment
    CF916E
    BZOJ2006 超级钢琴
    BZOJ4571
    凸包总结
    树形DP入门
    bzoj4300 绝世好题(位运算+DP)
    bzoj4552 [Tjoi2016&Heoi2016]排序 (线段树+二分)
    SP1716 GSS3
    Noip2009 Hankson 的趣味题 (简单数学)
  • 原文地址:https://www.cnblogs.com/wzj-xhjbk/p/11607902.html
Copyright © 2011-2022 走看看