POJ-1351裸DP

/*
这个DP挺有意思的~偶本来是想找个记忆化搜索的题来做的(这题也能记忆化,不过下面这样更好写一些)
抽象出模型后,应想到齿槽高度为1和4等价,2和3等价
我的DP方程中f[i][j][flag1][flag2]的j表示的是第i个齿槽的高度为j,flag1表示的是前i个齿槽中是否出现过相邻的1和4,flag2表示的是前i个齿槽中是否出现过2或3,dp[i][flag1][flag2]表示的是到第i个齿槽为止,对应状态的齿槽种数
代码还能优化,但没必要了~
转移方程见代码~
*/
#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;

int n;
long long f[20][5][2][2];

int main()
{
    memset(f,0,sizeof(f));
    f[1][1][0][0]=1;
    f[1][2][0][1]=1;
    f[1][3][0][1]=1;
    f[1][4][0][0]=1;
    for (int i=2; i<=17; i++)
    {
        f[i][4][0][0]=f[i][1][0][0]=f[i-1][1][0][0];
        f[i][4][0][1]=f[i][1][0][1]=f[i-1][1][0][1]+f[i-1][2][0][1]+f[i-1][3][0][1];
        f[i][4][1][0]=f[i][1][1][0]=f[i-1][1][1][0]+f[i-1][4][0][0]+f[i-1][4][1][0];
        f[i][4][1][1]=f[i][1][1][1]=f[i-1][1][1][1]+f[i-1][2][1][1]+f[i-1][3][1][1]+
                                    f[i-1][4][1][1]+f[i-1][4][0][1];
        f[i][3][0][1]=f[i][2][0][1]=f[i-1][1][0][0]+f[i-1][1][0][1]+f[i-1][2][0][1]+
                                    f[i-1][3][0][1]+f[i-1][4][0][0]+f[i-1][4][0][1];
        f[i][3][1][1]=f[i][2][1][1]=f[i-1][1][1][0]+f[i-1][1][1][1]+f[i-1][2][1][1]+
                                    f[i-1][3][1][1]+f[i-1][4][1][0]+f[i-1][1][1][1];
    }
    while (scanf("%d",&n),n!=-1)
    {
        printf("%d: ",n);
        printf("%I64d\n",f[n][1][1][1]+f[n][2][1][1]+f[n][3][1][1]+f[n][4][1][1]);
    }
    return 0;
}

你可能感兴趣的:(优化)