結果

問題 No.1504 ヌメロニム
コンテスト
ユーザー KoD
提出日時 2021-05-07 22:48:36
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,470 ms / 2,000 ms
コード長 5,488 bytes
コンパイル時間 3,667 ms
コンパイル使用メモリ 241,868 KB
最終ジャッジ日時 2025-01-21 08:36:04
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 61
権限があれば一括ダウンロードができます

ソースコード

diff # 

#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>;
using Poly = Vec<Fp>;

void operator+=(Poly& p, const Poly& q) {
    if (p.size() < q.size()) {
        p.resize(q.size(), Fp(0));
    }
    for (const auto i : rep(0, q.size())) {
        p[i] += q[i];
    }
}

Poly operator*(const Poly& p, const Poly& q) {
    return atcoder::convolution(p, q);
}

struct Data {
    Vec<Fp> left0, right1;
    Data() : left0(1, Fp(0)), right1(1, Fp(0)) {}
};

void F_main() {
    usize N;
    std::cin >> N;
    Vec<bool> S(N);
    for (const auto i : rep(0, N)) {
        char c;
        std::cin >> c;
        S[i] = (c == 'n');
    }
    const usize H = ceil_log2(N);
    const usize M = 1 << H;
    Vec<Data> data(2 * M);
    for (const auto i : rep(0, N)) {
        if (S[i]) {
            data[M + i].right1[0] = Fp(1);
        } else {
            data[M + i].left0[0] = Fp(1);
        }
    }
    Vec<Fp> ans(N - 1);
    Vec<Fp> mul;
    for (const auto i : revrep(1, M)) {
        {
            usize idx = 0;
            for (const auto x : data[2 * i].left0* data[2 * i + 1].right1) {
                if (idx <= N - 2) {
                    ans[idx] += x;
                }
                idx += 1;
            }
        }
        const auto size = data[2 * i].left0.size();
        if (mul.size() < size + 1) {
            mul.assign(size + 1, Fp(0));
            for (const auto i : rep(0, size + 1)) {
                mul[i] = Util::binom(size, i);
            }
        }
        data[i].left0 = std::move(data[2 * i + 1].left0);
        data[i].left0 += data[2 * i].left0 * mul;
        data[i].right1 = std::move(data[2 * i].right1);
        data[i].right1 += data[2 * i + 1].right1 * mul;
    }
    u32 sum = 0;
    for (const auto x : ans) {
        sum ^= x.val();
    }
    std::cout << sum << '\n';
}

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