結果
問題 | No.117 組み合わせの数 |
ユーザー | はまやんはまやん |
提出日時 | 2017-05-26 12:00:34 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 1,132 ms / 5,000 ms |
コード長 | 2,316 bytes |
コンパイル時間 | 1,420 ms |
コンパイル使用メモリ | 168,676 KB |
実行使用メモリ | 19,456 KB |
最終ジャッジ日時 | 2024-09-19 20:04:02 |
合計ジャッジ時間 | 4,118 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ソースコード
#include<bits/stdc++.h> #define rep(i,a,b) for(int i=a;i<b;i++) using namespace std; void _main(); int main() { cin.tie(0); ios::sync_with_stdio(false); _main(); } //--------------------------------------------------------------------------------------------------- int mod = 1000000007; int add(int x, int y) { return (x += y) >= mod ? x - mod : x; } template<class... T> int add(int x, T... y) { return add(x, add(y...)); } int mul(int x, int y) { return 1LL * x * y % mod; } template<class... T> int mul(int x, T... y) { return mul(x, mul(y...)); } int sub(int x, int y) { return add(x, mod - y); } int modpow(int a, long long b) { int ret = 1; while (b > 0) { if (b & 1) ret = 1LL * ret * a % mod; a = 1LL * a * a % mod; b >>= 1; } return ret; } int modinv(int a) { return modpow(a, mod - 2); } #define FAC_MAX 2010101 int fac[FAC_MAX], ifac[FAC_MAX]; void initfac() { fac[0] = ifac[0] = 1; rep(i, 1, FAC_MAX) fac[i] = 1LL * i * fac[i - 1] % mod; rep(i, 1, FAC_MAX) ifac[i] = modinv(fac[i]); } int aPb(int a, int b) { if (b < 0 || a < b) return 0; return mul(fac[a], ifac[a - b]); } int aCb(int a, int b) { if (b < 0 || a < b) return 0; return mul(fac[a], ifac[a - b], ifac[b]); } int nHk(int n, int k) { if (n == 0 && k == 0) return 1; if (n <= 0 || k < 0) return 0; return aCb(n + k - 1, k); } /*--------------------------------------------------------------------------------------------------- ∧_∧ ∧_∧ (´<_` ) Welcome to My Coding Space! ( ´_ゝ`) / ⌒i / \ | | / / ̄ ̄ ̄ ̄/ | __(__ニつ/ _/ .| .|____ \/____/ (u ⊃ ---------------------------------------------------------------------------------------------------*/ //--------------------------------------------------------------------------------------------------- void _main() { initfac(); int T; cin >> T; rep(t, 0, T) { string s; cin >> s; char c; int x, y; sscanf(s.c_str(), "%c(%d,%d)", &c, &x, &y); int ans; if (c == 'C') ans = aCb(x, y); else if (c == 'P') ans = aPb(x, y); else ans = nHk(x, y); printf("%d\n", ans); } }