第一章动态规划(八)

状态机模型

状态机:一系列有顺序的事件。
比如买入、卖出作为有序的两个动作,构成了一个事件。

例题:大盗阿福
原题链接
【题目描述】

阿福是一名经验丰富的大盗。趁着月黑风高,阿福打算今晚洗劫一条街上的店铺。

这条街上一共有 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
在这里插入图片描述
状态机分析法的过程:

整个分析过程相当于图论建模

  1. 首先设计状态,把几个状态画出来,作为节点
  2. 然后我们考虑这几个节点之间有什么逻辑关系
    并依据此给它们之间加上有向边(还有自环)
  3. 之后给边加上权值
  4. 最后根据每个状态的入度写出状态转移方程
    在这里插入图片描述
    在这里插入图片描述

边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);
    }
}
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值