hdu 5136 Yue Fei's Battle 树的dp计数

一棵二叉树的最长链长度为k，节点数不限，求所有形态数。

f[i]表示深度为i的二叉树形态数，dp[i]表示深度<=i的树的形态数，若k为偶数，从中间看，上面是一棵深度为k/2的二叉树，下面是一棵深度为k/2的二叉树，若k为奇数，中间也是一棵二叉树，分两种情况，1.中间的二叉树深度为k/2，2.中间的二叉树深度<=k/2.

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
#include<vector>
#include<set>
#include<map>
#include<queue>
#include<bitset>
#define fi first
#define se second
#define pii pair<int,int>
#define ll long long
#define inf 1<<30
#define eps 1e-8
using namespace std;
const int maxn=100005;
const ll mod=1000000007;
ll f[maxn];
ll dp[maxn];
ll sum[maxn];
ll inv[10];
ll powMod(ll a,ll b)
{
    if(b==0) return 1;
    ll ans=powMod(a,b/2);
    ans=ans*ans%mod;
    if(b&1)
        ans=ans*a%mod;
    return ans;
}
int main()
{
    int k;
    inv[2]=powMod(2,mod-2);
    inv[3]=powMod(3,mod-2);
    f[0]=1;
    dp[0]=1;
    for(int i=1;i<=50000;i++) {
        if(i-2>=0)
            f[i]=f[i-1]*dp[i-2]%mod;
        else
            f[i]=0;
        f[i]=(f[i]+f[i-1]*(f[i-1]+1)%mod*inv[2]%mod)%mod;
        dp[i]=(dp[i-1]+f[i])%mod;
    }
    while(~scanf("%d",&k) && k) {
        ll ans;
        if(k&1) {
            ans=f[k/2]*(f[k/2]+1)%mod*inv[2]%mod*dp[k/2-1]%mod;
            ans=((ans+f[k/2])%mod+f[k/2]*(f[k/2]-1)%mod)%mod;
            ans=(ans+f[k/2]*(f[k/2]-1)%mod*(f[k/2]-2)%mod*inv[2]%mod*inv[3]%mod)%mod;
        }
        else {
            ans=f[k/2]*(f[k/2]+1)%mod*inv[2]%mod;
        }
        printf("%I64d\n",ans);
    }
    return 0;
}