LeetCode 按摩师

题目

题目链接

思路

首先这道题是一道动态规划题目，那么我们来找一下状态转移方程；
动态规划一般要将问题分成小问题，解出来小问题后再利用小问题的解去解大问题；
那么这道题的子问题就是将数据量变小，先解只有一个元素，然后两个，三个.....;
对于每个子问题，考虑末尾元素只有取和不取两种可能，如果取末尾元素，那么前一个元素必然不能取；
如果不取末尾元素，那么前一个元素可以取，也可以不取；

代码

class Solution {
public:
    int massage(vector<int>& nums) {
    //dp0用来记录前一个子问题的末尾元素不取时的最优解；
    //dp1用来记录前一个子问题的末尾元素取时的最优解;
        int ans = 0, dp0 = 0, dp1 = 0, size = nums.size();
        if(size == 0) return 0;
        dp1 = nums[0];
        for(int i = 1; i < nums.size(); i++) {
        //需要用临时变量来存储上一状态的数值
            int temp1 = 0, temp0 = 0;
            temp1 = dp0 + nums[i];
            temp0 = max(dp1, dp0);
            dp1 = temp1;
            dp0 = temp0;
        }
        return max(dp0, dp1);
    }
};