思路:动态规划的思路,第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;
}