Description
幻魔皇拉比艾尔很喜欢斐波那契树,他想找到神奇的节点对。
所谓斐波那契树,根是一个白色节点,每个白色节点都有一个黑色节点儿子,而每个黑色节点则有一个白色和一个黑色节点儿子。神奇的节点对则是指白色节点对。
请问对于深度为n的斐波那契树,其中距离为i的神奇节点对有多少个?拉比艾尔需要你对于1<=i<=2n的所有i都求出答案。
Solution
只要细心一下,就会发现对于每个黑点,与它的距离为i的点的个数为斐波拉切数列第i项个,
所以,用这个设两个DP式,
fi,j
表示在第i层,当前点的颜色为白,距离为j的点对有对少对,
gi,j
表示在第i层,当前点的颜色为黑,距离为j的点对有对少对,
从最后一层开始做,
转移式显然,
最后加上一个预处理的优化,
复杂度: O(2n2)
要卡常!!!
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;
}