#655519 (C++17) No.1504 ヌメロニム

提出ソース

結果

問題	No.1504 ヌメロニム
ユーザー	KoD
提出日時	2021-05-08 10:26:53
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	185 ms / 2,000 ms
コード長	4,365 bytes
コンパイル時間	3,366 ms
コンパイル使用メモリ	242,028 KB
実行使用メモリ	28,368 KB
最終ジャッジ日時	2023-10-14 13:49:50
合計ジャッジ時間	8,941 ms
ジャッジサーバーID （参考情報）	judge11 / judge14

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 4,372 KB
testcase_01	AC	1 ms 4,372 KB
testcase_02	AC	2 ms 4,368 KB
testcase_03	AC	1 ms 4,368 KB
testcase_04	AC	1 ms 4,368 KB
testcase_05	AC	1 ms 4,368 KB
testcase_06	AC	2 ms 4,368 KB
testcase_07	AC	1 ms 4,372 KB
testcase_08	AC	1 ms 4,368 KB
testcase_09	AC	1 ms 4,368 KB
testcase_10	AC	2 ms 4,376 KB
testcase_11	AC	2 ms 4,376 KB
testcase_12	AC	2 ms 4,376 KB
testcase_13	AC	1 ms 4,368 KB
testcase_14	AC	1 ms 4,368 KB
testcase_15	AC	1 ms 4,368 KB
testcase_16	AC	2 ms 4,368 KB
testcase_17	AC	1 ms 4,372 KB
testcase_18	AC	1 ms 4,372 KB
testcase_19	AC	1 ms 4,368 KB
testcase_20	AC	3 ms 4,372 KB
testcase_21	AC	2 ms 4,372 KB
testcase_22	AC	3 ms 4,372 KB
testcase_23	AC	1 ms 4,372 KB
testcase_24	AC	180 ms 27,928 KB
testcase_25	AC	94 ms 16,644 KB
testcase_26	AC	180 ms 28,348 KB
testcase_27	AC	96 ms 17,204 KB
testcase_28	AC	95 ms 17,208 KB
testcase_29	AC	178 ms 28,152 KB
testcase_30	AC	176 ms 28,292 KB
testcase_31	AC	93 ms 16,864 KB
testcase_32	AC	93 ms 17,392 KB
testcase_33	AC	176 ms 28,248 KB
testcase_34	AC	23 ms 6,640 KB
testcase_35	AC	22 ms 6,156 KB
testcase_36	AC	90 ms 15,612 KB
testcase_37	AC	11 ms 4,764 KB
testcase_38	AC	181 ms 28,288 KB
testcase_39	AC	180 ms 28,212 KB
testcase_40	AC	177 ms 28,192 KB
testcase_41	AC	178 ms 28,360 KB
testcase_42	AC	177 ms 28,268 KB
testcase_43	AC	177 ms 28,200 KB
testcase_44	AC	177 ms 28,296 KB
testcase_45	AC	178 ms 28,204 KB
testcase_46	AC	185 ms 28,188 KB
testcase_47	AC	179 ms 28,368 KB
testcase_48	AC	94 ms 16,824 KB
testcase_49	AC	91 ms 16,636 KB
testcase_50	AC	3 ms 4,372 KB
testcase_51	AC	2 ms 4,368 KB
testcase_52	AC	2 ms 4,368 KB
testcase_53	AC	2 ms 4,372 KB
testcase_54	AC	2 ms 4,372 KB
testcase_55	AC	22 ms 6,132 KB
testcase_56	AC	1 ms 4,368 KB
testcase_57	AC	1 ms 4,368 KB
testcase_58	AC	2 ms 4,372 KB
testcase_59	AC	1 ms 4,368 KB
testcase_60	AC	1 ms 4,372 KB

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

ソースコード

raw source code

#include <bits/stdc++.h>
#include <atcoder/convolution>
#include <atcoder/modint>

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; }
};

class revrep {
    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 revrep(const usize first, const usize last) noexcept
        : first(last - 1), last(std::min(first, last) - 1) {}
    constexpr Iter begin() const noexcept { return first; }
    constexpr Iter end() const noexcept { return last; }
};

constexpr u64 ceil_log2(const u64 x) {
    u64 e = 0;
    while (((u64)1 << e) < x) ++e;
    return e;
}

template <class F> class AutoRealloc {
    using R = typename decltype(std::declval<F>()((usize)0))::value_type;

    F func;
    mutable std::vector<R> data;

  public:
    template <class G>
    explicit AutoRealloc(G&& g) : func(std::forward<G>(g)), data() {}

    void reserve(const usize size) const {
        if (data.size() < size) data = func(((usize)1 << ceil_log2(size)));
    }
    R operator[](const usize i) const {
        reserve(i + 1);
        return data[i];
    }
};

template <class G> explicit AutoRealloc(G&&) -> AutoRealloc<std::decay_t<G>>;

template <class M> struct ModintUtil {
    static inline const auto fact = AutoRealloc([](const usize n) {
        std::vector<M> ret(n);
        ret[0] = M(1);
        for (const usize i : rep(1, n)) {
            ret[i] = ret[i - 1] * M(i);
        }
        return ret;
    });
    static inline const auto inv = AutoRealloc([](const usize n) {
        std::vector<M> ret(n);
        if (n == 1) return ret;
        ret[1] = M(1);
        for (const usize i : rep(2, n)) {
            ret[i] = -M(M::mod() / i) * ret[M::mod() % i];
        }
        return ret;
    });
    static inline const auto inv_fact = AutoRealloc([](const usize n) {
        std::vector<M> ret(n);
        ret[0] = M(1);
        for (const usize i : rep(1, n)) {
            ret[i] = ret[i - 1] * inv[i];
        }
        return ret;
    });
    static M binom(const usize n, const usize k) {
        assert(k <= n);
        return fact[n] * inv_fact[n - k] * inv_fact[k];
    }
    static M factpow(const usize n, const usize k) {
        assert(k <= n);
        return fact[n] * inv_fact[n - k];
    }
    static M homo(const usize n, const usize k) {
        if (n == 0 and k == 0) return M(1);
        return binom(n + k - 1, k);
    }
};

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

using Fp = atcoder::modint998244353;
using Util = ModintUtil<Fp>;

void F_main() {
    usize N;
    std::cin >> N;
    Vec<Fp> A(N), B(N);
    for (const auto i : rep(0, N)) {
        char c;
        std::cin >> c;
        (c == 'i' ? A[N - i - 1] : B[i]) = Fp(1);
    }
    auto C = atcoder::convolution(A, B);
    C.erase(C.begin(), C.begin() + N);
    Vec<Fp> D(N - 1);
    for (const auto i : rep(0, N - 1)) {
        C[i] *= Util::fact[i];
        D[i] = Util::inv_fact[N - i - 2];
    }
    const auto E = atcoder::convolution(C, D);
    u32 ans = 0;
    for (const auto i : rep(N - 2, 2 * N - 3)) {
        ans ^= (E[i] * Util::inv_fact[i - (N - 2)]).val();
    }
    std::cout << ans << '\n';
}

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

yukicoder

結果

テストケース

ソースコード