状态机——奇葩的状态表示

简述

在动态规划中，遇到有一个点有多个状态，混在一起无法表示，那么就可以把状态分开，并且构造出不同状态之间的转移关系，然后再求出状态转移方程，之后就OK了。

题目

1049. 大盗阿福

2
3
1 8 2
4
10 7 6 14

8
24

方法一：使用传统方法解

状态表示f[i]表示选择到这一个点，最多可以有多少money

状态转移：

1. 不选择这一个点，那么f[i] = f[i-1]
2. 选择这一个点，那么f[i] = w + f[i-2]（上一个点不能选）
3. 初始状态：f[1] = w1; f[2] = max(w1, w2)

#include <bits/stdc++.h>
using namespace std;
#define N 100200
int a[N];
int f[N];
int main()
{
    int T;
    cin >> T;
    while(T--)
    {
        int n;
        scanf("%d", &n);
        for(int i = 1; i<= n; i++) scanf("%d", a+i);
        f[1] = a[1];
        for(int i = 2 ; i <= n; i++){
            f[i] = max(f[i- 1], f[i - 2] + a[i]);
        }
        printf("%d\n", f[n]);
    }
    return 0;
}

方法二：使用状态机进行表示

​

#include <bits/stdc++.h>
using namespace std;
#define N 100200
int a[N];
int f[N][2];
int main()
{
    int T;
    cin >> T;
    while(T--)
    {
        int n;
        scanf("%d", &n);
        for(int i = 1; i<= n; i++) scanf("%d", a+i);
        for(int i = 1 ; i <= n; i++){
            f[i][0] = max(f[i-1][1], f[i-1][0]);
            f[i][1] = f[i - 1][0] + a[i];
        }
        printf("%d\n", max(f[n][0], f[n][1]));
    }
    return 0;
}

1057. 股票买卖 IV

3 2
2 4 1

2

6 2
3 2 6 5 0 3

7

题解

状态机定义

​

状态表示：
f[i][j][0]表示在第 i 天，进行总共进行了 k 次交易，手中无货的所有情况。
f[i][j][1]表示在第 i 天，之前总共进行了 k-1 次交易，手中有货的所有情况。
属性为max

状态转移：

对于f[i][j][0]：

1. f[i-1][j][0]没有买货
2. f[i-1][j][1]+w[i]卖了货，现在手里面没有货，已经进行了j次交易

对于f[i][j][1]：

1. f[i-1][j][1]没有卖货
2. f[i-1][j-1][0]-w[i]买了货，现在手里面有货，已经进行了j-1次交易，这一次是第 j 次

#include <bits/stdc++.h>
using namespace std;
#define N 100200
int f[N][105][2];
int a[N];
int n, m;
int main()
{
    scanf("%d%d", &n, &m);
    for(int i = 1; i <= n; i++){
        scanf("%d", a+i);
    }
    // 初值为0并不是合法值，必须要经过初始化
    memset(f, -0x3f, sizeof f);
    for(int i = 0; i <= n; i++) f[i][0][0] = 0;
  
    for(int i = 1; i <= n; i++)
    {
        for(int j = 1; j <= m; j++)
        {
            f[i][j][0] = max(f[i-1][j][0], f[i-1][j][1] + a[i]);
            f[i][j][1] = max(f[i-1][j][1], f[i-1][j-1][0] - a[i]);
        }
    }
    int ans = 0;
    for(int i = 1; i<= m; i++) ans = max({ans, f[n][i][0], f[n][i][1]});
    printf("%d", ans);
    return 0;
}

1058. 股票买卖 V

5
1 2 3 0 2

3

题解

对于这一道题目，与上一道相比，没有的次数限制，所以没有必要使用第二维。

状态机设计：

• 表示状态
• 表示转移的边（对于每一个点枚举三种操作（忽略，买入，卖出）来构成边）
• 寻找入口以及出口

​

入口是最右面的状态。

出口是中间以及最右面的状态。

#include <bits/stdc++.h>
using namespace std;
const int N = 1e5+5;
int a[N];
int f[N][3];
int main()
{
    int n;
    scanf("%d", &n);
    for(int i = 1; i <= n; i++)
    {
        scanf("%d", a+i);
    }
    // memset(f, -0x3f, sizeof f); 其实没有这样的必要
    f[0][1] = f[0][0] = -0x3f3f3f3f;
    f[0][2] = 0;
    for(int i = 1; i <= n; i++)
    {
        f[i][0] = max(f[i-1][0], f[i-1][2] - a[i]);
        f[i][1] = f[i-1][0] + a[i];
        f[i][2] = max(f[i-1][1], f[i-1][2]);
    }
    printf("%d\n", max(f[n][1], f[n][2]));
    // for(int i = 1; i <= n; i++){
    //     printf("%d\t%d\t%d\n", f[i][0], f[i][1], f[i][2]);
    // }
    return 0;
}

1052. 设计密码

2
a

625

4
cbc

456924

#include <bits/stdc++.h>
using namespace std;
#define N 55
int n, m;
char a[N];
int nxt[N];
long long f[N][N];
const int mod = 1e9 + 7;

int main()
{
    scanf("%d%s", &n, a+1);
    m = strlen(a+1);
    for(int i = 2, j = 0; i <= m; i++)
    {
        while(j && a[j+1] != a[i]) j = nxt[j];
        if(a[j+1] == a[i]) j++;
        nxt[i] = j;
    }
    f[0][0] = 1;
// BUG 注意是从当前的状态往后推，而不是求从哪里转移过来
    for(int i = 0; i < n; i++)
    {
        for(int j = 0; j < m; j++)
        {
            for(char k = 'a'; k <= 'z'; k++)
            {
                int p = j;
                while(p && a[p+1] != k) p = nxt[p];
                if(a[p+1] == k) p++;
                if(p < m) f[i+1][p] = (f[i+1][p] + f[i][j]) % mod;
            }
        }
    }
    long long ans = 0;
    for(int i = 0; i < m; i++)
    {
        ans = (ans + f[n][i]) % mod;
    }
    printf("%lld\n", ans);
    return 0;
}