結果

問題 No.1458 Segment Function
ユーザー KoDKoD
提出日時 2021-03-31 21:18:43
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 9 ms / 2,000 ms
コード長 2,319 bytes
コンパイル時間 2,228 ms
コンパイル使用メモリ 204,816 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-10-12 13:33:06
合計ジャッジ時間 3,556 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 2 ms
5,248 KB
testcase_03 AC 2 ms
5,248 KB
testcase_04 AC 2 ms
5,248 KB
testcase_05 AC 2 ms
5,248 KB
testcase_06 AC 2 ms
5,248 KB
testcase_07 AC 2 ms
5,248 KB
testcase_08 AC 3 ms
5,248 KB
testcase_09 AC 3 ms
5,248 KB
testcase_10 AC 4 ms
5,248 KB
testcase_11 AC 3 ms
5,248 KB
testcase_12 AC 3 ms
5,248 KB
testcase_13 AC 7 ms
5,248 KB
testcase_14 AC 7 ms
5,248 KB
testcase_15 AC 7 ms
5,248 KB
testcase_16 AC 7 ms
5,248 KB
testcase_17 AC 7 ms
5,248 KB
testcase_18 AC 9 ms
5,248 KB
testcase_19 AC 9 ms
5,248 KB
testcase_20 AC 9 ms
5,248 KB
testcase_21 AC 9 ms
5,248 KB
testcase_22 AC 9 ms
5,248 KB
testcase_23 AC 2 ms
5,248 KB
testcase_24 AC 3 ms
5,248 KB
testcase_25 AC 2 ms
5,248 KB
testcase_26 AC 9 ms
5,248 KB
testcase_27 AC 9 ms
5,248 KB
testcase_28 AC 2 ms
5,248 KB
testcase_29 AC 3 ms
5,248 KB
testcase_30 AC 3 ms
5,248 KB
testcase_31 AC 8 ms
5,248 KB
testcase_32 AC 3 ms
5,248 KB
testcase_33 AC 3 ms
5,248 KB
testcase_34 AC 3 ms
5,248 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

using i32 = std::int32_t;
using u32 = std::uint32_t;
using i64 = std::int64_t;
using u64 = std::uint64_t;
using i128 = __int128_t;
using u128 = __uint128_t;
using isize = std::ptrdiff_t;
using usize = std::size_t;

class rep {
    struct Iter {
        usize itr;
        constexpr Iter(const usize pos) noexcept: itr(pos) { }
        constexpr void operator ++ () noexcept { ++itr; }
        constexpr bool operator != (const Iter& other) const noexcept { return itr != other.itr; }
        constexpr usize operator * () const noexcept { return itr; }
    };
    const Iter first, last;
public:
    explicit constexpr rep(const usize first, const usize last) noexcept: first(first), last(std::max(first, last)) { }
    constexpr Iter begin() const noexcept { return first; }
    constexpr Iter end() const noexcept { return last; }
};

template <class T>
using Vec = std::vector<T>;

constexpr usize SEG[] = { 6, 2, 5, 5, 4, 5, 6, 4, 7, 6 };

u64 f(const u64 x) {
    if (x < 10) {
        return SEG[x];
    }
    return SEG[x % 10] + f(x / 10);
}

u64 take_mod(const std::string& S, const u64 mod) {
    u64 ret = 0;
    for (const auto c: S) {
        ret = (ret * 10) % mod;
        ret += c - '0';
        ret %= mod;
    }
    return ret;
}

void F_main() {
    std::string P;
    std::cin >> P;
    std::string N;
    std::cin >> N;
    if (N == "0") {
        std::cout << P << '\n';
        return;
    }
    std::array<u64, 1001> A{};
    std::reverse(P.begin(), P.end());
    if (P.back() == '-') {
        A[1] += 1;
        P.pop_back();
    }
    while (!P.empty()) {
        A[1] += SEG[P.back() - '0'];
        P.pop_back();
    }
    for (auto i: rep(1, 1000)) {
        A[i + 1] = f(A[i]);
    }
    if (N.size() <= 4) {
        const auto n = std::stoi(N);
        if (n <= 1000) {
            std::cout << A[n] << '\n';
            return;
        }
    }
    Vec<u64> loop;
    std::array<bool, 10> done{};
    {
        u64 x = A[1000];
        while (!done[x]) {
            loop.push_back(x);
            done[x] = true;
            x = SEG[x];
        }
    }
    std::cout << loop[take_mod(N, loop.size()) - 1000 % loop.size()] << '\n';
    return;
}

int main() {
    std::ios_base::sync_with_stdio(false);
    std::cin.tie(nullptr);
    F_main(); 
    return 0;
}
0