[动态规划]线性DP集合

近期做一下简单DP找找做题的状态。
51nod 1412
题目：
平衡二叉树(AVL树），是指左右子树高度差至多为1的二叉树，并且该树的左右两个子树也均为AVL树。 现在问题来了，给定AVL树的节点个数n，求有多少种形态的AVL树恰好有n个节点(n<=2000)。
思路：
深度不深，最多可以10几层，所以可以枚举深度，设dp[i][k]表示第i个节点， 第k层。对于每一个新来的点都把它当做根节点，枚举左子树结点的个数， 因为最多深度差了1,所以有两个转移：1、dp[i][k] += dp[i - 1 - j][k - 1] * dp[j][k - 1] 2、dp[i][k] += 2 * dp[i - 1 - j][k - 2] * dp[j][k - 1]
代码：

#include <iostream>
#include <cstring>
#include <cstdio>
#include <cstring>
typedef long long ll;
using namespace std;
const int maxn = 2e3+ 7;
const int Mod = 1000000007;
ll dp[maxn][maxn];
int main()
{
    dp[0][0] = 1;
    dp[1][1] = 1;
    int n;
    scanf("%d", &n);
    for(int i = 2; i <= n; i++)
    {
        for(int k = 2; k <= 16; k++)
        {
            for(int j = 0; j < i; j++)
            {
                dp[i][k] += dp[j][k-1] * dp[i-1-j][k-1] + dp[j][k-1] * dp[i-1-j][k-2] * 2;
                dp[i][k] %= Mod;
            }
        }
    }
    ll ans = 0;
    for(int i = 1; i <= 20; i++)
        ans += dp[n][i], ans %= Mod;
    cout << ans << endl;
    return 0;
}

CodeForces - 1081C
题目： 有n个砖，m种颜色，把每个砖图成某一种颜色，有k个砖他左面的颜色跟他不同，问有几种涂颜料的方法？
思路：简单dp，dp[i][j]表示有i块砖，有j个上面描述的砖， 所以转移也很简单了：dp[i][j] += dp[i-1][j] + dp[i-1][j-1]*(m-1) 分别代表跟左面的颜色相同与不同
代码：

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
using namespace std;
const int maxn = 2000 + 7;
const int Mod = 998244353;
typedef long long ll;
ll dp[maxn][maxn];

int main()
{
    int n, m, k;
    while(~scanf("%d%d%d", &n, &m, &k))
    {
        memset(dp, 0, sizeof(dp));
        dp[1][0] = m;
        for(int i = 2; i <= n; i++)
        {
            dp[i][0] = m;
            for(int j = 1; j <= k; j++)
            {
                dp[i][j] += dp[i-1][j] + dp[i-1][j-1]*(m-1);
                dp[i][j] %= Mod;
            }
        }
        printf("%lld\n", dp[n][k]);
    }
    return 0;
}

CodeForces - 1084C
题目：

思路：设dp[i]为前i个字符的方案数，对于当前$s_i$如果是a 那么首先他可以作为一个字符继承前面的情况，在就是与前面最近的一个 b所结尾的序列组成一个新的序列，所以转移方程如代码

代码：

#include <iostream>
#include <cstring>
#include <string>
#include <cstdio>
#include <algorithm>
#include <cmath>
using namespace std;
const int maxn = 1e5 + 7;
const int Mod = 1e9 + 7;
typedef long long ll;
ll dp[maxn];
char str[maxn];
int main()
{
    while(~scanf("%s", str))
    {
        int n = strlen(str);
        memset(dp, 0, sizeof(dp));
        int last = -1;
        for(int i = 0; i < n; i++)
        {
            if(str[i] == 'a')
            {
                if(i > 0) dp[i] = dp[i-1] + 1;
                else dp[i] = 1;

                if(last != -1)
                    dp[i] += dp[last], dp[i] %= Mod;
            }
            else if(str[i] == 'b')
                last = i, dp[i] = dp[i-1];
            else
                dp[i] = dp[i-1];
        }
        printf("%lld\n", dp[n-1]);
    }
    return 0;
}