結果

問題 No.389 ロジックパズルの組み合わせ
ユーザー rickythetarickytheta
提出日時 2016-07-08 22:28:35
言語 C++11
(gcc 13.3.0)
結果
AC  
実行時間 39 ms / 2,000 ms
コード長 1,560 bytes
コンパイル時間 1,163 ms
コンパイル使用メモリ 157,420 KB
実行使用メモリ 14,592 KB
最終ジャッジ日時 2024-10-13 04:26:59
合計ジャッジ時間 4,790 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 99
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:70:8: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   70 |   scanf("%d",&m);
      |   ~~~~~^~~~~~~~~

ソースコード

diff # 

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;
typedef vector<int> vi;
typedef vector<ll> vl;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;

typedef int _loop_int;
#define REP(i,n) for(_loop_int i=0;i<(_loop_int)(n);++i)
#define FOR(i,a,b) for(_loop_int i=(_loop_int)(a);i<(_loop_int)(b);++i)
#define FORR(i,a,b) for(_loop_int i=(_loop_int)(b)-1;i>=(_loop_int)(a);--i)

#define DEBUG(x) cout<<#x<<": "<<x<<endl
#define DEBUG_VEC(v) cout<<#v<<":";REP(i,v.size())cout<<" "<<v[i];cout<<endl
#define ALL(a) (a).begin(),(a).end()

#define CHMIN(a,b) a=min((a),(b))
#define CHMAX(a,b) a=max((a),(b))

// mod
const ll MOD = 1000000007ll;
#define FIX(a) ((a)%MOD+MOD)%MOD

// floating
typedef double Real;
const Real EPS = 1e-11;
#define EQ0(x) (abs(x)<EPS)
#define EQ(a,b) (abs(a-b)<EPS)
typedef complex<Real> P;

int m;
int k;
int h[1252525];

ll fact[1252525];
void init(){
  fact[0] = 1;
  FOR(i,1,1252525){
    fact[i] = i*fact[i-1]%MOD;
  }
}

ll inv(ll x){
  ll k = MOD-2;
  ll ret = 1;
  while(k){
    if(k&1)ret=ret*x%MOD;
    x=x*x%MOD;
    k>>=1;
  }
  return ret;
}

int comb(int a,int b){
  ll ret = 1;
  init();
  ret *= fact[a];
  ret %= MOD;
  ret *= inv(fact[b]);
  ret %= MOD;
  ret *= inv(fact[a-b]);
  ret %= MOD;
  return (int)ret;
}

int main(){
  scanf("%d",&m);
  k=0;
  while(~scanf("%d",&h[k++]));
  --k;
  if(h[0]==0){
    puts("1");
    return 0;
  }
  int sum = 0;
  REP(i,k)sum += h[i];
  m -= sum;
  if(m+1<k){
    puts("NA");
    return 0;
  }
  // ans = m+1 C k
  printf("%d\n",comb(m+1,k));
  return 0;
}
0