結果

問題 No.117 組み合わせの数
ユーザー Taiki0715Taiki0715
提出日時 2023-11-25 19:38:13
言語 C++17(clang)
(17.0.6 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 3,894 bytes
コンパイル時間 4,585 ms
コンパイル使用メモリ 182,016 KB
実行使用メモリ 28,200 KB
最終ジャッジ日時 2024-09-26 11:10:03
合計ジャッジ時間 5,629 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#if __has_include(<atcoder/all>)
#include <atcoder/all>
using namespace atcoder;
template<int mod>istream &operator>>(istream &is,static_modint<mod> &a){long long b;is>>b;a=b;return is;}
istream &operator>>(istream &is,modint &a){long long b;cin>>b;a=b;return is;}
#endif
#ifdef LOCAL
#include "debug.h"
#else
#define debug(...) static_cast<void>(0)
#define debugg(...) static_cast<void>(0)
template<typename T1,typename T2>ostream &operator<<(ostream &os,const pair<T1,T2>&p){os<<p.first<<' '<<p.second;return os;}
#endif
using ll=long long;
using ull=unsigned long long;
using P=pair<ll,ll>;
template<typename T>using minque=priority_queue<T,vector<T>,greater<T>>;
template<typename T>bool chmax(T &a,const T &b){return (a<b?(a=b,true):false);}
template<typename T>bool chmin(T &a,const T &b){return (a>b?(a=b,true):false);}
template<typename T1,typename T2>istream &operator>>(istream &is,pair<T1,T2>&p){is>>p.first>>p.second;return is;}
template<typename T>istream &operator>>(istream &is,vector<T> &a){for(auto &i:a)is>>i;return is;}
template<typename T1,typename T2>void operator++(pair<T1,T2>&a,int n){a.first++,a.second++;}
template<typename T1,typename T2>void operator--(pair<T1,T2>&a,int n){a.first--,a.second--;}
template<typename T>void operator++(vector<T>&a,int n){for(auto &i:a)i++;}
template<typename T>void operator--(vector<T>&a,int n){for(auto &i:a)i--;}
#define reps(i,a,n) for(int i=(a);i<(n);i++)
#define rep(i,n) reps(i,0,n)
#define all(x) x.begin(),x.end()
#define pcnt(x) __builtin_popcount(x)
ll myceil(ll a,ll b){return (a+b-1)/b;}
template<typename T,size_t n,size_t id=0>
auto vec(const int (&d)[n],const T &init=T()){
  if constexpr (id<n)return vector(d[id],vec<T,n,id+1>(d,init));
  else return init;
}
void SOLVE();
int main(){
  ios::sync_with_stdio(false);
  cin.tie(nullptr);
  #ifdef LOCAL
  clock_t start=clock();
  #endif
  int testcase=1;
  //cin>>testcase;
  for(int i=0;i<testcase;i++){
    SOLVE();
  }
  #ifdef LOCAL
  cerr<<"time:";
  cerr<<(clock()-start)/1000;
  cerr<<"ms\n";
  #endif
}
template<typename T>
struct factorial_table{
private:
  int mod,capacity;
  vector<T>fact,factinv,inv;
  void resize(int n){
    fact.resize(n+1),factinv.resize(n+1),inv.resize(n+1);
    for(int i=capacity+1;i<=n;i++){
      fact[i]=fact[i-1]*i;
      inv[i]=-inv[mod%i]*(mod/i);
      factinv[i]=factinv[i-1]*inv[i];
    }
    capacity=n;
  }
public:
  factorial_table():capacity(1),mod(T::mod()),fact{1,1},factinv{1,1},inv{0,1}{}
  T C(int n,int k){
    if(n<k)return 0;
    if(k<0)return 0;
    if(capacity<n)resize(n);
    return fact[n]*factinv[k]*factinv[n-k];
  }
  T P(int n,int k){
    if(n<k)return 0;
    if(k<0)return 0;
    if(capacity<n)resize(n);
    return fact[n]*factinv[n-k];
  }
  T H(int n,int k){
    return C(n+k-1,k);
  }
  T factorial(int n){
    if(capacity<n)resize(n);
    return fact[n];
  }
  T S(ll n,int k){
    if(n<0)return 0;
    if(n<k)return 0;
    T ret=0;
    if(capacity<k)resize(k);
    rep(i,k+1){
      ret+=fact[k]*factinv[i]*factinv[k-i]*T::raw(i).pow(n)*((k-i)&1?-1:1);
    }
    return ret*factinv[k];
  }
  template<typename... INT>
  T O(INT...k){
    int n=0;
    for(int i:initializer_list<int>{k...}){
      if(i<0)return 0;
      n+=i;
    }
    if(capacity<n)resize(n);
    T ret=fact[n];
    for(int i:initializer_list<int>{k...})ret*=factinv[i];
    return ret;
  }
};
using mint=modint1000000007;
void SOLVE(){
  int n;
  cin>>n;
  factorial_table<mint>table;
  rep(i,n){
    string s;
    cin>>s;
    char c=s[0];
    s.erase(s.begin());
    s.erase(s.begin());
    s.pop_back();
    int g=-1;
    rep(j,s.size())if(s[j]==','){
      g=j;
      break;
    }
    int n=stoi(s.substr(0,g)),k=stoi(s.substr(g+1));
    if(c=='C')cout<<table.C(n,k).val()<<endl;
    else if(c=='P')cout<<table.P(n,k).val()<<endl;
    else cout<<table.H(n,k).val()<<endl;
  }
}
0