#571652 (C++17) No.1269 I hate Fibonacci Number

• 問題: No.1269 I hate Fibonacci Number
• ユーザー: tokusakurai
• 提出日時: 2020-10-26 09:56:23
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 343 ms / 3,000 ms
• コード長: 4,626 bytes
• コンパイル時間: 2,474 ms
• コンパイル使用メモリ: 218,160 KB
• 実行使用メモリ: 17,664 KB
• 最終ジャッジ日時: 2024-07-21 21:18:01
• 合計ジャッジ時間: 4,764 ms
• ジャッジサーバーID （参考情報）: judge3 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	3 ms; 5,376 KB
testcase_03	AC	2 ms; 5,376 KB
testcase_04	AC	1 ms; 5,376 KB
testcase_05	AC	2 ms; 5,376 KB
testcase_06	AC	1 ms; 5,376 KB
testcase_07	AC	2 ms; 5,376 KB
testcase_08	AC	2 ms; 5,376 KB
testcase_09	AC	2 ms; 5,376 KB
testcase_10	AC	2 ms; 5,376 KB
testcase_11	AC	2 ms; 5,376 KB
testcase_12	AC	2 ms; 5,376 KB
testcase_13	AC	44 ms; 7,424 KB
testcase_14	AC	59 ms; 8,500 KB
testcase_15	AC	7 ms; 5,376 KB
testcase_16	AC	48 ms; 7,296 KB
testcase_17	AC	3 ms; 5,376 KB
testcase_18	AC	51 ms; 7,636 KB
testcase_19	AC	32 ms; 6,528 KB
testcase_20	AC	9 ms; 5,376 KB
testcase_21	AC	29 ms; 6,016 KB
testcase_22	AC	25 ms; 5,632 KB
testcase_23	AC	29 ms; 6,016 KB
testcase_24	AC	15 ms; 5,376 KB
testcase_25	AC	17 ms; 5,376 KB
testcase_26	AC	3 ms; 5,376 KB
testcase_27	AC	2 ms; 5,376 KB
testcase_28	AC	9 ms; 5,376 KB
testcase_29	AC	16 ms; 5,376 KB
testcase_30	AC	7 ms; 5,376 KB
testcase_31	AC	111 ms; 9,728 KB
testcase_32	AC	18 ms; 5,376 KB
testcase_33	AC	343 ms; 17,664 KB
testcase_34	AC	2 ms; 5,376 KB
testcase_35	AC	6 ms; 5,376 KB
testcase_36	AC	2 ms; 5,376 KB
testcase_37	AC	1 ms; 5,376 KB
testcase_38	AC	2 ms; 5,376 KB

ソースコード

#include <bits/stdc++.h>
using namespace std;
#define rep(i, n) for(int i = 0; i < n; i++)
#define rep2(i, x, n) for(int i = x; i <= n; i++)
#define rep3(i, x, n) for(int i = x; i >= n; i--)
#define elif else if
#define sp(x) fixed << setprecision(x)
#define pb push_back
#define eb emplace_back
#define all(x) x.begin(), x.end()
#define sz(x) (int)x.size()
using ll = long long;
using pii = pair<int, int>;
using pil = pair<int, ll>;
using pli = pair<ll, int>;
using pll = pair<ll, ll>;
const int MOD = 1000000007;
//const int MOD = 998244353;
const int inf = (1<<30)-1;
const ll INF = (1LL<<60)-1;
const double pi = acos(-1.0);
const double EPS = 1e-10;
template<typename T> bool chmax(T &x, const T &y) {return (x < y)? (x = y, true) : false;};
template<typename T> bool chmin(T &x, const T &y) {return (x > y)? (x = y, true) : false;};

template<int char_size, char base>
struct Trie{
    struct Node{
        vector<int> next, accept;
        int count; //子以下に追加された文字列の数

        Node() : count(0){
            next.assign(char_size, -1);
        }
    };

    vector<Node> nodes;

    Trie() {nodes.eb();}

    int count() const {return nodes.front().count;}

    int size() const {return sz(nodes);}

    void insert(const string &s, int id){
        int now = 0;
        rep(i, sz(s)){
            int &next = nodes[now].next[s[i]-base];
            if(next == -1){
                next = size(), nodes.eb();
            }
            nodes[now].count++, now = next;
        }
        nodes[now].count++, nodes[now].accept.pb(id);
    }

    void insert(const string &s) {insert(s, count());}

    bool search(const string &s, bool prefix = false) const{
        int now = 0;
        rep(i, sz(s)){
            now = nodes[now].next[s[i]-base];
            if(now == -1) return false;
        }
        return (prefix)? true : !nodes[now].accept.empty();
    }
};

template<int char_size, char base>
struct Aho_Corasick : Trie<char_size+1, base>{
    using trie = Trie<char_size+1, base>;
    const int failure = char_size;
    vector<int> correct;

    void build(bool heavy = false){
        correct.resize(this->size());
        rep(i, this->size()){
            correct[i] = sz(this->nodes[i].accept);
        }
        queue<int> que;
        rep(i, char_size+1){
            if(this->nodes[0].next[i] != -1){
                this->nodes[this->nodes[0].next[i]].next[failure] = 0;
                que.push(this->nodes[0].next[i]);
            }
            else{
                this->nodes[0].next[i] = 0;
            }
        }
        while(!que.empty()){
            auto &now = this->nodes[que.front()];
            correct[que.front()] += correct[now.next[failure]];
            que.pop();
            rep(i, char_size){
                if(now.next[i] == -1) continue;
                int fail = now.next[failure];
                while(this->nodes[fail].next[i] == -1) fail = this->nodes[fail].next[failure];
                this->nodes[now.next[i]].next[failure] = this->nodes[fail].next[i];
                if(heavy){
                    auto &u = this->nodes[now.next[i]].accept;
                    auto &v = this->nodes[this->nodes[fail].next[i]].accept;
                    vector<int> accept;
                    set_union(all(u), all(v), back_inserter(accept));
                    u = accept;
                }
                que.emplace(now.next[i]);
            }
        }
    }

    map<int, int> match(const string &s){
        int now = 0;
        map<int, int> ret;
        for(auto &e: s){
            while(this->nodes[now].next[e-base] == -1) now = this->nodes[now].next[failure];
            now = this->nodes[now].next[e-base];
            for(auto &u: this->nodes[now].accept) ret[u]++;
        }
        return ret;
    }

    pii move(int now, const int c){
        while(this->nodes[now].next[c] == -1) now = this->nodes[now].next[failure];
        now = this->nodes[now].next[c];
        return pii(correct[now], now);
    }
};

int main(){
    int N;
    ll L, R;
    cin >> N >> L >> R;
    Aho_Corasick<10, '0'> trie;
    ll p = 1, q = 1;
    while(q <= R){
        if(q >= L) trie.insert(to_string(q));
        p += q, swap(p, q);
    }
    int K = trie.size();
    trie.build();
    int dp[N+1][K+1];
    fill(dp[0], dp[N+1], 0);
    dp[0][0] = 1;
    rep(i, N){
        rep(j, K){
            rep(k, 10){
                auto [p, q] = trie.move(j, k);
                if(p == 0) dp[i+1][q] += dp[i][j], dp[i+1][q] %= MOD;
            }
        }
    }
    int ans = MOD-1;
    rep(i, K+1) ans += dp[N][i], ans %= MOD;
    cout << ans << endl;
}