結果

問題 No.389 ロジックパズルの組み合わせ
コンテスト
ユーザー ei1333333
提出日時 2016-07-08 23:21:04
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 1,082 ms / 2,000 ms
コード長 1,153 bytes
コンパイル時間 2,495 ms
コンパイル使用メモリ 161,124 KB
実行使用メモリ 161,280 KB
最終ジャッジ日時 2024-10-13 07:04:06
合計ジャッジ時間 91,717 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 99
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:48:8: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   48 |   scanf("%d", &M);
      |   ~~~~~^~~~~~~~~~

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
const int MOD = 1e9 + 7;
struct CombinationMod
{
  long long mod, size;
  vector< long long > fact;
  vector< long long > inv;
 
  void make_fact()
  {
    fact[0] = 1;
    for(int  i = 1; i < size; i++){
      fact[i] = fact[i-1] * i % mod;
    }
  }
 
  void make_inv()
  {
    inv[0] = inv[1] = 1;
    for(int i = 2; i < size; i++){
      inv[i] = inv[mod % i] * (mod - mod / i) % mod;
    }
    for(int i = 2; i < size; i++){
      inv[i] = inv[i-1] * inv[i] % mod;
    }
  }
  
  CombinationMod(int mod, int size = 10000000) : mod(mod), size(size+1)
  {
    fact.resize(size);
    inv.resize(size);
    make_fact();
    make_inv();
  }
  long long nCr(long long n, long long k)
  {
    if(k==0 || n==k) return 1;
    long long ret = fact[n] * inv[k] % mod * inv[n-k] % mod;
    return ret;
  }
};

int M, K;
int H[10000000];
int main()
{
  scanf("%d", &M);
  while(~scanf("%d", &H[K])) ++K;
  long long ALL = accumulate(H, H + K, 0LL);
  CombinationMod c(MOD);
  if(K == 1 && H[0] == 0) cout << 1 << endl;
  else if(M - ALL + 1 < K) cout << "NA" << endl;
  else cout << c.nCr(M - ALL + 1, K) << endl;
}
0