学习笔记：状态机模型

概念

状态机模型，有一点不同于我们平常意义的状态机，这里指的是每一个子问题会有多种状态可以选择的一种模型，比如股票，可以在某一时刻选择买入或者卖出。

做法

这种类型的问题一般没有什么特殊的做法。但是有一个技巧，因为子问题有多种状态，所以可以用$f_{i,k}$表示第$i$个问题的第$k$个状态。这样两个状态之间的关系就会简单明了一些。

例题

AcWing 1049

法一

这个题就是两个之间只能选一个或者不选。因为有两种状态，而且状态之间的关系不明显，考虑把状态分开观察。首先，拆成$0,1$两个状态，$0$表示不抢，$1$表示抢。首先$f_{i,0}$，那么前面的就可以选择抢或者不抢，$f_{i,0}=\max (f_{i-1,1},f_{i-1,0})$。那么$f_{i,1}$呢？首先上一个不能抢，然后这个抢了就有$a_i$的收益，$f_{i-1,1}=f_{i-1,0}+a_i$。

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

法二

考虑状态合并。假设$f_i$就是选择了抢还是不抢的最优方案。那么$f_i$有可能是$f_{i,0}$或$f_{i,1}$。首先考虑第一种，不抢。所以我们不用担心$f_{i-1}$是用的$f_{i,0}$还是$f_{i,1}$，而且$f_{i-1}$使用的是两个的最大值，相当于前面的$f_{i,0}=\max (f_{i-1,1},f_{i-1,0})$，则$f_i=f_{i-1}$。那么如果选择了抢呢？首先，前面一个有可能使用的是抢，所以我们不能用它来迭代。发现$f_{i-2}$的选择对于$f_i$的选择是没有影响的，则考虑$f_{i-2}$是不是覆盖了全部的状态。首先，原来是$f_{i-1,1}=f_{i-1,0}+a_i$，那么看看$f_{i-1,0}$是怎么迭代的。原来$f_{i-1,0}=max(f_{i-2,1},f_{i-2,0})$，刚好是新定义的$f_{i-2}$，所以新的选择抢的状态转移方程就是$f_{i-2}+a_i$，因为要选择两个方案的最大值，则状态转移方程$f_i=\max (f_{i-1},f_{i-2}+a_i)$。

#include<bits/stdc++.h>
using namespace std;
const int NN=100004;
int a[NN],f[NN];
int main()
{
	int t;
	scanf("%d",&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;
}

AcWing 1057

这个题每个时刻都有两种状态：手中没有股票和没股票。因为本题可以买多股股票，设$j$为交易了几组（一次买入卖出为一组）股票的情况。设为$f_{i,j,0}$和$f_{i,j,1}$表示$1...i$时刻最多买$j$组股票的两种状态。先考虑不买，则可以选择继续不买或者买一个，买一个就要花费当天的股价$(a_i)$，则$f_{i,j,0}=\max (f_{i-1,j,0},f_{i-1,j,1}-a_i)$。考虑手中有股票，则不能购买股票，选择继续等待或者卖出去，卖出去会获得$a_i$且使用了一次交易，则$f_{i,j,1}=\max (f_{i-1,j,1},f_{i-1,j-1,0}+a_i)$。然后我们发现，$i$只需要$i-1$的状态，且$j$只会用更小的，则可以考虑滚动数组。$j$只会用更小的，可以从大到小枚举$j$以防用到的$f$被提前更新。最后输出最大买$m$个股票且手中没有持有股票的值（因为持有股票在之前卖了肯定更划算）。需要注意的是，要把第$0$天购买的状态设为负无穷，因为第$0$天没办法买。

#include<bits/stdc++.h>
using namespace std;
int w[100004],f[104][2];
int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
        scanf("%d",&w[i]);
    for(int i=1;i<=m;i++)
        f[i][1]=-1e9;
    for(int i=1;i<=n;i++)
        for(int j=m;j>=1;j--)
        {
            f[j][0]=max(f[j][0],f[j][1]+w[i]);
            f[j][1]=max(f[j][1],f[j-1][0]-w[i]);
        }
    printf("%d",f[m][0]);
    return 0;
}

AcWing 1058

这个题目可以把上一题的思路中卖出的情况拆成两种：刚卖出、出了冷冻期。则设$f_{i,0},f_{i,1},f_{i,2}$分别表示手中持有股票、今天卖出和过了冷冻期的方案。首先考虑买入，买入必须过了冷冻期，所以不能用刚卖出的方案更新买入，$f_{i,0}=\max (f_{i-1,0},f_{i-1,2}-w_i)$。然后考虑刚卖出的方案，只能选择卖出，$f_{i,1}=f_{i-1,0}+w_i$。接着考虑出了冷冻期，则前一天要么已经出了冷冻期，要么刚卖出，则$f_{i,2}=\max (f_{i-1,1},f_{i-1,2})$。最后考虑边界，第$0$天不能买入或者是卖出，那么就不能用它们更新，设为负无穷即可。其实这个题用的全部都是$i-1$的状态，所以可以设两个数组，一个存$i-1$，一个存$i$来滚动数组。但是本题数据范围不大，滚动数组细节太多还是少用为妙。

#include<bits/stdc++.h>
using namespace std;
const int NN=100004;
int w[NN],f[NN][3];
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%d",&w[i]);
    f[0][0]=f[0][1]=-1e9;
    for(int i=1;i<=n;i++)
    {
        f[i][0]=max(f[i-1][0],f[i-1][2]-w[i]);
        f[i][1]=f[i-1][0]+w[i];
        f[i][2]=max(f[i-1][2],f[i-1][1]);
    }
    printf("%d",max(f[n][1],f[n][2]));
    return 0;
}

AcWing 1052

这个题目要求一个字符串不包含另一个字符串，我们称其为串$b$，设长度为$m$。肯定需要匹配两个字符串，则可以考虑$KMP$。首先$ne$数组是可以先求出来的，因为题目给出了用于匹配的串。现在可以考虑，每次加一个数，如果匹配的长度就不可以用该方案了。于是，我们就可以按匹配长度设计状态。设$f_{i,j}$为已经设计了$i$位可行的密码，最后有$j$位匹配串$b$的方案数。只有最后$j<m$才是可行的方案，所以最终答案是$\sum _{j=0}^{m-1}{f}_{n,j}$。考虑状态转移，首先因为$n$很小，枚举$i$和$j$，也可以枚举现在加上哪个字母。首先之前匹配的指的是串$b$的前缀（如果从中间匹配就不可能出现包含的情况，$KMP$也不会从中间匹配），所以匹配以前匹配的串加上当前枚举的新加的字母组成的串。如果匹配长度大于$m$直接跳过，否则就可以用原串加上该字母得到一个新串，所以$f_{i,newj}+=f_{i-1,j}$。

#include<bits/stdc++.h>
using namespace std;
const int NN=54,P=1e9+7;
int ne[NN],f[NN][NN];
char str[54];
int main()
{
    int n,m;
    scanf("%d%s",&n,str+1);
    m=strlen(str+1);
    int j=0;
    for(int i=2;i<=m;i++)
    {
        while(j&&str[i]!=str[j+1])
            j=ne[j];
        if(str[i]==str[j+1])
            j++;
        ne[i]=j;
    }
    f[0][0]=1;
    for(int i=1;i<=n;i++)
        for(int j=0;j<m;j++)
            for(char k='a';k<='z';k++)
            {
                int u=j;
                while(u&&k!=str[u+1])
                    u=ne[u];
                if(k==str[u+1])
                    u++;
                if(u<m)
                    (f[i][u]+=f[i-1][j])%=P;
            }
    int res=0;
    for(int i=0;i<m;i++)
        (res+=f[n][i])%=P;
    printf("%d",res);
    return 0;
}

AcWing 1053

这个题目和上一题非常像，就是匹配的串变成了多个。多个字符串匹配，可以用$AC$自动机匹配，本题把上一题的$KMP$改成$AC$自动机就行了。目前我还没有写$AC$自动机的学习笔记，可以先在$CSDN$上搜一下。

#include<bits/stdc++.h>
using namespace std;
const int NN=1004;
int ne[NN],tr[NN][4],f[NN][NN],cnt;
char s[NN];
bool num[NN];
int get(char c)
{
    if(c=='A')
        return 0;
    if(c=='T')
        return 1;
    if(c=='G')
        return 2;
    return 3;
}
void insert()
{
    int u=0,len=strlen(s+1);
    for(int i=1;i<=len;i++)
        if(tr[u][get(s[i])])
            u=tr[u][get(s[i])];
        else
            u=tr[u][get(s[i])]=++cnt;
    num[u]=true;
}
void build()
{
    queue<int>q;
    for(int i=0;i<4;i++)
        if(tr[0][i])
            q.push(tr[0][i]);
    while(q.size())
    {
        int t=q.front();
        q.pop();
        for(int i=0;i<4;i++)
        {
            int u=tr[t][i];
            if(!u)
                tr[t][i]=tr[ne[t]][i];
            else
            {
                ne[u]=tr[ne[t]][i];
                q.push(u);
                num[u]|=num[ne[u]];
            }
        }
    }
}
int main()
{
    int kase=0,n;
    while(scanf("%d",&n)&&n)
    {
        memset(num,false,sizeof(num));
        memset(tr,0,sizeof(tr));
        memset(ne,0,sizeof(ne));
        cnt=0;
        for(int i=1;i<=n;i++)
        {
            scanf("%s",s+1);
            insert();
        }
        build();
        scanf("%s",s+1);
        int len=strlen(s+1);
        memset(f,0x3f,sizeof(f));
        f[0][0]=0;
        for(int i=1;i<=len;i++)
            for(int j=0;j<=cnt;j++)
                for(int k=0;k<4;k++)
                {
                    int u=tr[j][k],t=get(s[i])!=k;
                    if(!num[u])
                        f[i][u]=min(f[i][u],f[i-1][j]+t);
                }
        int ans=1e9;
        for(int i=0;i<=cnt;i++)
            ans=min(ans,f[len][i]);
        if(ans==1e9)
            ans=-1;
        printf("Case %d: %d\n",++kase,ans);
    }
    return 0;
}