[JZOJ4921] 幻魔皇

Description

幻魔皇拉比艾尔很喜欢斐波那契树，他想找到神奇的节点对。
所谓斐波那契树，根是一个白色节点，每个白色节点都有一个黑色节点儿子，而每个黑色节点则有一个白色和一个黑色节点儿子。神奇的节点对则是指白色节点对。
请问对于深度为n的斐波那契树，其中距离为i的神奇节点对有多少个？拉比艾尔需要你对于1<=i<=2n的所有i都求出答案。
对于100%的数据n<=5000。
对123456789取模

Solution

可以先把树构出来YY一下
枚举$i$
显然随手dp题
$f[j][0,1]$表示深度为 j <script type="math/tex" id="MathJax-Element-195">j</script>，根的颜色。
明显每一层的白点数也是Fibonacci数列
如果根是白点，新增根与子树中所有满足的白点。

根是黑点，那么就两边的加起来加上一个类似于卷积和的东西，对应层的白点数乘起来。

Code

#include <cstdio>
#include <iostream>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <cstdlib>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fod(i,a,b) for(int i=a;i>=b;i--)
#define LL long long
#define N 5005
#define mo 123456789
using namespace std;
LL n,f[N],sum[N],g[N][2];
int main()
{
    freopen("raviel.in","r",stdin);
    freopen("raviel.out","w",stdout);
    cin>>n;
    f[0]=1;
    f[1]=0;
    fo(i,2,n) 
    {
        f[i]=(f[i-1]+f[i-2])%mo;
    }
    sum[1]=0;
    fo(i,1,2*n)
    {
        memset(g,0,sizeof(g));
        memset(sum,0,sizeof(sum));
        int p1,p2;
        if(i%2==0) p1=p2=i/2+1;
        else p1=i/2+1,p2=i/2+2;
        fo(j,p2,n)
        {
            g[j][0]=(f[i]*(j>i)+g[j-1][1])%mo;
            if(j>i) sum[j]=sum[j-1]; 
            else 
            {
                if(p1==p2) sum[j]=(sum[j-1]+f[p1-2]*f[p2-1])%mo;
                else sum[j]=(sum[j-1]+f[p1-2]*f[p2-1]+f[p1-1]*f[p2-2])%mo;
                p1--;
                p2++;
            }
            g[j][1]=(g[j-1][1]+g[j-1][0]+sum[j])%mo;
        }
        printf("%lld ",g[n][0]);
    }
}