結果

問題 No.117 組み合わせの数
ユーザー koprickykopricky
提出日時 2017-07-17 16:48:59
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 280 ms / 5,000 ms
コード長 1,993 bytes
コンパイル時間 1,541 ms
コンパイル使用メモリ 160,492 KB
実行使用メモリ 50,304 KB
最終ジャッジ日時 2024-04-16 23:51:19
合計ジャッジ時間 2,536 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 280 ms
50,304 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:53:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   53 |         scanf("%s",c);
      |         ~~~~~^~~~~~~~

ソースコード

diff # 

#include <bits/stdc++.h>
#define ll long long
#define INF 1000000005
#define MOD 1000000007
#define EPS 1e-10
#define rep(i,n) for(int i=0;i<(int)n;++i)
#define each(a, b) for(auto (a): (b))
#define all(v) (v).begin(),(v).end()
#define fi first
#define se second
#define pb push_back
#define show(x) cout <<#x<<" = "<<(x)<<endl
#define spair(p) cout <<#p<<": "<<p.fi<<" "<<p.se<<endl
#define svec(v) cout<<#v<<":";rep(kbrni,v.size())cout<<" "<<v[kbrni];cout<<endl
#define sset(s) cout<<#s<<":";each(kbrni,s)cout <<" "<<kbrni;cout<<endl

using namespace std;

typedef pair<int,int>P;

const int MAX_N = 2000001;

//どこかでmake()することを忘れずに
ll inv[MAX_N],fac[MAX_N],finv[MAX_N];

void make()
{
	fac[0] = fac[1] = 1;
	finv[0] = finv[1] = 1;
	inv[1] = 1;
	for(int i=2;i<MAX_N;i++){
		inv[i] = MOD - inv[MOD%i] * (MOD/i) % MOD;
		fac[i] = fac[i-1] * (ll) i % MOD;
		finv[i] = finv[i-1] * inv[i] % MOD;
	}
}

ll comb(int a,int b)
{
	if(a<b){
		return 0;
	}
	return fac[a] * (finv[b] * finv[a-b] % MOD) % MOD;
}

int main()
{
    int n;
    cin >> n;
    make();
    rep(i,n){
        char c[20];
        scanf("%s",c);
        int st=2;
        string s1,s2;
        while(1){
            if(!isdigit(c[st])){
                break;
            }else{
                s1.pb(c[st]);
            }
            st++;
        }
        st++;
        while(1){
            if(!isdigit(c[st])){
                break;
            }else{
                s2.pb(c[st]);
            }
            st++;
        }
        int u = stoi(s1),v = stoi(s2);
        if(c[0] == 'C'){
            cout << comb(u,v) << endl;
        }else if(c[0] == 'P'){
            if(u - v < 0){
                cout << "0\n";
            }else{
                cout << fac[u] * finv[u-v] % MOD << endl;
            }
        }else{
            if(u == 0 && v == 0){
                cout << "1\n";
            }else{
                cout << comb(u+v-1,v) << endl;
            }
        }
    }
    return 0;
}
0