#774370 (C++17) No.389 ロジックパズルの組み合わせ

提出ソース

結果

問題	No.389 ロジックパズルの組み合わせ
ユーザー	recososo
提出日時	2022-06-29 00:21:58
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	45 ms / 2,000 ms
コード長	3,975 bytes
コンパイル時間	4,412 ms
コンパイル使用メモリ	264,732 KB
実行使用メモリ	15,716 KB
最終ジャッジ日時	2024-05-01 14:45:38
合計ジャッジ時間	8,934 ms
ジャッジサーバーID （参考情報）	judge3 / judge2

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 5,248 KB
testcase_01	AC	2 ms 5,248 KB
testcase_02	AC	2 ms 5,376 KB
testcase_03	AC	2 ms 5,376 KB
testcase_04	AC	14 ms 10,880 KB
testcase_05	AC	2 ms 5,376 KB
testcase_06	AC	2 ms 5,376 KB
testcase_07	AC	2 ms 5,376 KB
testcase_08	AC	2 ms 5,376 KB
testcase_09	AC	19 ms 12,748 KB
testcase_10	AC	6 ms 5,376 KB
testcase_11	AC	20 ms 12,928 KB
testcase_12	AC	5 ms 5,376 KB
testcase_13	AC	28 ms 15,332 KB
testcase_14	AC	8 ms 5,376 KB
testcase_15	AC	6 ms 5,376 KB
testcase_16	AC	10 ms 7,552 KB
testcase_17	AC	15 ms 9,984 KB
testcase_18	AC	32 ms 11,872 KB
testcase_19	AC	2 ms 5,376 KB
testcase_20	AC	2 ms 5,376 KB
testcase_21	AC	2 ms 5,376 KB
testcase_22	AC	2 ms 5,376 KB
testcase_23	AC	2 ms 5,376 KB
testcase_24	AC	2 ms 5,376 KB
testcase_25	AC	2 ms 5,376 KB
testcase_26	AC	2 ms 5,376 KB
testcase_27	AC	2 ms 5,376 KB
testcase_28	AC	2 ms 5,376 KB
testcase_29	AC	1 ms 5,376 KB
testcase_30	AC	2 ms 5,376 KB
testcase_31	AC	2 ms 5,376 KB
testcase_32	AC	2 ms 5,376 KB
testcase_33	AC	2 ms 5,376 KB
testcase_34	AC	2 ms 5,376 KB
testcase_35	AC	2 ms 5,376 KB
testcase_36	AC	2 ms 5,376 KB
testcase_37	AC	2 ms 5,376 KB
testcase_38	AC	2 ms 5,376 KB
testcase_39	AC	2 ms 5,376 KB
testcase_40	AC	2 ms 5,376 KB
testcase_41	AC	2 ms 5,376 KB
testcase_42	AC	2 ms 5,376 KB
testcase_43	AC	2 ms 5,376 KB
testcase_44	AC	2 ms 5,376 KB
testcase_45	AC	2 ms 5,376 KB
testcase_46	AC	2 ms 5,376 KB
testcase_47	AC	2 ms 5,376 KB
testcase_48	AC	2 ms 5,376 KB
testcase_49	AC	35 ms 12,424 KB
testcase_50	AC	7 ms 5,632 KB
testcase_51	AC	4 ms 5,376 KB
testcase_52	AC	18 ms 12,564 KB
testcase_53	AC	8 ms 5,760 KB
testcase_54	AC	22 ms 14,516 KB
testcase_55	AC	5 ms 5,376 KB
testcase_56	AC	4 ms 5,376 KB
testcase_57	AC	20 ms 9,312 KB
testcase_58	AC	34 ms 14,356 KB
testcase_59	AC	27 ms 12,132 KB
testcase_60	AC	12 ms 6,500 KB
testcase_61	AC	45 ms 15,716 KB
testcase_62	AC	17 ms 9,700 KB
testcase_63	AC	32 ms 12,388 KB
testcase_64	AC	20 ms 13,296 KB
testcase_65	AC	8 ms 5,376 KB
testcase_66	AC	10 ms 5,856 KB
testcase_67	AC	17 ms 9,572 KB
testcase_68	AC	21 ms 14,236 KB
testcase_69	AC	14 ms 11,264 KB
testcase_70	AC	12 ms 8,960 KB
testcase_71	AC	8 ms 6,784 KB
testcase_72	AC	16 ms 12,928 KB
testcase_73	AC	8 ms 7,168 KB
testcase_74	AC	8 ms 6,912 KB
testcase_75	AC	8 ms 6,912 KB
testcase_76	AC	11 ms 8,832 KB
testcase_77	AC	11 ms 8,960 KB
testcase_78	AC	5 ms 5,376 KB
testcase_79	AC	7 ms 6,272 KB
testcase_80	AC	5 ms 5,376 KB
testcase_81	AC	3 ms 5,376 KB
testcase_82	AC	16 ms 12,416 KB
testcase_83	AC	9 ms 7,552 KB
testcase_84	AC	5 ms 5,376 KB
testcase_85	AC	17 ms 13,568 KB
testcase_86	AC	17 ms 12,488 KB
testcase_87	AC	4 ms 5,376 KB
testcase_88	AC	2 ms 5,376 KB
testcase_89	AC	17 ms 8,804 KB
testcase_90	AC	29 ms 15,180 KB
testcase_91	AC	35 ms 12,808 KB
testcase_92	AC	8 ms 5,760 KB
testcase_93	AC	26 ms 11,748 KB
testcase_94	AC	13 ms 10,752 KB
testcase_95	AC	24 ms 9,444 KB
testcase_96	AC	28 ms 14,880 KB
testcase_97	AC	25 ms 13,408 KB
testcase_98	AC	21 ms 13,192 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include<bits/stdc++.h>
#include<atcoder/all>
using namespace std;
using namespace atcoder;
typedef long long ll;
typedef long double ld;
#define REP(i, n) for (int i = 0; i < (n); ++i)
#define REPR(i, n) for (int i = n - 1; i >= 0; --i)
#define FOR(i, m, n) for (int i = m; i < n; ++i)
#define FORR(i, m, n) for (int i = m; i >= n; --i)
#define ALL(v) (v).begin(),(v).end()
#define ALLR(v) (v).rbegin(),(v).rend()
#define fi first
#define se second
#define PB push_back
#define EB emplace_back
template<typename T> using PQ = priority_queue<T>;
template<typename T> using QP = priority_queue<T,vector<T>,greater<T>>;
template<typename T>void debug(const T &v,ll h,ll w){for(ll i=0;i<h;i++){for(ll j=0;j<w;j++)cerr<<v[i][j]<<" \n"[j==w-1];}};
template<typename T>void debug(const T &v,ll n){for(ll i=0;i<n;i++)cerr<<v[i]<<" \n"[i==n-1];};
template<typename T>void debug(const vector<T>&v){debug(v,v.size());}
template<typename T>void debug(const vector<vector<T>>&v){for(auto &vv:v)debug(vv,vv.size());}
template<typename T>void debug(stack<T> st){while(!st.empty()){cerr<<st.top()<<" ";st.pop();}cerr<<endl;}
template<typename T>void debug(queue<T> st){while(!st.empty()){cerr<<st.front()<<" ";st.pop();}cerr<<endl;}
template<typename T>void debug(deque<T> st){while(!st.empty()){cerr<<st.front()<<" ";st.pop_front();}cerr<<endl;}
template<typename T>void debug(PQ<T> st){while(!st.empty()){cerr<<st.top()<<" ";st.pop();}cerr<<endl;}
template<typename T>void debug(QP<T> st){while(!st.empty()){cerr<<st.top()<<" ";st.pop();}cerr<<endl;}
template<typename T>void debug(const set<T>&v){for(auto z:v)cerr<<z<<" ";cerr<<endl;}
template<typename T>void debug(const multiset<T>&v){for(auto z:v)cerr<<z<<" ";cerr<<endl;}
template<typename T,size_t size>void debug(const array<T, size> &a){for(auto z:a)cerr<<z<<" ";cerr<<endl;}
template<typename T,typename V>void debug(const map<T,V>&v){for(auto z:v)cerr<<"["<<z.first<<"]="<<z.second<<",";cerr<<endl;}
void foo() {cerr << endl;}
template <class Head, class... Tail>void foo(Head&& head, Tail&&... tail){cerr << head << " ";foo(move(tail)...);}
template<class T>bool chmax(T &a, const T &b) { if (a<b) { a=b; return 1; } return 0; }
template<class T>bool chmin(T &a, const T &b) { if (b<a) { a=b; return 1; } return 0; }
int popcount(ll x){return __builtin_popcountll(x);};
int poplow(ll x){return __builtin_ctzll(x);};
int pophigh(ll x){return 63 - __builtin_clzll(x);};
const ll INF=1LL<<60;
const int inf=(1<<30)-1;
// const int mod=1e9+7;
const int mod=998244353;
int dx[8]={1,0,-1,0,-1,-1,1,1};
int dy[8]={0,1,0,-1,-1,1,-1,1};

// using mint=modint998244353;
using mint=modint1000000007;

template< typename T >
struct Combination {
  vector< T > _fact, _rfact, _inv;

  Combination(int sz) : _fact(sz + 1), _rfact(sz + 1), _inv(sz + 1) {
    _fact[0] = _rfact[sz] = _inv[0] = 1;
    for(int i = 1; i <= sz; i++) _fact[i] = _fact[i - 1] * i;
    _rfact[sz] /= _fact[sz];
    for(int i = sz - 1; i >= 0; i--) _rfact[i] = _rfact[i + 1] * (i + 1);
    for(int i = 1; i <= sz; i++) _inv[i] = _rfact[i] * _fact[i - 1];
  }

  inline T fact(int k) const { return _fact[k]; }

  inline T rfact(int k) const { return _rfact[k]; }

  inline T inv(int k) const { return _inv[k]; }

  T P(int n, int r) const {
    if(r < 0 || n < r) return 0;
    return fact(n) * rfact(n - r);
  }

  T C(int p, int q) const {
    if(q < 0 || p < q) return 0;
    return fact(p) * rfact(q) * rfact(p - q);
  }

  T H(int n, int r) const {
    if(n < 0 || r < 0) return (0);
    return r == 0 ? 1 : C(n + r - 1, r);
  }
};

int main(){
    cin.tie(0);
    ios::sync_with_stdio(false);
    int m;cin >> m;
    vector<int> h;
    int a;
    while(cin >> a) h.PB(a);
    int n=h.size();
    if(h[0]==0){
        cout << 1 << endl;
        return 0;
    }
    Combination<mint> com(m+10);
    int d=n-1;
    REP(i,n) d+=h[i];
    if(d>m){
        cout << "NA" << endl;
        return 0;
    }
    mint ans=com.C(m-d+n,m-d);
    cout << ans.val() << endl;
}

yukicoder

結果

テストケース

ソースコード