結果

問題 No.389 ロジックパズルの組み合わせ
ユーザー ei1333333ei1333333
提出日時 2016-07-08 23:16:18
言語 C++11
(gcc 13.3.0)
結果
WA  
実行時間 -
コード長 1,141 bytes
コンパイル時間 1,431 ms
コンパイル使用メモリ 161,372 KB
実行使用メモリ 36,352 KB
最終ジャッジ日時 2024-10-13 06:42:50
合計ジャッジ時間 18,745 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 89 WA * 10
権限があれば一括ダウンロードができます
コンパイルメッセージ
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 = 2000000) : 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[1000000];
int main()
{
  scanf("%d", &M);
  while(~scanf("%d", &H[K])) ++K;
  int ALL = accumulate(H, H + K, 0);
  CombinationMod c(MOD);
  if(K == 1 && H[K] == 0) cout << 1 << endl;
  else if(M - ALL + 1 < K) cout << "NA" << endl;
  else cout << c.nCr(M - ALL + 1, K) << endl;
}
0