#967931 (C++23) No.1269 I hate Fibonacci Number

提出ソース

結果

問題	No.1269 I hate Fibonacci Number
ユーザー	cureskol
提出日時	2024-03-31 20:44:33
言語	C++23 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	450 ms / 3,000 ms
コード長	10,607 bytes
コンパイル時間	4,499 ms
コンパイル使用メモリ	295,544 KB
実行使用メモリ	6,548 KB
最終ジャッジ日時	2024-03-31 20:44:42
合計ジャッジ時間	8,641 ms
ジャッジサーバーID （参考情報）	judge12 / judge14

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 6,548 KB
testcase_01	AC	2 ms 6,548 KB
testcase_02	AC	14 ms 6,548 KB
testcase_03	AC	2 ms 6,548 KB
testcase_04	AC	2 ms 6,548 KB
testcase_05	AC	2 ms 6,548 KB
testcase_06	AC	2 ms 6,548 KB
testcase_07	AC	2 ms 6,548 KB
testcase_08	AC	2 ms 6,548 KB
testcase_09	AC	2 ms 6,548 KB
testcase_10	AC	2 ms 6,548 KB
testcase_11	AC	2 ms 6,548 KB
testcase_12	AC	2 ms 6,548 KB
testcase_13	AC	6 ms 6,548 KB
testcase_14	AC	344 ms 6,548 KB
testcase_15	AC	19 ms 6,548 KB
testcase_16	AC	436 ms 6,548 KB
testcase_17	AC	3 ms 6,548 KB
testcase_18	AC	450 ms 6,548 KB
testcase_19	AC	212 ms 6,548 KB
testcase_20	AC	48 ms 6,548 KB
testcase_21	AC	239 ms 6,548 KB
testcase_22	AC	182 ms 6,548 KB
testcase_23	AC	190 ms 6,548 KB
testcase_24	AC	115 ms 6,548 KB
testcase_25	AC	108 ms 6,548 KB
testcase_26	AC	5 ms 6,548 KB
testcase_27	AC	2 ms 6,548 KB
testcase_28	AC	5 ms 6,548 KB
testcase_29	AC	47 ms 6,548 KB
testcase_30	AC	15 ms 6,548 KB
testcase_31	AC	271 ms 6,548 KB
testcase_32	AC	143 ms 6,548 KB
testcase_33	AC	11 ms 6,548 KB
testcase_34	AC	2 ms 6,548 KB
testcase_35	AC	12 ms 6,548 KB
testcase_36	AC	2 ms 6,548 KB
testcase_37	AC	2 ms 6,548 KB
testcase_38	AC	2 ms 6,548 KB

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

ソースコード

raw source code

#pragma region template
#pragma GCC optimize("Ofast")
#include <bits/stdc++.h>
using namespace std;

using ll = long long;
using ld = long double;
using vi = vector<int>;
using vll = vector<ll>;
using pi = pair<int, int>;
using pll = pair<ll, ll>;

#define overload2(a, b, c, ...) c
#define overload3(a, b, c, d, ...) d
#define overload4(a, b, c, d, e, ...) e
#define overload5(a, b, c, d, e, f, ...) f

#define TYPE1(T) template <typename T>
#define TYPE2(T, U) template <typename T, typename U>
#define TYPE(...) overload2(__VA_ARGS__, TYPE2, TYPE1)(__VA_ARGS__)
#define TYPES1(T) template <typename... T>
#define TYPES2(H, T) template <typename H, typename... T>
#define TYPES(...) overload2(__VA_ARGS__, TYPES2, TYPES1)(__VA_ARGS__)

#define REP4(i, s, n, d) for (int i = (s); i < (n); i += (d))
#define REP3(i, s, n) REP4(i, s, n, 1)
#define REP2(i, n) REP3(i, 0, n)
#define REP1(n) REP2(tomato, n)
#define REP(...) overload4(__VA_ARGS__, REP4, REP3, REP2, REP1)(__VA_ARGS__)

#define RREP4(i, n, s, d) for (int i = (n)-1; i >= (s); i -= (d))
#define RREP3(i, n, s) RREP4(i, n, s, 1)
#define RREP2(i, n) RREP3(i, n, 0)
#define RREP1(n) RREP2(tomato, n)
#define RREP(...)                                                              \
    overload4(__VA_ARGS__, RREP4, RREP3, RREP2, RREP1)(__VA_ARGS__)

#define FOR4(a, b, c, d, v) for (auto [a, b, c, d] : v)
#define FOR3(a, b, c, v) for (auto [a, b, c] : v)
#define FOR2(a, b, v) for (auto [a, b] : v)
#define FOR1(a, v) for (auto a : v)
#define FOR(...) overload5(__VA_ARGS__, FOR4, FOR3, FOR2, FOR1)(__VA_ARGS__)

#define AFOR4(a, b, c, d, v) for (auto &[a, b, c, d] : v)
#define AFOR3(a, b, c, v) for (auto &[a, b, c] : v)
#define AFOR2(a, b, v) for (auto &[a, b] : v)
#define AFOR1(a, v) for (auto &a : v)
#define AFOR(...)                                                              \
    overload5(__VA_ARGS__, AFOR4, AFOR3, AFOR2, AFOR1)(__VA_ARGS__)

#define CFOR4(a, b, c, d, v) for (const auto &[a, b, c, d] : v)
#define CFOR3(a, b, c, v) for (const auto &[a, b, c] : v)
#define CFOR2(a, b, v) for (const auto &[a, b] : v)
#define CFOR1(a, v) for (const auto &a : v)
#define CFOR(...)                                                              \
    overload5(__VA_ARGS__, CFOR4, CFOR3, CFOR2, CFOR1)(__VA_ARGS__)

#define PFOR(v)                                                                \
    for (bool f = true; (f ? exchange(f, false) : next_permutation(ALL(v)));)

#define ALL(v) v.begin(), v.end()
#define RALL(v) v.rbegin(), v.rend()
#define SORT(v) sort(ALL(v))
#define RSORT(v) sort(RALL(v))
#define REVERSE(v) reverse(ALL(v))
#define UNIQUE(v) SORT(v), v.erase(unique(ALL(v)), v.end())
#define SZ(v) int(v.size())
#define MAX(v) *max_element(ALL(v))
#define MIN(v) *min_element(ALL(v))
template <typename T>
conditional_t<is_same_v<T, int>, ll, T> SUM(const vector<T> &v) {
    return accumulate(ALL(v), conditional_t<is_same_v<T, int>, ll, T>(0));
}

TYPES(T) void input(T &...a) { (cin >> ... >> a); }
#define DECLARE(T, ...)                                                        \
    T __VA_ARGS__;                                                             \
    input(__VA_ARGS__);
#define INT(...) DECLARE(int, __VA_ARGS__)
#define STR(...) DECLARE(string, __VA_ARGS__)
#define LL(...) DECLARE(long long, __VA_ARGS__)
#define CHR(...) DECLARE(char, __VA_ARGS__)
#define DBL(...) DECLARE(double, __VA_ARGS__)
#define VI(n, v)                                                               \
    vi v(n);                                                                   \
    cin >> v;
#define VLL(n, v)                                                              \
    vll v(n);                                                                  \
    cin >> v;
#define VS(n, s)                                                               \
    vector<string> s(n);                                                       \
    cin >> s;

TYPE(T) istream &operator>>(istream &is, vector<T> &v) {
    for (auto &a : v)
        cin >> a;
    return is;
}
TYPE(T) ostream &operator<<(ostream &os, const vector<T> &v) {
    if (&os == &cerr)
        os << "[";
    REP (i, v.size()) {
        os << v[i];
        if (i + 1 < v.size())
            os << (&os == &cerr ? "," : " ");
    }
    if (&os == &cerr)
        os << "]";
    return os;
}
TYPE(T, S) istream &operator>>(istream &is, pair<T, S> &p) {
    cin >> p.first >> p.second;
    return is;
}

#ifdef __DEBUG
#include <debug>
#else
#define debug(...) void(0)
#endif

void print() { cout << '\n'; }
TYPES(T, Ts) void print(const T &a, const Ts &...b) {
    cout << a;
    (cout << ... << (cout << ' ', b));
    cout << '\n';
}

TYPE(T) using pq = priority_queue<T>;
TYPE(T) using pqg = priority_queue<T, vector<T>, greater<T>>;
TYPE(T) T pick(queue<T> &que) {
    assert(que.size());
    T a = que.front();
    que.pop();
    return a;
}
TYPE(T) T pick(pq<T> &que) {
    assert(que.size());
    T a = que.top();
    que.pop();
    return a;
}
TYPE(T) T pick(pqg<T> &que) {
    assert(que.size());
    T a = que.top();
    que.pop();
    return a;
}
TYPE(T) T pick(stack<T> &sta) {
    assert(sta.size());
    T a = sta.top();
    sta.pop();
    return a;
}
TYPE(T) void clear(T &v) { v = decltype(v)(); }

string YES(bool f = true) { return (f ? "YES" : "NO"); }
string Yes(bool f = true) { return (f ? "Yes" : "No"); }
string yes(bool f = true) { return (f ? "yes" : "no"); }

constexpr int INF = 1e9 + 7;
constexpr ll LINF = ll(1e18) + 7;
constexpr ld EPS = 1e-10;

vi iota(int n) {
    vi a(n);
    iota(ALL(a), 0);
    return a;
}
TYPE(T) vector<pair<T, int>> query_sort(const vector<T> &v) {
    vector<pair<T, int>> res(v.size());
    REP (i, v.size())
        res[i] = {v[i], i};
    SORT(res);
    return res;
}
TYPE(T) void add(vector<T> &v, T a = 1) {
    AFOR (p, v)
        p += a;
}
TYPE(T) T rev(T a) {
    REVERSE(a);
    return a;
}
TYPE(T) void fin(T a) {
    cout << a << '\n';
    exit(0);
}
TYPE(T) bool chmax(T &a, T b) { return (a < b && (a = b, true)); }
TYPE(T) bool chmin(T &a, T b) { return (a > b && (a = b, true)); }
TYPES(T, Ns) auto make_vector(T x, int n, Ns... ns) {
    if constexpr (sizeof...(ns) == 0)
        return vector<T>(n, x);
    else
        return vector(n, make_vector<T>(x, ns...));
}
bool in(const ll S, const int a) { return (S >> a) & 1; }
int popcount(const ll S) { return __builtin_popcountll(S); }
int digit(char c) { return (c >= '0' and c <= '9' ? c - '0' : -1); }
ll sqrtll(ll a) {
    for (ll b = sqrt(a); b * b <= a; b++)
        if (b * b == a)
            return b;
    for (ll b = sqrt(a); b >= 0 and b * b >= a; b--)
        if (b * b == a)
            return b;
    return -1;
}
#pragma endregion template

#include <bits/stdc++.h>

template <size_t SIGMA = 26, char MARGIN = 'a'> class Trie {
    struct Node {
        std::array<int, SIGMA> nxt;
        Node() { std::ranges::fill(nxt, -1); }
    };

  protected:
    std::vector<Node> nodes;

  public:
    Trie() : nodes(1, Node()) {}

    int &nxt(size_t v, char c) { return nodes[v].nxt[c - MARGIN]; }

    size_t add(const std::string &s) {
        size_t now = 0;
        for (char c : s) {
            if (nxt(now, c) < 0) {
                nxt(now, c) = nodes.size();
                nodes.push_back(Node());
            }
            now = nxt(now, c);
        }
        return now;
    }

    std::vector<size_t> add(const std::vector<std::string> &v) {
        std::vector<size_t> ret;
        ret.reserve(v.size());
        for (const std::string &s : v)
            ret.push_back(add(s));
        return ret;
    }
};

template <size_t SIGMA = 26, char MARGIN = 'a'>
class AhoCorasick : Trie<SIGMA, MARGIN> {
    using super = Trie<SIGMA, MARGIN>;
    using super::nodes;

    std::vector<std::map<size_t, uint32_t>> vertex_counts;

  public:
    using super::nxt;
    using super::Trie;

    std::vector<size_t> add(const std::vector<std::string> &v) {
        auto ret = super::add(v);
        size_t n = nodes.size();
        vertex_counts.resize(n);

        std::vector<bool> exist(n, false);
        for (size_t v : ret)
            exist[v] = true;

        std::vector<size_t> suffix(n);
        std::queue<size_t> que;
        que.push(0);
        while (que.size()) {
            size_t v = que.front();
            que.pop();
            if (v) {
                if (exist[v])
                    vertex_counts[v][v]++;
                for (const auto &[key, value] : vertex_counts[suffix[v]])
                    vertex_counts[v][key] += value;
            }

            for (size_t i = 0; i < SIGMA; i++) {
                int &to = nodes[v].nxt[i];
                if (~to) {
                    suffix[to] = v ? nodes[suffix[v]].nxt[i] : 0;
                    que.push(to);
                } else
                    to = v ? nodes[suffix[v]].nxt[i] : 0;
            }
        }
        return ret;
    }

    const std::map<size_t, uint32_t> &count(size_t v) const {
        return vertex_counts[v];
    }

    std::map<size_t, uint32_t> path(const std::string &s) const {
        std::map<size_t, uint32_t> ret;
        size_t v = 0;
        for (const char &c : s) {
            v = nxt(v, c);
            for (const auto &[key, value] : vertex_counts)
                ret[key] += value;
        }
        return ret;
    }
};

#include <atcoder/modint>
using namespace atcoder;
using mint = modint1000000007;
namespace atcoder {
ostream &operator<<(ostream &os, mint a) {
    os << a.val();
    return os;
}
istream &operator>>(istream &is, mint &a) {
    long long b;
    is >> b;
    a = b;
    return is;
}
} // namespace atcoder

int main() {
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);

    int n;
    std::cin >> n;
    uint64_t l, r;
    std::cin >> l >> r;

    std::vector<uint64_t> fib{1, 2};
    while (fib.back() < r)
        fib.push_back(fib[fib.size() - 2] + fib[fib.size() - 1]);

    std::vector<std::string> v;
    for (const auto &x : fib) {
        if (std::clamp(x, l, r) != x)
            continue;
        v.push_back(std::to_string(x));
    }

    AhoCorasick<10, '0'> T;
    T.add(v);

    std::map<size_t, mint> dp{{0, 1}};
    REP (n) {
        std::map<size_t, mint> nxt;
        for (const auto &[v, val] : dp) {
            for (char c = '0'; c <= '9'; c++) {
                size_t to = T.nxt(v, c);
                if (T.count(to).size())
                    continue;
                nxt[to] += val;
            }
        }
        dp = nxt;
    }
    mint ans = -1;
    for (const auto &[key, val] : dp)
        ans += val;
    std::cout << ans << '\n';
}

yukicoder

結果

テストケース

ソースコード