結果

問題 No.117 組み合わせの数
ユーザー ojisan_ITojisan_IT
提出日時 2017-10-11 22:08:24
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 678 ms / 5,000 ms
コード長 2,365 bytes
コンパイル時間 962 ms
コンパイル使用メモリ 90,048 KB
実行使用メモリ 34,252 KB
最終ジャッジ日時 2023-08-10 23:17:28
合計ジャッジ時間 2,756 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

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

ソースコード

diff # 

#include<iostream>
#include<cstdio>
#include<vector>
#include<map>
#include<tuple>
#include<string>
#include<sstream>
#include<cmath>
#include<climits>
#include<algorithm>
#include<bitset>
#include<set>
#include<stack>
using namespace std;  
typedef long long ll;
typedef vector<int> vi;
typedef vector<vi> vvi;
#define rep(i,n) for(ll i=0;i<(n);i++)  
#define tii tuple<int,int>
#define tiii tuple<int,int,int>
#define mt make_tuple
#define pb push_back  
#define ALL(a) (a).begin(),(a).end()
#define FST first
#define SEC second  
const int INF = (INT_MAX/2);
const ll LLINF = (LLONG_MAX/2);
const double eps = 1e-5;
const double PI = M_PI;  
#define DEB cerr<<"!"<<endl
#define SHOW(a,b) cerr<<(a)<<" "<<(b)<<endl
#define SHOWARRAY(ar,i,j) REP(a,i)REP(b,j)cerr<<ar[a][b]<<((b==j-1)?((a==i-1)?("\n\n"):("\n")):(" "))
#define DIV 1000000007
ll pow(ll x,ll n,ll m){ll r=1;while(n>0){if((n&1)==1)r=r*x%m;x=x*x%m;n>>=1;}return r%m;}

class FermatCombination{
public:
  vector<ll> factrial; // 階乗
  vector<ll> inverse; // 逆元
  FermatCombination(int size){
    factrial.resize(size+1);
    inverse.resize(size+1);
    factrial[0] = 1;
    inverse[0] = 1;
    for(int i = 1; i < size+1; i++){
      factrial[i] = factrial[i-1] * i % DIV;
      inverse[i] = pow(factrial[i],DIV-2,DIV);
    }
  }
  ll combination(int n, int k){
    if(n < k) return 0; 
    return factrial[n]* inverse[k] % DIV * inverse[n - k] % DIV;
  }
  ll permutation(int n, int k){
    if(n < k) return 0; 
    return factrial[n] * inverse[n-k] % DIV;
  }
  ll multi_choose(int n, int k){
    if(n == 0 && k == 0) return 1;
    else return combination(n+k-1,k);
  }
};

//何故かローカルでstoiが動かない
int stoi_(string s){
  stringstream ss;
  ss << s;
  int ret; ss >> ret;
  return ret;
}

int to1(string s){
  string ret ="";
  for(int i = 2; s[i] != ','; i++)
    ret = ret + s[i];
  return stoi_(ret);
}
int to2(string s){
  string ret ="";
  for(int i = s.size()-2; s[i] != ','; i--)
    ret = s[i] + ret;
  return stoi_(ret);
}

int main(){
  int n; cin >> n;
  FermatCombination FC(2000000);
  rep(i,n){
    string s; cin >> s;
    ll ans = 0;
    if(s[0] == 'C'){
      ans = FC.combination(to1(s),to2(s));
    }else if(s[0] == 'P'){
      ans = FC.permutation(to1(s),to2(s));
    }else{
      ans = FC.multi_choose(to1(s),to2(s));
    }
    cout << ans << endl;
  }
}
0