#670424 (C++17) No.1574 Swap and Repaint

提出ソース

結果

問題	No.1574 Swap and Repaint
ユーザー	KoD
提出日時	2021-06-26 20:08:17
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	WA
実行時間	-
コード長	5,397 bytes
コンパイル時間	4,118 ms
コンパイル使用メモリ	245,248 KB
実行使用メモリ	21,396 KB
最終ジャッジ日時	2024-04-14 13:47:31
合計ジャッジ時間	25,943 ms
ジャッジサーバーID （参考情報）	judge5 / judge1

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 6,816 KB
testcase_01	AC	2 ms 6,940 KB
testcase_02	AC	2 ms 6,940 KB
testcase_03	AC	2 ms 6,940 KB
testcase_04	AC	2 ms 6,944 KB
testcase_05	WA	-
testcase_06	AC	2 ms 6,948 KB
testcase_07	WA	-
testcase_08	WA	-
testcase_09	WA	-
testcase_10	WA	-
testcase_11	WA	-
testcase_12	WA	-
testcase_13	WA	-
testcase_14	WA	-
testcase_15	WA	-
testcase_16	WA	-
testcase_17	WA	-
testcase_18	WA	-
testcase_19	WA	-
testcase_20	WA	-
testcase_21	WA	-
testcase_22	WA	-
testcase_23	WA	-
testcase_24	WA	-
testcase_25	WA	-
testcase_26	WA	-
testcase_27	WA	-
testcase_28	WA	-
testcase_29	WA	-
testcase_30	WA	-
testcase_31	WA	-
testcase_32	WA	-
testcase_33	WA	-
testcase_34	WA	-
testcase_35	WA	-

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

ソースコード

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

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

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

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

  public:
    explicit AutoReallocation(F&& f) : func(std::forward<F>(f)), 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 F> decltype(auto) auto_realloc(F&& f) {
    using G = std::decay_t<F>;
    return AutoReallocation<G>(std::forward<G>(f));
}

template <class M> struct ModintUtil {
    static inline const auto fact = auto_realloc([](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 = auto_realloc([](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 = auto_realloc([](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>;

const auto INV2 = Fp(1) / Fp(2);

void main_() {
    usize N;
    std::cin >> N;
    Vec<Fp> A(N);
    for (auto& x : A) {
        u32 y;
        std::cin >> y;
        x = y;
    }
    Vec<Fp> C(N);
    {
        Vec<Fp> more(N);
        for (const auto i : rep(0, N)) {
            more[i] = Util::factpow(N - 1, (N - 1) - i);
        }
        Vec<Fp> pow2(2 * N);
        pow2[0] = 1;
        for (const auto i : rep(1, 2 * N)) {
            pow2[i] = pow2[i - 1] * 2;
        }
        Fp sum;
        for (const auto i : rep(0, N - 1)) {
            sum += (more[i] - more[i + 1] * INV2) * pow2[2 * ((N - 1) - i) + i];
            C[i] = sum;
        }
        for (const auto i : rep(0, N)) {
            C[N - 1] += more[i] * pow2[2 * ((N - 1) - i) + i];
        }
    }
    A.resize(2 * N);
    C.resize(2 * N);
    for (const auto i : rep(0, N)) {
        A[2 * N - i - 1] = A[i];
        C[2 * N - i - 1] = C[i];
    }
    const auto N2 = 2 * N;
    std::reverse(C.begin(), C.end());
    auto F = atcoder::convolution(A, C);
    for (const auto i : rep(N2, F.size())) {
        F[i - N2] += F[i];
    }
    F.resize(N2);
    std::reverse(F.begin(), F.end());
    const usize K = std::min(N2, (usize)(5 * std::sqrt(N)));
    Vec<Vec<Fp>> D_base(K + 1);
    D_base[0] = {Fp(1)};
    for (const auto i : rep(1, K + 1)) {
        D_base[i].resize(D_base[i - 1].size() + 2);
        for (const auto j : rep(0, D_base[i - 1].size())) {
            D_base[i][j] += D_base[i - 1][j];
            D_base[i][j + 1] += D_base[i - 1][j] * Fp(N - 3);
            D_base[i][j + 2] += D_base[i - 1][j];
        }
    }
    Vec<typename Vec<Fp>::const_iterator> D(K);
    for (const auto i : rep(0, K)) {
        D[i] = D_base[i].cbegin() + (D_base[i].size() / 2);
    }
    for (const auto i : rep(0, N + 1)) {
        if (i % K == 0 and i != 0) {
            const auto tmp = atcoder::convolution(F, D_base[K]);
            F.assign(N2, 0);
            for (const auto i : rep(0, tmp.size())) {
                F[(i < K ? i + N2 - K : i - K)] += tmp[i];
            }
        }
        Fp ans;
        const isize width = i % K;
        for (isize i = -width; i <= width; i += 1) {
            ans += F[(i < 0 ? i + N2 : i)] * D[width][i];
        }
        std::cout << (ans * INV2).val() << '\n';
    }
}

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

yukicoder

結果

テストケース

ソースコード