結果
問題 | No.117 組み合わせの数 |
ユーザー | しらっ亭 |
提出日時 | 2016-08-11 01:35:13 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
RE
|
実行時間 | - |
コード長 | 1,708 bytes |
コンパイル時間 | 1,604 ms |
コンパイル使用メモリ | 167,996 KB |
実行使用メモリ | 18,908 KB |
最終ジャッジ日時 | 2024-11-07 10:37:54 |
合計ジャッジ時間 | 3,132 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ソースコード
#include <bits/stdc++.h> using namespace std; #define _p(...) (void)printf(__VA_ARGS__) #define forr(x,arr) for(auto&& x:arr) #define _overload3(_1,_2,_3,name,...) name #define _rep2(i,n) _rep3(i,0,n) #define _rep3(i,a,b) for(int i=int(a);i<int(b);++i) #define rep(...) _overload3(__VA_ARGS__,_rep3,_rep2,)(__VA_ARGS__) #define _rrep2(i,n) _rrep3(i,0,n) #define _rrep3(i,a,b) for(int i=int(b)-1;i>=int(a);i--) #define rrep(...) _overload3(__VA_ARGS__,_rrep3,_rrep2,)(__VA_ARGS__) #define ALL(x) (x).begin(), (x).end() #define BIT(n) (1LL<<(n)) #define SZ(x) ((int)(x).size()) #define fst first #define snd second typedef vector<int> vi;typedef vector<vi> vvi;typedef pair<int,int> pii;typedef vector<pii> vpii; typedef long long ll; const long long mod = 1000000007; long long powMod(long long x, long long y) { long long r = 1, a = x % mod; while (y) { if (y & 1) r = (r * a) % mod; a = a * a % mod; y /= 2; } return r; } #define FACT_MAX 1000001 long long fact[FACT_MAX]; // 階乗 long long revFact[FACT_MAX]; // 階乗の逆元 void setFact() { const int N = FACT_MAX; fact[0] = 1; for (int i = 1; i < N; i++) { fact[i] = fact[i - 1] * i; fact[i] %= mod; } revFact[N - 1] = powMod(fact[N - 1], mod - 2); for (int i = N - 2; i >= 0; i--) { revFact[i] = revFact[i + 1] * (i + 1); revFact[i] %= mod; } } void Main() { char T; int N, K; scanf("%c(%d,%d)\n", &T, &N, &K); if (T == 'H') N += K - 1; ll ans = N < K ? 0 : T == 'P' ? fact[N] * revFact[N-K] % mod : fact[N] * revFact[N-K] % mod * revFact[K] % mod; _p("%lld\n", ans); } int main() { setFact(); int T; scanf("%d\n", &T); while (T--) Main(); return 0; }