Description
幻魔皇拉比艾尔很喜欢斐波那契树,他想找到神奇的节点对。
所谓斐波那契树,根是一个白色节点,每个白色节点都有一个黑色节点儿子,而每个黑色节点则有一个白色和一个黑色节点儿子。神奇的节点对则是指白色节点对。
请问对于深度为n的斐波那契树,其中距离为i的神奇节点对有多少个?拉比艾尔需要你对于1<=i<=2n的所有i都求出答案。
Solution
先看到题目的关键字“斐波那契”,就很容易要往斐波那契数列的方向思考。
构建出一颗这样的树后可以找到一个神奇的规律:每一层的白色节点个数是一个斐波那契数列,只是在前面多了两项。而黑色节点个数则是正统的斐波那契数列。对于一对白色节点,可以分为两种情况。第一种:lca为一个白色节点,即两个点无分叉关系;第二种情况的lca为一个黑色节点。
对于第一种情况很明显是当前层节点个数它子树的组合。
对于第二种情况就还要枚举一个lca左边节点的深度,计算答案也基本相同。
设
fi
表示从某一个节点开始,向下走i的深度后,对应层的白色节点个数,
ti
表示
fi
的前缀和(C++选手可以直接枚举计和,Pascal选手不打容易超时),
gi
表示第i层的黑色节点个数,
si
表示
gi
的前缀和。为了方便处理,我们可以把n减1,深度从0开始。
先从第一种情况入手:
因为我们预处理出的
fi
是对于任意白色节点而言定义都相同,所以枚举出深度以后,都用合法的开始节点个数乘上从此往下走i步的白色节点数,即
fi∗tn−i
。
第二种情况稍显复杂:
首先要想到的是左右相乘,即lca左边的节点乘上右边的节点,但是这样子直接做可能会有重复,所以要研究一下斐波那契数列在这道题目的性质:
fibi=fibi−1+fibi−2
,而这道题中,这两项就意味着两个子节点,所以可以用左边的一半乘上右边的一半,就能够避免重复了。
Code
const mo=123456789;
var
f,g,s,t:array[0..5005] of int64;
n,i,j,mj,st:longint;
ans:int64;
function min(x,y:longint):longint;
begin
if x<y then exit(x);exit(y);
end;
function max(x,y:longint):longint;
begin
if x>y then exit(x);exit(y);
end;
begin
readln(n);
mj:=2*n; dec(n);
f[0]:=1;t[0]:=1;t[1]:=1;g[1]:=1;s[1]:=1;
for i:=2 to n do
begin
f[i]:=f[i-1]+f[i-2];
if f[i]>=mo then dec(f[i],mo);
t[i]:=t[i-1]+f[i];
if t[i]>=mo then dec(t[i],mo);
g[i]:=g[i-1]+g[i-2];
if g[i]>=mo then dec(g[i],mo);
s[i]:=s[i-1]+g[i];
if s[i]>=mo then dec(s[i],mo);
end;
for i:=1 to mj do //deep
begin
ans:=0;
if n-i>=0 then ans:=t[n-i]*f[i] mod mo;
st:=1;
if i>n then st:=i-n;
for j:=st to min(i-1,n) do
ans:=(ans+(f[j]*f[i-j-1] mod mo)*s[min(n-j,n-i+j)]) mod mo;
write(ans,' ');
end;
writeln;
end.