結果
問題 | No.117 組み合わせの数 |
ユーザー |
![]() |
提出日時 | 2021-10-26 21:46:49 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 286 ms / 5,000 ms |
コード長 | 1,364 bytes |
コンパイル時間 | 5,395 ms |
コンパイル使用メモリ | 254,980 KB |
最終ジャッジ日時 | 2025-01-25 07:24:33 |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 1 |
ソースコード
#include <stdio.h>#include <bits/stdc++.h>#include <atcoder/all>using namespace atcoder;using mint = modint1000000007;using namespace std;#define rep(i,n) for (int i = 0; i < (n); ++i)#define Inf 1000000000struct combi{deque<mint> kaijou;deque<mint> kaijou_;combi(int n){kaijou.push_back(1);for(int i=1;i<=n;i++){kaijou.push_back(kaijou[i-1]*i);}mint b=kaijou[n].inv();kaijou_.push_front(b);for(int i=1;i<=n;i++){int k=n+1-i;kaijou_.push_front(kaijou_[0]*k);}}mint combination(int n,int r){if(r>n)return 0;mint a = kaijou[n]*kaijou_[r];a *= kaijou_[n-r];return a;}mint junretsu(int a,int b){mint x = kaijou_[a]*kaijou_[b];x *= kaijou[a+b];return x;}mint catalan(int n){return combination(2*n,n)/(n+1);}};int main(){int _t;cin>>_t;combi C(3000000);rep(_,_t){string s;cin>>s;char c = s[0];rep(i,2)s.erase(s.begin());int N=0,K=0;while(s[0]!=','){N *= 10;N += s[0] - '0';s.erase(s.begin());}s.erase(s.begin());s.pop_back();K = stoi(s);mint ans = 0;if(c=='C'){ans = C.combination(N,K);}if(c=='P'){ans = C.combination(N,K) * C.kaijou[K];}if(c=='H'){if(N>=1){ans = C.combination(N+K-1,N-1);}else if(K==0)ans = 1;}cout<<ans.val()<<endl;}return 0;}