Leetcode刷题笔记题解(C++):腾讯笔试-假期(动态规划)

Leetcode刷题笔记题解(C++):腾讯笔试-假期(动态规划)_第1张图片

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

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

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

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

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

代码如下:

#include
#include
#include
using namespace std;
#define INFINITY 65536
//dp[i][0],dp[i][1], dp[i][2] 分别记录第i天 休息/锻炼/工作时累计的最小休息天数.
int main() {
	int n;
	cin >> n;
	vectorgym(n);
	vectorcompany(n);
	for (int i = 0; i < n; i++)
		cin >> company[i];
	for (int i = 0; i < n; i++)
		cin >> gym[i];

	vector>dp(n + 1, vector(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;
}

 

你可能感兴趣的:(Leetcode算法题解,c++,算法,动态规划,腾讯笔试)