結果

問題 No.117 組み合わせの数
ユーザー koba-e964koba-e964
提出日時 2015-06-10 19:46:50
言語 C++11
(gcc 11.4.0)
結果
WA  
実行時間 -
コード長 1,522 bytes
コンパイル時間 818 ms
コンパイル使用メモリ 76,848 KB
実行使用メモリ 26,956 KB
最終ジャッジ日時 2024-07-06 15:24:14
合計ジャッジ時間 1,606 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

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

ソースコード

diff # 

#include <deque>
#include <functional>
#include <iomanip>
#include <iostream>
#include <list>
#include <map>
#include <numeric>
#include <queue>
#include <set>
#include <sstream>
#include <stack>
#include <string>
#include <utility>
#include <vector>

#define REP(i,s,n) for(int i=(int)(s);i<(int)(n);i++)

using namespace std;
typedef long long int ll;
typedef vector<int> VI;
typedef pair<int, int> PI;
const double EPS=1e-9;
const ll mod = 1e9+7;
const int N = 3e6;
ll fact[N];

ll invmod(ll v, ll m) {
  ll sum = 1, cur = v % m;
  ll e = m - 2;
  while (e > 0) {
    if (e % 2) {
      sum = sum * cur % m;
    }
    cur = cur * cur % m;
    e /= 2;
  }
  return sum;
}


ll comb(ll n, ll k) {
  if (n < k || k < 0 || n < 0) {
    return 0;
  }
  ll sum = fact[n];
  sum = sum * invmod(fact[k], mod) % mod;
  sum = sum * invmod(fact[n - k], mod) % mod;
  return sum;
}

void solve(const string &s) {
  stringstream ss;
  ss << s;
  char c = ss.get();
  ss.ignore();
  ll n,k;
  ss >> n;
  ss.ignore();
  ss >> k;
  switch(c) {
  case 'C':
    cout << comb(n, k) << endl;
    break;
  case 'H':
    cout << comb(n + k - 1, k) << endl;
    break;
  case 'P':
    {
      ll sum = fact[n];
      if (n >= k) {
	sum = sum * invmod(fact[n - k], mod) % mod;
      } else {
	sum = 0;
      }
      cout << sum << endl;
    }
    break;
  }
}

int main(void){
  int t;
  cin >> t;
  fact[0] = 1;
  REP(i, 1, N) {
    fact[i] = fact[i - 1] * i % mod;
  }
  REP(loop_var, 0, t) {
    string s;
    cin >> s;
    solve(s);
  }
}
0