Leetcode刷题笔记题解（C++）：腾讯笔试-假期（动态规划）_leetcode 健身计划 愉悦值 前一天健身 后天不能健身

思路：动态规划的思路，第i天与第i-1天的活动有关，以及公司营业或者 健身房营业都有关

如果当天选择休息，则上一天的 最小休息值+1即可

如果当天选择工作，则等于上一天选择休息 或者健身的最小休息值即可

如果当天选择健身，则等于上一天选择休息 或者工作的最小休息值即可

 注意的是还要 判断当天的公司以及健身房的情况

代码如下：

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
#define INFINITY 65536
//dp[i][0],dp[i][1], dp[i][2] 分别记录第i天 休息/锻炼/工作时累计的最小休息天数.
int main() {
	int n;
	cin >> n;
	vector<int>gym(n);
	vector<int>company(n);
	for (int i = 0; i < n; i++)
		cin >> company[i];
	for (int i = 0; i < n; i++)
		cin >> gym[i];
 
	vector<vector<int>>dp(n + 1, vector<int>(3, INFINITY));//初始化
	dp[0][0] = dp[0][1] = dp[0][2] = 0;//初始化
	for (int i = 1; i < n + 1; i++)
	{
		if (company[i - 1] == 1 && gym[i - 1] == 1)//如果第i天公司开门,健身房也开门
		{
			dp[i][0] = min(dp[i - 1][0], min(dp[i - 1][1], dp[i - 1][2])) + 1;//第i天可以选择休息
			dp[i][1] = min(dp[i - 1][0], dp[i - 1][2]);//第i天选择健身，则上一天只能是休息或者工作，第i天没有休息，只能在之前的选择的休息工作中取最小值
			dp[i][2] = min(dp[i - 1][0], dp[i - 1][1]);//第i天可以选择工作
		}
		else if (gym[i - 1] == 1)//如果第i天只有健身房开门
		{
			dp[i][0] = min(dp[i - 1][0], min(dp[i - 1][1], dp[i - 1][2])) + 1;//第i天可以选择休息
			dp[i][1] = min(dp[i - 1][0], dp[i - 1][2]);//第i天可以选择健身
		}
		else if (company[i - 1] == 1)//如果第i天只有公司开门
		{
			dp[i][0] = min(dp[i - 1][0], min(dp[i - 1][1], dp[i - 1][2])) + 1;//第i天可以选择休息
			dp[i][2] = min(dp[i - 1][0], dp[i - 1][1]);//第i天可以选择工作
		}
		else//如果第i天都不开门
		{
			dp[i][0] = min(dp[i - 1][0], min(dp[i - 1][1], dp[i - 1][2])) + 1;//第i天只能选择休息
		}
	}
 
	int res = min(dp[n][0], min(dp[n][1], dp[n][2]));//选出最小休息的时间
	cout << res << endl;
	return 0;
}