【JZOJ 4921】 幻魔皇

Description

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

Analysis

首先你把树画出来
对于那些白点对，分两类计算
1一个白点是另一个的lca
思路：枚举深度
2两个白点的lca是一个黑点
思路：还是枚举深度，左边白点可能的个数 * 右边白点可能的个数 * 可能的子树个数累加答案

Code

#include<cstdio>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
typedef long long ll;
const ll N=5010,mo=123456789;
ll n,f[N],ans[N*2];
int main()
{
    freopen("raviel.in","r",stdin);
    freopen("raviel.out","w",stdout);
    scanf("%lld",&n);
    f[1]=f[3]=f[4]=1;
    fo(i,5,n+3) f[i]=(f[i-1]+f[i-2])%mo;
    fo(i,2,n)
        fo(j,1,n-i) (ans[i]+=f[j]*f[1+i]%mo)%=mo;
    fo(i,1,n)
        fo(j,1,n)
            (ans[i+j]+=((f[n-max(i,j)+3]-1+mo)*f[i]%mo*f[j+1]%mo))%=mo;
    fo(i,1,n*2) printf("%lld ",ans[i]);
    return 0;
}