Petrozavodsk Winter Camp, Andrew, 2014, Dichromatic Trees

条件：

1:每个红色节点的儿子都是黑色节点

2.每个叶子到根路径上的黑点数相等，等于某个常数，称作树的black height

求给定black height和节点数的符合条件的方案数

$black_{h} = x (black_{h-1} + red_{h-1})^2$

$red_{h} = x black_{h}^2$

任意模数的fft

#include <bits/stdc++.h>
using namespace std;
#define rep(i, j, k) for (int i = int(j); i <= int(k); ++ i)
#define dwn(i, j, k) for (int i = int(j); i >= int(k); -- i)
typedef long long LL;
const int MOD = ;
const int N =  << ;
const long double PI = acos(-.);
LL red[][N], black[][N], M = ;
struct Complex {
    long double r, i;
    Complex(long double _r = , long double _i = ) {
        r = _r; i = _i;
    }
    Complex operator + (const Complex &rhs) const {
        return Complex(r + rhs.r, i + rhs.i);
    }
    Complex operator - (const Complex &rhs) const {
        return Complex(r - rhs.r, i - rhs.i);
    }
    Complex operator * (const Complex &rhs) const {
        return Complex(r * rhs.r - i * rhs.i, r * rhs.i + i * rhs.r);
    }
    Complex operator / (long double b) const {
        return Complex(r / b, i / b);
    }
};
int a[N], b[N], c[N];
Complex conj(Complex &a) {
    return Complex(a.r, -a.i);
}
void fft(Complex *a, int n, int t) {
    int k = ;
    while (( << k) < n) k ++;
    rep(i, , n - ) {
        int t = ;
        rep(j, , k - ) if ((i >> j) & ) t |= ( << (k -  - j));
        if (i < t) swap(a[i], a[t]);
    }

    for (int l = ; l <= n; l <<= ) {
        int m = l >> ;
        long double o =  * PI / l * t; Complex _w(cos(o), sin(o));
        for (int i = ; i < n; i += l) {
            Complex w(, );
            rep(j, , m - ) {
                Complex x = w * a[i + j + m];
                a[i + j + m] = a[i + j] - x;
                a[i + j] = a[i + j] + x;
                w = w * _w;
            }
        }
    }
    if (t == -) rep(i, , n - ) a[i] = a[i] / n;
}
void Mul(int *A, int *B, int *C, int len, LL P) {
    for(int i = ;i < len; ++i) (A[i] += P) %= P, (B[i] += P) %= P;
    static Complex a[N], b[N], Da[N], Db[N], Dc[N], Dd[N];
    for(int i = ;i < len; ++i) a[i] = Complex(A[i] & M, A[i] >> );
    for(int i = ;i < len; ++i) b[i] = Complex(B[i] & M, B[i] >> );
    fft(a, len, ); fft(b, len, );
    for(int i = ;i < len; ++i) {
        int j = (len - i) & (len - ); static Complex da, db, dc, dd;
        da = (a[i] + conj(a[j])) * Complex(0.5, );
        db = (a[i] - conj(a[j])) * Complex(, -0.5);
        dc = (b[i] + conj(b[j])) * Complex(0.5, );
        dd = (b[i] - conj(b[j])) * Complex(, -0.5);
        Da[j] = da * dc; Db[j] = da * dd; Dc[j] = db * dc; Dd[j] = db * dd; //顺便区间反转，方便等会直接用DFT代替IDFT
    }
    for(int i = ;i < len; ++i) a[i] = Da[i] + Db[i] * Complex(, );
    for(int i = ;i < len; ++i) b[i] = Dc[i] + Dd[i] * Complex(, );
    fft(a, len, ); fft(b, len, );
    for(int i = ;i < len; ++i) {
        int da = (LL) (a[i].r / len + 0.5) % P; //直接取实部和虚部
        int db = (LL) (a[i].i / len + 0.5) % P;
        int dc = (LL) (b[i].r / len + 0.5) % P;
        int dd = (LL) (b[i].i / len + 0.5) % P;
        C[i] = (da + ((LL)(db + dc) << ) + ((LL)dd << )) % P;
    }
}

int main() {
    red[][] = ; red[][] = ;
    int len1 =  << , len2 =  << ;

    rep(h, , ) {
        rep(i, , len1 - ) a[i] = (black[h - ][i] + red[h - ][i]) % MOD, b[i] = a[i];
        rep(i, len1, len2 - ) a[i] = b[i] = ;
        Mul(a, b, c, len2, MOD);
        rep(i, , len1) black[h][i] = c[i - ];
        rep(i, , len1 - ) a[i] = black[h][i], b[i] = a[i];
        rep(i, len1, len2 - ) a[i] = b[i] = ;
        Mul(a, b, c, len2, MOD);
        rep(i, , len1) red[h][i] = c[i - ];
     }

    rep(i, , len1) rep(j, , ) {
        (red[j][i] += red[j - ][i]) %= MOD;
        (black[j][i] += black[j - ][i]) %= MOD;
    }
    // rep(i, 0, 10) cout <<  black[1][i] << ' '; cout << '\n';
    int k, H, n;
    scanf("%d%d", &k, &H);
    rep(i, , k) {
        scanf("%d", &n);
        cout << (red[H][n] + black[H][n]) % MOD << '\n';
    }
}