poj 2537 Tight words

#include <iostream>            //DP
#include<cmath>
using namespace std;
int main()
{
    int k,n;
    double dp[15][105];        //dp[i][j]表示最后一位数字为i,长度为j的可能组合数
    while(cin>>k>>n)
    {
        memset(dp,0,sizeof(dp));
        for(int i=1;i<=k+1;++i)
            dp[i][1]=1;
        for(int j=2;j<=n;++j)    //长度为j
            for(int i=1;i<=k+1;++i)    //最后一位数字为i,这里1<=i<=k+1,而不是0<=i<=k,因为要考虑到dp[i-1][j-1]
                dp[i][j]=dp[i-1][j-1]+dp[i][j-1]+dp[i+1][j-1];        //状态转移方程
        double s=0;        //注意dp,s要声明为double变量
        for(int i=1;i<=k+1;++i)
            s+=dp[i][n];
        printf("%.5f\n",100*s/pow(k+1.0,n));
    }
    return 0;
}

　　

#include <iostream>        //dfs,很明显会TLE
#include <cmath>
using namespace std;
int data[105],k,n,s;
void dfs(int b)
{
    if(b==n)
    {
        s++;
        return ;
    }
    data[b+1]=data[b];
    dfs(b+1);
    if(data[b]+1<=k)
    {
        data[b+1]=data[b]+1;
        dfs(b+1);
    }
    if(data[b]-1>=0)
    {
        data[b+1]=data[b]-1;
        dfs(b+1);
    }
}
int main()
{
    while(cin>>k>>n)
    {
        s=0;
        memset(data,-1,sizeof(data));
        for(int i=0;i<=k;++i)
        {
            data[1]=i;
            dfs(1);
        }
        printf("%.5f\n",100*s/pow(k+1.0,n));
    }
    return 0;
}