結果
問題 | No.117 組み合わせの数 |
ユーザー | freecss00 |
提出日時 | 2020-02-25 23:53:14 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
RE
|
実行時間 | - |
コード長 | 3,204 bytes |
コンパイル時間 | 1,578 ms |
コンパイル使用メモリ | 170,992 KB |
実行使用メモリ | 20,168 KB |
最終ジャッジ日時 | 2024-10-13 14:39:36 |
合計ジャッジ時間 | 2,898 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ソースコード
#include <bits/stdc++.h> using namespace std; #define REP(i,m,n) for(int i=(m); i<(int)(n); i++) #define RREP(i,m,n) for(int i=(int)(n-1); i>=m; i--) #define rep(i,n) REP(i,0,n) #define rrep(i,n) RREP(i,0,n) #define all(a) (a).begin(),(a).end() #define rall(a) (a).rbegin(),(a).rend() #define aut(r,v) __typeof(v) r = (v) #define each(it,o) for(aut(it,(o).begin()); it!=(o).end(); ++it) #define reach(it,o) for(aut(it,(o).rbegin()); it!=(o).rend(); ++it) #define fi first #define se second #define debug(...) {cerr<<"[L"<<__LINE__<<"] "; _debug(__VA_ARGS__);} template<typename T1, typename T2> ostream& operator<<(ostream& o, const pair<T1, T2>& p) {return o<<"("<<p.first<<", "<<p.second<<")";} template<typename T>string join(const vector<T>&v, string del=", ") {stringstream s;rep(i,v.size())s<<del<<v[i];return s.str().substr(del.size());} template<typename T>ostream& operator<<(ostream& o, const vector<T>&v) {if(v.size())o<<"["<<join(v)<<"]";return o;} template<typename T>ostream& operator<<(ostream& o, const vector<vector<T> >&vv) {int l=vv.size();if(l){o<<endl;rep(i,l){o<<(i==0?"[ ":",\n ")<<vv[i]<<(i==l-1?" ]":"");}}return o;} template<typename T>ostream& operator<<(ostream& o, const set<T>& st) {vector<T> v(st.begin(),st.end());o<<"{ "<<join(v)<<" }";return o;} template<typename T1, typename T2>ostream& operator<<(ostream& o, const map<T1, T2>& m) {each(p,m){o<<(p==m.begin()?"{ ":",\n ")<<*p<<(p==--m.end()?" }":"");}return o;} inline void _debug(){cerr<<endl;} template<class First, class... Rest> void _debug(const First& first, const Rest&... rest){cerr<<first<<" ";_debug(rest...);} typedef long long ll; typedef pair<int, int> pii; typedef vector<int> vi; typedef vector<vi> vvi; typedef vector<ll> vl; typedef vector<vl> vvl; const double PI = (1*acos(0.0)); const double EPS = 1e-9; const ll INF = 0x3f3f3f3f; const ll INFL = 0x3f3f3f3f3f3f3f3fLL; const ll mod = 1e9 + 7; inline void finput(string filename) { freopen(filename.c_str(), "r", stdin); } ll mod_pow(ll x, ll n){ ll v = 1; for(; n; x=x*x%mod, n>>=1) if(n & 1) v = v * x % mod; return v; } ll mod_inv(ll x){ return mod_pow(x, mod-2); } struct Factorial{ vector<ll> fact; vector<ll> factinv; Factorial(int n){ fact = vector<ll>(n+1); factinv = vector<ll>(n+1); fact[0] = 1; REP(i,1,n+1) fact[i] = fact[i-1] * i % mod; factinv[n] = mod_inv(fact[n]); RREP(i,1,n+1) factinv[i-1] = factinv[i] * i % mod; } ll comb(int n, int k){ return fact[n] * factinv[n-k] % mod * factinv[k] % mod; } }; int main(){ ios_base::sync_with_stdio(0); // finput("./input"); Factorial fa(1000000); int t; cin >> t; rep(i,t){ string s; cin >> s; char c; int n, k; sscanf(s.c_str(), "%c(%d,%d)", &c, &n, &k); if(c == 'C'){ if(n < k) cout << 0 << endl; else cout << fa.comb(n,k) << endl; } if(c == 'P'){ if(n < k) cout << 0 << endl; else cout << fa.fact[n] * fa.factinv[n-k] % mod << endl; } if(c == 'H'){ cout << fa.comb(n+k-1, n-1) << endl; } } return 0; }