【JZOJ 4921】幻魔皇

Description

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

Solution

只要细心一下，就会发现对于每个黑点，与它的距离为i的点的个数为斐波拉切数列第i项个，
所以，用这个设两个DP式，
$f_{i,j}$表示在第i层，当前点的颜色为白，距离为j的点对有对少对，
$g_{i,j}$表示在第i层，当前点的颜色为黑，距离为j的点对有对少对，
从最后一层开始做，
转移式显然，
最后加上一个预处理的优化，

复杂度：$O(2n^2)$

要卡常！！！

code

#include <cstdio>
#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--)
using namespace std;
typedef long long LL;
const int N=5010,mo=123456789;
int n,ans;
int f[2][N*2],g[2][2*N];
LL fb[N*2];
int F[N*2][N];
int min(int q,int w){return q<w?q:w;}
int main()
{
    freopen("raviel.in","r",stdin);
    freopen("raviel.out","w",stdout);
    scanf("%d",&n);
    fb[1]=1;fo(i,2,n*2+1)fb[i]=(fb[i-1]+fb[i-2])%mo;
    fo(j,1,n*2)fo(k,1,n)F[j][k]=(F[j][k-1]+fb[k]*fb[j-k-3])%mo;
    bool q=n%2;
    fod(i,n-1,1)
    {
        q=!q;
        fo(j,2,n-i)
        {
            f[q][j]=(g[!q][j]+fb[j-1])%mo;
            g[q][j]=(f[!q][j]+g[!q][j]+fb[j-2]+F[j][j-4])%mo;
        }
        f[q][n-i+1]=g[!q][n-i+1];
        g[q][n-i+1]=(f[!q][n-i+1]+g[!q][n-i+1]+fb[n-i-1]+F[n-i+1][n-i-3])%mo;
        fo(j,n-i+2,(n-i)*2)
        {
            f[q][j]=g[!q][j];
            g[q][j]=(f[!q][j]+g[!q][j]+F[j][n-i-2]-F[j][j-n+i-3])%mo;
        }
    }
    fo(i,1,n*2)printf("%lld ",((LL)f[1][i]+mo)%mo);
    printf("\n");
    return 0;
}