引言
关于这个状态机模型,其实感觉是动态规划中比较简单的一种了,只要把问题中的各个状态找出来,并且分析出这些状态之间的关系和代价,其实代码就是照着这个关系图抄而已,就是这么的简单。所以主要在于分析,这是最重要的,分析也不是你自己一个人想出来的,也是要通过训练你在这方面的能力才行,不可能你什么都没学就能凭空想出来一个东西,那么你刚出生话都不会说字也不认识,能想的到吗,所以还是一个记忆的过程,可以说是一个训练思维的过程,你能想出来说明你之前做过这种题,你记住了这种思维,所以你能做出来,所以开始训练吧。
一、没有上司的舞会
标签:动态规划、树形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[i−1][0],f[i−1][1]) ,
f
[
i
]
[
1
]
=
f
[
i
−
1
]
[
0
]
+
w
[
i
]
f[i][1] = f[i-1][0] + w[i]
f[i][1]=f[i−1][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;
}