結果

問題 No.3046 yukicoderの過去問
ユーザー ミドリムシミドリムシ
提出日時 2019-04-01 21:38:03
言語 C++11
(gcc 11.4.0)
結果
TLE  
実行時間 -
コード長 3,762 bytes
コンパイル時間 489 ms
コンパイル使用メモリ 67,784 KB
実行使用メモリ 13,568 KB
最終ジャッジ日時 2024-05-05 04:09:17
合計ジャッジ時間 4,268 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 5 ms
13,568 KB
testcase_01 AC 4 ms
8,064 KB
testcase_02 AC 5 ms
8,192 KB
testcase_03 AC 6 ms
8,192 KB
testcase_04 AC 4 ms
7,936 KB
testcase_05 AC 5 ms
8,064 KB
testcase_06 AC 41 ms
8,448 KB
testcase_07 TLE -
testcase_08 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <algorithm>
#include <vector>
#include <string.h>
#include <queue>
#include <cassert>
using namespace std;
using lint = long long;
const long mod = 1e9 + 7;
#define all(x) (x).begin(), (x).end()
#define bitcount(n) __builtin_popcountl(long(n))
#define fcout cout << fixed << setprecision(15)
#define highest(x) (63 - __builtin_clzl(x))
template<class T> inline void YES(T condition){ if(condition) cout << "YES" << endl; else cout << "NO" << endl; }
template<class T> inline void Yes(T condition){ if(condition) cout << "Yes" << endl; else cout << "No" << endl; }
template<class T> inline void POSS(T condition){ if(condition) cout << "POSSIBLE" << endl; else cout << "IMPOSSIBLE" << endl; }
template<class T> inline void Poss(T condition){ if(condition) cout << "Possible" << endl; else cout << "Impossible" << endl; }
template<class T> inline void First(T condition){ if(condition) cout << "First" << endl; else cout << "Second" << endl; }
template<class T = string, class U = char>int character_count(T text, U character){ int ans = 0; for(U i: text){ ans += (i == character); } return ans; }
long power(long base, long exponent, long module){ if(exponent % 2){ return power(base, exponent - 1, module) * base % module; }else if(exponent){ long root_ans = power(base, exponent / 2, module); return root_ans * root_ans % module; }else{ return 1; }}
struct position{ int y, x; }; position mv[4] = {{0, -1}, {1, 0}, {0, 1}, {-1, 0}}; // double euclidean(position first, position second){ return sqrt((second.x - first.x) * (second.x - first.x) + (second.y - first.y) * (second.y - first.y)); }
template<class T, class U> string to_string(pair<T, U> x){ return to_string(x.first) + "," + to_string(x.second); } string to_string(string x){ return x; }
template<class itr> void array_output(itr start, itr goal){ string ans; for(auto i = start; i != goal; i++) ans += to_string(*i) + " "; if(!ans.empty()) ans.pop_back(); cout << ans << endl; }
template<class itr> void cins(itr start, itr goal){ for(auto i = start; i != goal; i++){ cin >> (*i); } }
template<class T> T gcd(T a, T b){ if(a && b){ return gcd(min(a, b), max(a, b) % min(a, b)); }else{ return a; }} template<class T> T lcm(T a, T b){ return a / gcd(a, b) * b; }
struct combination{ vector<long> fact, inv; combination(int sz) : fact(sz + 1), inv(sz + 1){ fact[0] = 1; for(int i = 1; i <= sz; i++){ fact[i] = fact[i - 1] * i % mod; } inv[sz] = power(fact[sz], mod - 2, mod); for(int i = sz - 1; i >= 0; i--){ inv[i] = inv[i + 1] * (i + 1) % mod; } } long C(int p, int q) const{ if(q < 0 || p < q) return 0; return (fact[p] * inv[q] % mod * inv[p - q] % mod); } };

int main(){
    int K, N;
    cin >> K >> N;
    int x[N];
    cins(x, x + N);
    if(N <= 50000){
        long dp[300000];
        dp[K] = 1;
        for(int i = K + 1; i < 300000; i++){
            dp[i] = 0;
        }
        for(int i = K - 1; i >= 0; i--){
            for(int j: x){
                dp[i] += dp[i + j];
            }
            dp[i] %= mod;
        }
        cout << dp[0] << endl;
    }else{
        vector<int> nex;
        for(int i = 1; i <= 100010; i++){
            if(upper_bound(x, x + N, i) == lower_bound(x, x + N, i)){
                nex.push_back(i);
            }
        }
        long dp[300000], dp_sum[300000];
        dp[K] = 1;
        dp_sum[K] = 1;
        for(int i = K + 1; i < 300000; i++){
            dp[i] = 0;
            dp_sum[i] = 0;
        }
        for(int i = K - 1; i >= 0; i--){
            for(int j: nex){
                dp[i] += dp[i + j];
            }
            dp[i] %= mod;
            dp[i] = (dp_sum[i + 1] + mod - dp[i]) % mod;
            dp_sum[i] = (dp[i] + dp_sum[i + 1]) % mod;
        }
        cout << dp[0] << endl;
    }
}
0