結果
問題 | No.117 組み合わせの数 |
ユーザー | karinohito |
提出日時 | 2021-07-26 11:26:55 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,534 bytes |
コンパイル時間 | 1,759 ms |
コンパイル使用メモリ | 176,772 KB |
実行使用メモリ | 50,064 KB |
最終ジャッジ日時 | 2024-07-22 00:11:17 |
合計ジャッジ時間 | 3,186 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ソースコード
//#include <atcoder/all> #include <bits/stdc++.h> using namespace std; //using namespace atcoder; using ll = long long; #define all(A) A.begin(),A.end() using vll = vector<ll>; #define rep(i, n) for (long long i = 0; i < (long long)(n); i++) using Graph = vector<vector<ll>>; Graph G; vll dist; vector<bool> seen; set<pair<ll, ll>> S; ll gcd(ll(a), ll(b)) { ll c = a; while (a % b != 0) { c = a % b; a = b; b = c; } return b; } vector<ll> fact, factinv, inv; ll mod = 1e9 + 7; void prenCkModp(ll n) { fact.resize(n + 5); factinv.resize(n + 5); inv.resize(n + 5); fact.at(0) = fact.at(1) = 1; factinv.at(0) = factinv.at(1) = 1; inv.at(1) = 1; for (ll i = 2; i < n + 5; i++) { fact.at(i) = (fact.at(i - 1) * i) % mod; inv.at(i) = mod - (inv.at(mod % i) * (mod / i)) % mod; factinv.at(i) = (factinv.at(i - 1) * inv.at(i)) % mod; } } ll nCk(ll n, ll k) { if (n < k) return 0; return fact.at(n) * (factinv.at(k) * factinv.at(n - k) % mod) % mod; } ll nPk(ll n,ll k){ if (n < k) return 0; return fact.at(n) * (factinv.at(n - k) % mod) % mod; } int main() { ll N; cin>>N; prenCkModp(1e6*2); rep(i,N){ string S; cin>>S; ll A=0,B=0; rep(k,S.size()){ if(S[k]=='('){ k++; while(S[k]!=','){ A+=S[k]-'0'; A*=10; k++; } k++; while(S[k]!=')'){ B+=S[k]-'0'; B*=10; k++; } } } A/=10;B/=10; if(S[0]=='C'){ cout<<nCk(A,B)<<endl; } else if(S[0]=='P'){ cout<<nPk(A,B)<<endl; } else{ cout<<nCk(A+B-1,B)<<endl; } } }