serna37's Library

Logo

C++ アルゴリズムとデータ構造のライブラリ

View the Project on GitHub serna37/library-cpp

:heavy_check_mark: 上位集合のゼータ・メビウス変換
(library/polynomial/fft/superset_zeta_moebius_transform.hpp)

上位集合のゼータ・メビウス変換

できること

Image

計算量

$O(N \cdot 2^N)$

使い方

superset_zeta_transform(f);
superset_moebius_transform(f);

Required by

Verified with

Code

#pragma once
template <typename T> void superset_zeta_transform(vector<T> &f) {
    const int n = (int)f.size();
    assert((n & (n - 1)) == 0ll);
    for (int i = 1; i < n; i <<= 1) {
        for (int j = 0; j < n; j += i << 1) {
            for (int k = 0; k < i; k++) {
                f[j + k] += f[j + k + i];
            }
        }
    }
}
template <typename T> void superset_moebius_transform(vector<T> &f) {
    const int n = (int)f.size();
    assert((n & (n - 1)) == 0ll);
    for (int i = 1; i < n; i <<= 1) {
        for (int j = 0; j < n; j += i << 1) {
            for (int k = 0; k < i; k++) {
                f[j + k] -= f[j + k + i];
            }
        }
    }
}
#line 2 "library/polynomial/fft/superset_zeta_moebius_transform.hpp"
template <typename T> void superset_zeta_transform(vector<T> &f) {
    const int n = (int)f.size();
    assert((n & (n - 1)) == 0ll);
    for (int i = 1; i < n; i <<= 1) {
        for (int j = 0; j < n; j += i << 1) {
            for (int k = 0; k < i; k++) {
                f[j + k] += f[j + k + i];
            }
        }
    }
}
template <typename T> void superset_moebius_transform(vector<T> &f) {
    const int n = (int)f.size();
    assert((n & (n - 1)) == 0ll);
    for (int i = 1; i < n; i <<= 1) {
        for (int j = 0; j < n; j += i << 1) {
            for (int k = 0; k < i; k++) {
                f[j + k] -= f[j + k + i];
            }
        }
    }
}
Back to top page