結果
問題 | No.117 組み合わせの数 |
ユーザー | TKTYI |
提出日時 | 2022-05-18 23:53:20 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 615 ms / 5,000 ms |
コード長 | 1,754 bytes |
コンパイル時間 | 3,875 ms |
コンパイル使用メモリ | 231,368 KB |
実行使用メモリ | 26,740 KB |
最終ジャッジ日時 | 2024-09-17 03:59:26 |
合計ジャッジ時間 | 5,545 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ソースコード
#include<bits/stdc++.h> #include<atcoder/all> using namespace std; using namespace atcoder; typedef long long int ll; typedef long double ld; #define FOR(i,l,r) for(ll i=l;i<r;i++) #define REP(i,n) FOR(i,0,n) #define RFOR(i,l,r) for(ll i=r-1;i>=l;i--) #define RREP(i,n) RFOR(i,0,n) #define ALL(x) x.begin(),x.end() #define PA pair<ll,ll> #define F first #define S second #define BS(A,x) binary_search(ALL(A),x) #define LB(A,x) (ll)(lower_bound(ALL(A),x)-A.begin()) #define UB(A,x) (ll)(upper_bound(ALL(A),x)-A.begin()) #define COU(A,x) (UB(A,x)-LB(A,x)) #define sz(c) ((ll)(c).size()) //#include<boost/multiprecision/cpp_int.hpp> //namespace mp=boost::multiprecision; //using Bint=mp::cpp_int; template<typename T>using min_priority_queue=priority_queue<T,vector<T>,greater<T>>; template<class T>bool chmax(T&a,T b){if(a<b){a=b;return 1;}return 0;} template<class T>bool chmin(T&a,T b){if(b<a){a=b;return 1;}return 0;} template<class T>void print(vector<T>A){REP(i,A.size()){if(i)cout<<" ";cout<<A[i];}cout<<endl;} ld dist(ld x1,ld y1,ld x2,ld y2){return sqrtl((x1-x2)*(x1-x2)+(y1-y2)*(y1-y2));} using mint=modint1000000007; //using mint=modint998244353; //const ll mod=1000000007; //const ll mod=998244353; //using mint=modint; int main(){ ll T;cin>>T; vector<mint>ex(3000000,1); FOR(i,1,3000000)ex[i]=i*ex[i-1]; vector<mint>r(3000000); REP(i,3000000)r[i]=1/ex[i]; while(T--){ string S;cin>>S; ll t=0; while(S[t]!=',')t++; ll N=stoi(S.substr(2,t-2)),K=stoi(S.substr(t+1,sz(S)-t-2)); mint ans=0; if(S[0]=='C')if(N>=K)ans=ex[N]*r[N-K]*r[K]; if(S[0]=='P')if(N>=K)ans=ex[N]*r[N-K]; if(S[0]=='H'){ if(N)ans=ex[N+K-1]*r[K]*r[N-1]; else if(!K)ans=1; } cout<<ans.val()<<endl; } return 0; }