結果

問題 No.117 組み合わせの数
ユーザー SugarDragon5SugarDragon5
提出日時 2018-08-31 09:07:26
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 238 ms / 5,000 ms
コード長 3,286 bytes
コンパイル時間 1,779 ms
コンパイル使用メモリ 171,736 KB
実行使用メモリ 11,032 KB
最終ジャッジ日時 2024-09-13 20:41:40
合計ジャッジ時間 2,678 ms
ジャッジサーバーID
(参考情報)
judge3 / judge2
このコードへのチャレンジ
(要ログイン)

テストケース

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

ソースコード

diff #

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define FOR(i,n,m) for(int i=(n);i<(m);i++)
#define REP(i,n) FOR(i,0,n)
#define REPR(i,n) for(int i=(n);i>=0;i--)
#define all(vec) vec.begin(),vec.end()
using vi=vector<int>;
using vvi=vector<vi>;
using vl=vector<ll>;
using vvl=vector<vl>;
using P=pair<int,int>;
using PP=pair<int,P>;
using Pl=pair<ll,ll>;
using PPl=pair<ll,Pl>;
using vs=vector<string>;
#define fi first
#define se second
#define pb push_back
template<class T>bool chmax(T &a,const T &b){if(a<b){a=b;return true;}return false;}
template<class T>bool chmin(T &a,const T &b){if(a>b){a=b;return true;}return false;}
const ll MOD=1000000007LL;
const int INF=1<<30;
const ll LINF=1LL<<60;
template<int mod>
struct ModInt{
    int x;
    ModInt():x(0){}
    ModInt(long long y):x(y>=0?y%mod:(mod-(-y)%mod)%mod){}
    ModInt &operator+=(const ModInt &p){
        if((x+=p.x)>=mod)x-=mod;
        return *this;
    }
    ModInt &operator-=(const ModInt &p){
        if((x+=mod-p.x)>=mod)x-=mod;
        return *this;
    }
    ModInt &operator*=(const ModInt &p){
        x=(int)(1LL*x*p.x%mod);
        return *this;
    }
    ModInt &operator/=(const ModInt &p){
        *this*=p.inverse();
        return *this;
    }
    ModInt operator-()const{return ModInt(-x);}
    ModInt operator+(const ModInt &p)const{return ModInt(*this)+=p;}
    ModInt operator-(const ModInt &p)const{return ModInt(*this)-=p;}
    ModInt operator*(const ModInt &p)const{return ModInt(*this)*=p;}
    ModInt operator/(const ModInt &p)const{return ModInt(*this)/=p;}
    bool operator==(const ModInt &p)const{return x==p.x;}
    bool operator!=(const ModInt &p)const{return x!=p.x;}
    ModInt inverse()const{
        int a=x,b=mod,u=1,v=0,t;
        while(b>0){
            t=a/b;
            a-=t*b;
            swap(a,b);
            u-=t*v;
            swap(u,v);
        }
        return ModInt(u);
    }
    friend ostream &operator<<(ostream &os,const ModInt<mod> &p){
        return os<<p.x;
    }
    friend istream &operator>>(istream &is,ModInt<mod> &a){
        long long x;
        is>>x;
        a=ModInt<mod>(x);
        return (is);
    }
};

template<int mod=1e9+7>
struct Combination{
    using mint=ModInt<mod>;
    vector<mint> fact;
    Combination(int n){
        n++;
        fact.resize(n);
        for(int i=0;i<n;i++){
            if(!i){
                fact[i]=1;
            }else{
                fact[i]=fact[i-1]*i;
            }
        }
    }
    mint P(int n,int r){
        if(n<r){
            return 0;
        }
        return fact[n]/fact[n-r];
    }
    mint C(int n,int r){
        if(n<r){
            return 0;
        }
        return fact[n]/fact[r]/fact[n-r];
    }
    mint H(int n,int r){
        if(!n&&!r){
            return 1;
        }
        if(n+r-1<r){
            return 0;
        }
        return C(n+r-1,r);
    }
};

const int mod=1e9+7;
using mint=ModInt<mod>;
int main(){
    int Q;
    cin>>Q;
    Combination<mod> comb(2e6);
    REP(i,Q){
        char c;
        int x,y;
        scanf("\n%c(%d,%d)",&c,&x,&y);
        if(c=='C'){
            cout<<comb.C(x,y)<<endl;
        }else if(c=='P'){
            cout<<comb.P(x,y)<<endl;
        }else{
            cout<<comb.H(x,y)<<endl;
        }
    }
    return 0;
}
0