Submission #69714451

---

Source Code Expand

#include <bits/stdc++.h>
using namespace std;

/////////////////// メイン ///////////////////

int main () {
  
  /////////////////// 前入力 ///////////////////
  
  int q, mod;
  cin >> q >> mod;

  /////////////////// 前処理 ///////////////////

  // 二項係数の範囲
  int m=5001;

  // パスカルの三角形
  vector<vector<long long>> comb(m,vector<long long>(m,0));
  for (int i=0; i<m; i++) {
    comb.at(i).at(0) = 1;
    for (int j=1; j<=i; j++) {
      comb.at(i).at(j) = (comb.at(i-1).at(j-1)+comb.at(i-1).at(j))%mod;
    } 
  }

  /////////////////// ループ ///////////////////

  for (int loop=0; loop<q; loop++) {
    
    //////////////////// 入力 ////////////////////

    int n;
    cin >> n;

    vector<int> a(n);
    for (int i=0; i<n; i++) {
      cin >> a.at(i);
    }

    //////////////// 出力変数定義 ////////////////

    long long result = 1;

    //////////////////// 処理 ////////////////////

    // とりあえず全部の和
    int sum = accumulate(a.begin(),a.end(),0);

    // それぞれの数を残っているマスの中で選んでいく
    for (int i : a) {
      result *= comb.at(sum).at(i);
      result %= mod;
      sum -= i;
    }

    //////////////////// 出力 ////////////////////

    cout << result << endl;
    
  }

  /////////////////// 後処理 ///////////////////



  //////////////////// 終了 ////////////////////

  return 0;

}

Submission Info

Judge Result