結果

問題 No.117 組み合わせの数
ユーザー karinohitokarinohito
提出日時 2021-07-26 11:26:55
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 1,534 bytes
コンパイル時間 1,759 ms
コンパイル使用メモリ 176,772 KB
実行使用メモリ 50,064 KB
最終ジャッジ日時 2024-07-22 00:11:17
合計ジャッジ時間 3,186 ms
ジャッジサーバーID
(参考情報)
judge4 / judge2
このコードへのチャレンジ
(要ログイン)

テストケース

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

ソースコード

diff #

//#include <atcoder/all>
#include <bits/stdc++.h>
using namespace std;
//using namespace atcoder;
using ll = long long;
#define all(A) A.begin(),A.end()
using vll = vector<ll>;
#define rep(i, n) for (long long i = 0; i < (long long)(n); i++)
using Graph = vector<vector<ll>>;
Graph G;
vll dist;
vector<bool> seen;
set<pair<ll, ll>> S;

ll gcd(ll(a), ll(b)) {
	ll c = a;
	while (a % b != 0) {
		c = a % b;
		a = b;
		b = c;
	}
	return b;
}
vector<ll> fact, factinv, inv;
ll mod = 1e9 + 7;
void prenCkModp(ll n) {
	fact.resize(n + 5);
	factinv.resize(n + 5);
	inv.resize(n + 5);
	fact.at(0) = fact.at(1) = 1;
	factinv.at(0) = factinv.at(1) = 1;
	inv.at(1) = 1;
	for (ll i = 2; i < n + 5; i++) {
		fact.at(i) = (fact.at(i - 1) * i) % mod;
		inv.at(i) = mod - (inv.at(mod % i) * (mod / i)) % mod;
		factinv.at(i) = (factinv.at(i - 1) * inv.at(i)) % mod;
	}

}
ll nCk(ll n, ll k) {
	if (n < k) return 0;
	return fact.at(n) * (factinv.at(k) * factinv.at(n - k) % mod) % mod;
}
ll nPk(ll n,ll k){
	if (n < k) return 0;
	return fact.at(n) * (factinv.at(n - k) % mod) % mod;
}

int main() {
	ll N;
	cin>>N;
	prenCkModp(1e6*2);
	rep(i,N){
		string S;
		cin>>S;
		ll A=0,B=0;
		rep(k,S.size()){
			if(S[k]=='('){
				k++;
				while(S[k]!=','){
					A+=S[k]-'0';
					A*=10;
					k++;
				}
				k++;
				while(S[k]!=')'){
					B+=S[k]-'0';
					B*=10;
					k++;
				}
				
			}
			
		}
		A/=10;B/=10;
		if(S[0]=='C'){
			cout<<nCk(A,B)<<endl;
		}
		else if(S[0]=='P'){
			cout<<nPk(A,B)<<endl;
		}
		else{
			cout<<nCk(A+B-1,B)<<endl;
		}
	}


}
0