算法刷题day51:状态机模型

引言

关于这个状态机模型,其实感觉是动态规划中比较简单的一种了,只要把问题中的各个状态找出来,并且分析出这些状态之间的关系和代价,其实代码就是照着这个关系图抄而已,就是这么的简单。所以主要在于分析,这是最重要的,分析也不是你自己一个人想出来的,也是要通过训练你在这方面的能力才行,不可能你什么都没学就能凭空想出来一个东西,那么你刚出生话都不会说字也不认识,能想的到吗,所以还是一个记忆的过程,可以说是一个训练思维的过程,你能想出来说明你之前做过这种题,你记住了这种思维,所以你能做出来,所以开始训练吧。


一、没有上司的舞会

标签:动态规划、树形DP

思路:本质上是一个树形 D P DP DP + 状态机 D P DP DP ,状态定义为 f [ i ] [ j ] f[i][j] f[i][j] 代表以 i i i 为根结点的子树去或者不去的最大值。本题中每个人有两个状态,去或者不去,如果去的话,那就是它的下属不去的最大值之和,然后去求每个下属不去的最大值,如果不去那就是每个下属去或者不去取最大值之和,我们可以用递归的思路,先把子结点计算出来,然后用子结点计算父结点,即可,可以得出这样的式子: f [ u ] [ 0 ] = ∑ m a x ( f [ j ] [ 0 ] , f [ j ] [ 1 ] ) f[u][0] = \sum max(f[j][0], f[j][1]) f[u][0]=max(f[j][0],f[j][1]) f [ u ] [ 1 ] = ∑ f [ j ] [ 0 ] f[u][1] = \sum f[j][0] f[u][1]=f[j][0] ,然后这样把每个结点的状态推导出来,然后最终的结果就是 m a x ( f [ r o o t ] [ 0 ] , f [ r o o t ] [ 1 ] ) max(f[root][0], f[root][1]) max(f[root][0],f[root][1])

题目描述:

Ural 大学有 N 名职员,编号为 1∼N。

他们的关系就像一棵以校长为根的树,父节点就是子节点的直接上司。

每个职员有一个快乐指数,用整数 Hi 给出,其中 1≤i≤N。

现在要召开一场周年庆宴会,不过,没有职员愿意和直接上司一起参会。

在满足这个条件的前提下,主办方希望邀请一部分职员参会,使得所有参会职员的快乐指数总和最大,求这个最大值。

输入格式
第一行一个整数 N。

接下来 N 行,第 i 行表示 i 号职员的快乐指数 Hi。

接下来 N−1 行,每行输入一对整数 L,K,表示 K 是 L 的直接上司。(注意一下,后一个数是前一个数的父节点,不要搞反)。

输出格式
输出最大的快乐指数。

数据范围
1≤N≤6000,−128≤Hi≤127
输入样例:
7
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5
输出样例:
5

示例代码:

#include <bits/stdc++.h>

using namespace std;

typedef long long LL;
typedef pair<int,int> PII;
#define x first
#define y second

const int N = 6010, M = N * 2, INF = 0x3f3f3f3f;

int n, m;
int w[N];
int h[N], e[M], ne[M], idx;
bool has_fa[N];
int f[N][2];

void add(int a, int b)
{
	e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}

void dfs(int u, int fa)
{
	f[u][1] = w[u];
	
	for(int i = h[u]; i != -1; i = ne[i])
	{
		int j = e[i];
		if(j == fa) continue;
		dfs(j,u);
		
		f[u][0] += max(f[j][0], f[j][1]);
		f[u][1] += f[j][0];
	}
}

int main()
{
	ios::sync_with_stdio(0); cin.tie(0); cout.tie(0);
	
	memset(h, -1, sizeof h);
	cin >> n;
	for(int i = 1; i <= n; ++i) cin >> w[i];
	for(int i = 0; i < n - 1; ++i) 
	{
		int a, b; cin >> a >> b;
		add(b,a), add(a,b);
		has_fa[a] = true;
	}
	
	int root = 1;
	while(has_fa[root]) root++;
	
	dfs(root,-1);
	
	cout << max(f[root][0], f[root][1]) << endl;
	
	return 0;
}

二、大盗阿福

标签:DP、线性DP、状态机

思路:这道题每个店铺有两个状态要么偷,要么不偷,我们定义 f [ i ] [ 0 ] , f [ i ] [ 1 ] f[i][0],f[i][1] f[i][0],f[i][1] 代表前 i i i 家店铺(包含)偷或者不偷的最多收益,那么可以得到方程: f [ i ] [ 0 ] = m a x ( f [ i − 1 ] [ 0 ] , f [ i − 1 ] [ 1 ] ) f[i][0] = max(f[i-1][0], f[i-1][1]) f[i][0]=max(f[i1][0],f[i1][1]) f [ i ] [ 1 ] = f [ i − 1 ] [ 0 ] + w [ i ] f[i][1] = f[i-1][0] + w[i] f[i][1]=f[i1][0]+w[i] ,然后循环遍历即可。

题目描述:

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

这条街上一共有 N 家店铺,每家店中都有一些现金。

阿福事先调查得知,只有当他同时洗劫了两家相邻的店铺时,街上的报警系统才会启动,然后警察就会蜂拥而至。

作为一向谨慎作案的大盗,阿福不愿意冒着被警察追捕的风险行窃。

他想知道,在不惊动警察的情况下,他今晚最多可以得到多少现金?

输入格式
输入的第一行是一个整数 T,表示一共有 T 组数据。

接下来的每组数据,第一行是一个整数 N ,表示一共有 N 家店铺。

第二行是 N 个被空格分开的正整数,表示每一家店铺中的现金数量。

每家店铺中的现金数量均不超过1000。

输出格式
对于每组数据,输出一行。

该行包含一个整数,表示阿福在不惊动警察的情况下可以得到的现金数量。

数据范围
1≤T≤50,1≤N≤105
输入样例:
2
3
1 8 2
4
10 7 6 14
输出样例:
8
24
样例解释
对于第一组样例,阿福选择第2家店铺行窃,获得的现金数量为8。

对于第二组样例,阿福选择第1和4家店铺行窃,获得的现金数量为10+14=24。

示例代码:

#include <bits/stdc++.h>

using namespace std;

typedef long long LL;
typedef pair<int,int> PII;
#define x first
#define y second

const int N = 1e5+10, M = N, INF = 0x3f3f3f3f;

int n, m;
int w[N], f[N][2];

int main()
{
	ios::sync_with_stdio(0); cin.tie(0); cout.tie(0);
	
	int T; cin >> T;
	while(T--)
	{
		cin >> n;
		for(int i = 1; i <= n; ++i) cin >> w[i];
		for(int i = 1; i <= n; ++i)
		{
			f[i][0] = max(f[i-1][0], f[i-1][1]);
			f[i][1] = f[i-1][0] + w[i];
		}
		cout << max(f[n][0], f[n][1]) << endl;
	}
	
	return 0;
}

三、股票买卖 IV

标签:DP、线性DP、状态机

思路:自己有两种状态,有股票和没股票的状态,定义状态 f [ i ] [ j ] [ k ] f[i][j][k] f[i][j][k] 代表前 i i i 天,已经交易了 j j j 次,并且当前状态是 k k k ,那么我们可以分析出来如下图的状态机,然后根绝分析编写代码即可。具体的也可以参考之前写过的 状态机模型
在这里插入图片描述

题目描述:

给定一个长度为 N 的数组,数组中的第 i 个数字表示一个给定股票在第 i 天的价格。

设计一个算法来计算你所能获取的最大利润,你最多可以完成 k 笔交易。

注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。一次买入卖出合为一笔交易。

输入格式
第一行包含整数 N 和 k,表示数组的长度以及你可以完成的最大交易笔数。

第二行包含 N 个不超过 10000 的非负整数,表示完整的数组。

输出格式
输出一个整数,表示最大利润。

数据范围
1≤N≤105,1≤k≤100
输入样例1:
3 2
2 4 1
输出样例1:
2
输入样例2:
6 2
3 2 6 5 0 3
输出样例2:
7
样例解释
样例1:在第 1 天 (股票价格 = 2) 的时候买入,在第 2 天 (股票价格 = 4) 的时候卖出,这笔交易所能获得利润 = 4-2 = 2 。

样例2:在第 2 天 (股票价格 = 2) 的时候买入,在第 3 天 (股票价格 = 6) 的时候卖出, 这笔交易所能获得利润 = 6-2 = 4 。随后,
在第 5 天 (股票价格 = 0) 的时候买入,在第 6 天 (股票价格 = 3) 的时候卖出, 这笔交易所能获得利润 = 3-0 = 3 。共计利润
 4+3 = 7.

示例代码:

#include <bits/stdc++.h>

using namespace std;

typedef long long LL;
typedef pair<int,int> PII;
#define x first
#define y second

const int N = 1e5+10, M = 110, INF = 0x3f3f3f3f;

int n, m;
int w[N], f[N][M][2];

int main()
{
	ios::sync_with_stdio(0); cin.tie(0); cout.tie(0);
	
	cin >> n >> m;
	for(int i = 1; i <= n; ++i) cin >> w[i];
	
	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] + w[i]);
			f[i][j][1] = 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 = max(res, f[n][i][0]);
	cout << res << endl;
	
	return 0;
}

四、股票买卖 V

标签:DP、线性DP、状态机

思路:我们可以根据题目画出如下的状态机,然后根据其编写代码即可。
在这里插入图片描述

题目描述:

给定一个长度为 N 的数组,数组中的第 i 个数字表示一个给定股票在第 i 天的价格。

设计一个算法计算出最大利润。在满足以下约束条件下,你可以尽可能地完成更多的交易(多次买卖一支股票):

你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。卖出股票后,你无法在第二天买入股票 (即冷冻期为 1 天)。

输入格式
第一行包含整数 N,表示数组长度。

第二行包含 N 个不超过 10000 的正整数,表示完整的数组。

输出格式
输出一个整数,表示最大利润。

数据范围
1≤N≤105
输入样例:
5
1 2 3 0 2
输出样例:
3
样例解释
对应的交易状态为: [买入, 卖出, 冷冻期, 买入, 卖出],第一笔交易可得利润 2-1 = 1,第二笔交易可得利润 2-0 = 2,共得利润 1+2 = 3。

示例代码:

#include <bits/stdc++.h>

using namespace std;

typedef long long LL;
typedef pair<int,int> PII;
#define x first
#define y second

const int N = 1e5+10, M = N, INF = 0x3f3f3f3f;

int n, m;
int w[N], f[N][3];  // 无二 有 无一 

int main()
{
	ios::sync_with_stdio(0); cin.tie(0); cout.tie(0);
	
	cin >> n;
	for(int i = 1; i <= n; ++i) cin >> w[i];
	
	f[0][1] = -INF;  // 非法状态 
	for(int i = 1; i <= n; ++i)
	{
		f[i][0] = max(f[i-1][0], f[i-1][2]);
		f[i][1] = max(f[i-1][1], f[i-1][0] - w[i]);
		f[i][2] = f[i-1][1] + w[i];
	}
	
	cout << max(f[n][0], f[n][2]) << endl;
	
	return 0;
}
  • 22
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

lijiachang030718

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值