algo — C++23 日本語リファレンス GitHub

test/yosupo/set_power_series/subset_convolution.test.cpp

AC · 最終更新 2026-09-07 · GitHub

問題: https://judge.yosupo.jp/problem/subset_convolution

依存するヘッダ(4)
実装
// competitive-verifier: PROBLEM https://judge.yosupo.jp/problem/subset_convolution
#include <iostream>
#include <vector>
#include "convolution/subset_convolution.hpp"
#include "number_theory/modint.hpp"

using Mint = modint998;

int main(void) {
    int n;
    std::cin >> n;
    std::vector<Mint> a(1 << n), b(1 << n);
    for (auto &e : a) std::cin >> e;
    for (auto &e : b) std::cin >> e;
    auto ans = subset_convolution(a, b);
    for (int i = 0; i < (int)ans.size(); ++i) std::cout << ans[i] << (i == (int)ans.size() - 1 ? '\n' : ' ');

    return 0;
}