1.问题描述
一个有名的按摩师会收到源源不断的预约请求,每个预约都可以选择接或者不接。在每次预约服务之间要有休息时间,因此她不能接受相邻的预约。给定一个预约请求序列,替按摩师找到最优的预约集合(总预约时间最长),返回总的分钟数。
(1)示例1
输入:[1, 2, 3, 1]
输出:4
解释:选择一号和三号预约
(2)示例2
输入:[2, 7, 9, 3, 1]
输出:12
解释:选择一号、三号和五号预约
2.解决方法
首先应该明确为了得到总预约时间最长的集合,不会存在连着跳过3个预约,因为3个预约中的中间预约总是可以加到这个集合中。
使用分治法和动态规划的思想,我们从预约请求序列的末尾开始考虑。如果有n个预约,则第n个预约可以选择接也可以选择不接。
(1)如果不接,则我们的问题转化为从n-1个预约序列中寻求最长时间;
(2)如果接,则第n-1个一定不能接,问题转化为求n-2个预约序列中的最长时间,再加上第n个预约的时间。
(3)求得上述两种情况的结果后应当进行比较,较大的一个就是最终结果。
很明显,我们可以不断地减小问题规模,直到问题规模为1或者2,为1时,直接返回第一个预约的时间,若为2,则返回前两个预约中时间较长的一个。
因此这个问题可以通过递归或者迭代来实现。
3.代码实现
(1)递归实现
int max(int a,int b)
{
return a>b?a:b;
}
int massage(vector<int>& nums,int start,int end)
{
int n=end-start+1;
if(n==1)
return nums[0];
if(n==2)
return max(nums[0],nums[1]);
int r1=massage(nums,0,end-2)+nums[end];
int r2=massage(nums,0,end-1);
return max(r1,r2);
}
(2)迭代实现
int max(int a,int b)
{
return a>b?a:b;
}
int massage(vector<int>& nums)
{
int len=nums.size();
if(len==0)
return 0;
if(len==1)
return nums[0];
vector<int> dp(len);//dp[i]表示前i+1个预约的最长时间
dp[0]=nums[0];
dp[1]=max(nums[0],nums[1]);
for(int i=2;i<len;i++)
dp[i]=max(dp[i-2]+nums[i],dp[i-1]);
return dp[len-1];
}