状态机模型
状态机:一系列有顺序的事件。
比如买入、卖出作为有序的两个动作,构成了一个事件。
例题:大盗阿福
原题链接
【题目描述】
阿福是一名经验丰富的大盗。趁着月黑风高,阿福打算今晚洗劫一条街上的店铺。
这条街上一共有 N家店铺,每家店中都有一些现金。阿福事先调查得知,只有当他同时洗劫了两家相邻的店铺时,街上的报警系统才会启动,然后警察就会蜂拥而至。
作为一向谨慎作案的大盗,阿福不愿意冒着被警察追捕的风险行窃。他想知道,在不惊动警察的情况下,他今晚最多可以得到多少现金?
【输入】
输入的第一行是一个整数T(T≤50),表示一共有T组数据。
接下来的每组数据,第一行是一个整数N(1≤N≤100,000),表示一共有N家店铺。第二行是N个被空格分开的正整数,表示每一家店铺中的现金数量。每家店铺中的现金数量均不超过1000。
【输出】
对于每组数据,输出一行。该行包含一个整数,表示阿福在不惊动警察的情况下可以得到的现金数量。
【输入样例】
2
3
1 8 2
4
10 7 6 14
【输出样例】
8
24
【提示】
对于第一组样例,阿福选择第2家店铺行窃,获得的现金数量为8。
对于第二组样例,阿福选择第1和4家店铺行窃,获得的现金数量为10+14=24。
————————————————————————————————————————————————————
(和LeetCode打家劫舍一样)
f[i] :抢劫前 i 家店铺的最大收益。
状态计算:
- 不抢第 i 家,f[i] = f[i - 1]
- 抢第 i 家,f[i] = f[i - 2] + w[i]
f[i] = 两者取 max
从上面的分析可以看出,我们用到了 i-2 层的状态(这个方法是对的,只不过想用状态机做一下),现在我们希望只用到上一层 i-1 的状态,为此我们将状态表示改为二维。
f[i][0] : 不抢第 i 家
f[i][1] : 抢第 i 家
状态计算:
f[i][0] = max(f[i-1][0], f[i-1][1]) ,此时上一家可抢可不抢
f[i][1] = f[i-1][0] + w[i] ,此时上一家不能抢
(这种将 f[i] 的状态分离开的做法就是状态机,这么做是为了更清楚的表示状态,便于分析计算)
import java.util.Scanner;
public class Main {
static int N = 100010;
static int T;
static int n;
static int[] w = new int[N];
static int[][] f = new int[N][2];
public static void main(String[] args){
Scanner sc = new Scanner(System.in);
T = sc.nextInt();
while(T-- > 0) {
n = sc.nextInt();
for(int i = 1;i <= n;i++) w[i] = sc.nextInt();
f[0][0] = 0;
f[0][1] = Integer.MIN_VALUE;
for(int i = 1;i <= n;i++) {
f[i][0] = Math.max(f[i - 1][0], f[i - 1][1]);
f[i][1] = f[i - 1][0] + w[i];
}
int res = Math.max(f[n][0], f[n][1]);
System.out.println(res);
}
sc.close();
}
}
例题:1057. 股票买卖 IV
状态机分析法的过程:
整个分析过程相当于图论建模
- 首先设计状态,把几个状态画出来,作为节点
- 然后我们考虑这几个节点之间有什么逻辑关系
并依据此给它们之间加上有向边(还有自环) - 之后给边加上权值
- 最后根据每个状态的入度写出状态转移方程
边2那里为什么是 j 而不是 j-1 呢?
因为必须严格按照先买入再卖出的顺序,且再次买入前必须卖掉手中股票,所以我们以买入的时间节点作为一次交易,边2 是卖出,所以不算一次交易。
初始化:
如果一种状态不合法,或者不希望从这个状态转移过来 ,那么就把它设成正无穷或负无穷
因为这个题要求最大值,所以把不合法的设成负无穷,也这样这个状态不可能用来更新后来的状态
例如这道题中f[i][0][1]表示,如果我们处理了0笔股票,并且我们手中居然还有票,这显然是不可能的
如果我们处理了0笔股票并且目前手里没有票,那收入就是0,即f[i][0][0] = 0
所以我们将 f 先全置成负无穷,再单独处理为0的情况。
【样例都对,TLE…】
import java.util.Scanner;
public class Main {
static int N = 100010;
static int M = 110;
static int INF = Integer.MAX_VALUE >> 1;
static int n;
static int m;
static int[] w = new int[N];
static int[][][] f = new int[N][M][2];
public static void main(String[] args){
Scanner sc = new Scanner(System.in);
n = sc.nextInt();
m = sc.nextInt();
for(int i = 1;i <= n;i++) w[i] = sc.nextInt();
sc.close();
//初始化
for(int i = 0;i < f.length;i++) {
for(int j = 0;j < f[i].length;j++) {
for(int k = 0;k < f[i][j].length;k++) {
f[i][j][k] = -INF;
}
}
}
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] = Math.max(f[i - 1][j][0], f[i - 1][j][1] + w[i]);
f[i][j][1] = Math.max(f[i - 1][j][1], f[i - 1][j - 1][0] - w[i]);
}
}
int res = 0;
for(int i = 0;i <= m;i++) {//枚举交易次数
res = Math.max(res, f[n][i][0]);//最终状态是手中无货
}
System.out.println(res);
}
}
例题:1058股票交易V
————————————————————————————————————————————————————
状态机分析清楚后,转移方程看着状态机直接写就行了
f[i][j] : 考虑前 i 天,状态是 j 的最大收益。
入口选在 2 是因为,不能是0 ,不能是1(此时处于冷却,不能买股票,与实际不符)。
import java.util.Scanner;
public class Main {
static int N = 100010;
static int INF = Integer.MAX_VALUE >> 1;
static int n;
static int[] w = new int[N];
static int[][] f = new int[N][3];
public static void main(String[] args){
Scanner sc = new Scanner(System.in);
n = sc.nextInt();
for(int i = 1;i <= n;i++) w[i] = sc.nextInt();
sc.close();
f[0][0] = -INF;
f[0][1] = -INF;
f[0][2] = 0;
for(int i = 1;i <= n;i++) {
f[i][0] = Math.max(f[i - 1][0], f[i - 1][2] - w[i]);
f[i][1] = f[i - 1][0] + w[i];
f[i][2] = Math.max(f[i - 1][1], f[i - 1][2]);
}
System.out.println(Math.max(f[n][1], f[n][2]));
}
}
例题:设计密码
输入样例1:
2
a
输出样例1:
625
输入样例2:
4
cbc
输出样例2:
456924
————————————————————————————————————————————————————
没听懂,代码也有错…以后再看
import java.util.Scanner;
public class Main {
static int N = 55;
static int mod = (int) (1e9 + 7);
static int n;
static int m;
static char[] str = new char[N];
static int[][] f = new int[N][N];
public static void main(String[] args){
Scanner sc = new Scanner(System.in);
n = sc.nextInt();
String T = sc.next();
sc.close();
T = "&" + T;//转为下标从1开始
str = T.toCharArray();
m = T.length();
int[] next = new int[N];
for(int i = 2, j = 0;i <= n;i++) {
while(j > 0 && str[i] != str[j + 1]) j = next[j];
if(str[i] == str[j + 1]) j++;
next[i] = j;
}
f[0][0] = 1;
for(int i = 0;i < n;i++) {
for(int j = 0;j < m;j++) {
for(char k = 'a';k <= 'z';k++) {
int u = j;
while(u > 0 && k != str[u + 1]) {
u = next[u];
}
if(k == str[u + 1]) u++;
if(u < m) f[i + 1][u] = (f[i + 1][u] + f[i][j]) % mod;
}
}
}
int res = 0;
for(int i = 0;i < m;i++) res = (res + f[n][i]) % mod;
System.out.println(res);
}
}