結果
問題 | No.117 組み合わせの数 |
ユーザー | Haar |
提出日時 | 2019-06-24 12:08:02 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 219 ms / 5,000 ms |
コード長 | 3,500 bytes |
コンパイル時間 | 1,874 ms |
コンパイル使用メモリ | 171,792 KB |
実行使用メモリ | 34,560 KB |
最終ジャッジ日時 | 2024-06-10 23:28:45 |
合計ジャッジ時間 | 2,954 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ソースコード
#include <bits/stdc++.h> #define LLI long long int #define FOR(v, a, b) for(LLI v = (a); v < (b); ++v) #define FORE(v, a, b) for(LLI v = (a); v <= (b); ++v) #define REP(v, n) FOR(v, 0, n) #define REPE(v, n) FORE(v, 0, n) #define REV(v, a, b) for(LLI v = (a); v >= (b); --v) #define ALL(x) (x).begin(), (x).end() #define RALL(x) (x).rbegin(), (x).rend() #define ITR(it, c) for(auto it = (c).begin(); it != (c).end(); ++it) #define RITR(it, c) for(auto it = (c).rbegin(); it != (c).rend(); ++it) #define EXIST(c,x) ((c).find(x) != (c).end()) #define fst first #define snd second #define popcount __builtin_popcount #define UNIQ(v) (v).erase(unique(ALL(v)), (v).end()) #define bit(i) (1LL<<(i)) #define sz(v) ((LLI)(v).size()) #ifdef DEBUG #include <misc/C++/Debug.cpp> #else #define dump(...) ((void)0) #endif #define gcd __gcd using namespace std; template <class T> constexpr T lcm(T m, T n){return m/gcd(m,n)*n;} template <typename I> void join(ostream &ost, I s, I t, string d=" "){for(auto i=s; i!=t; ++i){if(i!=s)ost<<d; ost<<*i;}ost<<endl;} template <typename T> istream& operator>>(istream &is, vector<T> &v){for(auto &a : v) is >> a; return is;} template <typename T, typename U> istream& operator>>(istream &is, pair<T,U> &p){is >> p.first >> p.second; return is;} template <typename T, typename U> bool chmin(T &a, const U &b){return (a>b ? a=b, true : false);} template <typename T, typename U> bool chmax(T &a, const U &b){return (a<b ? a=b, true : false);} template <typename T, size_t N, typename U> void fill_array(T (&a)[N], const U &v){fill((U*)a, (U*)(a+N), v);} LLI power(LLI n, LLI p, LLI m){ LLI ret = 1; while(p>0){ if(p&1) (ret *= n) %= m; (n *= n) %= m; p /= 2; } return ret; } LLI mod_inv(LLI n, LLI p){return power(n,p-2,p);} template <LLI MOD> class Combination{ public: static vector<LLI> facto; static vector<LLI> ifacto; static void init(int N){ facto.assign(N+1, 1); ifacto.assign(N+1, 1); FORE(i,1,N){ (facto[i] = facto[i-1] * i) %= MOD; } ifacto[N] = mod_inv(facto[N],MOD); REV(i,N-1,0){ ifacto[i] = ifacto[i+1] * (i+1) % MOD; } } static LLI factorial(LLI i){ assert(i < facto.size()); return facto[i]; } static LLI factorial_inverse(LLI i){ assert(i < ifacto.size()); return ifacto[i]; } static LLI P(LLI n, LLI k); static LLI C(LLI n, LLI k); static LLI H(LLI n, LLI k); static LLI stirling_number(LLI n, LLI k); static LLI bell_number(LLI n, LLI k); }; template <LLI MOD> vector<LLI> Combination<MOD>::facto = vector<LLI>(); template <LLI MOD> vector<LLI> Combination<MOD>::ifacto = vector<LLI>(); template <LLI MOD> LLI Combination<MOD>::H(LLI n, LLI k){ if(n == 0 and k == 0) return 1; return C(n+k-1, k); } template <LLI MOD> LLI Combination<MOD>::C(LLI n, LLI k){ if(n < k or n < 0 or k < 0) return 0; return P(n,k) * factorial_inverse(k) % MOD; } template <LLI MOD> LLI Combination<MOD>::P(LLI n, LLI k){ if(n < k or n < 0 or k < 0) return 0; return factorial(n) * factorial_inverse(n-k) % MOD; } const LLI mod = 1e9+7; int main(){ cin.tie(0); ios::sync_with_stdio(false); using C = Combination<mod>; C::init(2000000); int T; scanf("%d",&T); char c; int n,k; while(T--){ scanf(" %c(%d,%d)", &c, &n, &k); if(c == 'C'){ cout << C::C(n,k) << endl; }else if(c == 'P'){ cout << C::P(n,k) << endl; }else{ cout << C::H(n,k) << endl; } } return 0; }