結果

問題 No.389 ロジックパズルの組み合わせ
ユーザー kosakkunkosakkun
提出日時 2016-11-08 18:10:04
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 140 ms / 2,000 ms
コード長 1,517 bytes
コンパイル時間 900 ms
コンパイル使用メモリ 97,184 KB
実行使用メモリ 6,824 KB
最終ジャッジ日時 2024-11-25 04:56:32
合計ジャッジ時間 4,777 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 99
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <string>
#include <vector>
#include <queue>
#include <stack>
#include <map>
#include <algorithm>
#include <sstream>
#include <cmath>
#include <set>
#include <iomanip>
#include <deque>
#include <list>
#include <stdio.h>
using namespace std;

#define REP(i,n) for(long long (i)=0;(i)<(long long)(n);(i)++)
#define RREP(i,n) for(long long (i)=(long long)(n)-1;i>=0;i--)
#define REMOVE(Itr,n) (Itr).erase(remove((Itr).begin(),(Itr).end(),n),(Itr).end())
#define UNIQUE(Itr) sort((Itr).begin(),(Itr).end()); (Itr).erase(unique((Itr).begin(),(Itr).end()),(Itr).end())

typedef long long ll;

#define MOD 1000000007

ll extgcd(ll a, ll b, ll& x, ll& y) {
    if(b==0){
        x=(a>=0)?1:-1;
        y=0;
        return abs(a);
    }else{
        ll res=extgcd(b,a%b,y,x);
        y-=(a/b)*x;
        return res;
    }
}

ll mod_inverse(ll a, ll m) {
    ll x,y;
    extgcd(a,m,x,y);
    return (m+x%m)%m;
}

ll c(ll n, ll k) {
    ll res=1;
    for(int i=0;i<k;i++)res=(res*((n-i)%MOD))%MOD;
    for(int i=1;i<=k;i++)res=(res*mod_inverse(i,MOD))%MOD;
    return res;
}

int main(){
    
    int N; cin>>N;
    vector<int> num;
    int t;
    while(cin>>t)num.push_back(t);
   if(num[0]==0){
cout<<1<<endl;
return 0;   
}    

    ll sum=0;
    REP(i,num.size())sum+=num[i]+1;
    sum--;
    if(sum==N){
        cout<<1<<endl;
        return 0;
    }else if(sum>N){
        cout<<"NA"<<endl;
        return 0;
    }
    
    ll x=N-sum;
    cout<<c(num.size()+x,x)<<endl;
    
    return 0;
}
0