day18动态规划5

139.单词拆分

bool wordBreak(string s, vector<string>& wordDict) {
        unordered_set<string> wordSet(wordDict.begin(),wordDict.end());
        vector<bool> dp(s.size()+1,false);
        dp[0]=true;
        for(int i=1;i<=s.size();i++){// 遍历背包
            for(int j=0;j<i;j++){// 遍历物品
                string word=s.substr(j,i-j);//substr(起始位置，截取的个数)
                if(wordSet.find(word)!=wordSet.end() && dp[j]){
                    dp[i]=true;
                }
            }
        }
        return dp[s.size()];
    }

198.打家劫舍

int rob(vector<int>& nums) {
        if (nums.size() == 0) return 0;
        if (nums.size() == 1) return nums[0];
        vector<int> dp(nums.size());
        dp[0]=nums[0];
        dp[1]=max(nums[1],nums[0]);
        for(int i=2;i<nums.size();i++){
            dp[i]=max(dp[i-2]+nums[i],dp[i-1]);
        }
        return dp[nums.size()-1];
    }

213.打家劫舍II

class Solution {// 注意注释中的情况二情况三，以及把198.打家劫舍的代码抽离出来了
public:
    // 198.打家劫舍的逻辑
    int robRange(vector<int>& nums,int start,int end) {
        // if (nums.size() == 0) return 0;
        // if (nums.size() == 1) return nums[0];
        if(end==start) return nums[start];
        vector<int> dp(nums.size());
        dp[start]=nums[start];   //dp[0]=nums[0];
        dp[start+1]=max(nums[start],nums[start+1]);  //dp[1]=max(nums[1],nums[0]);
        for(int i=start+2;i<=end;i++){
            dp[i]=max(dp[i-2]+nums[i],dp[i-1]);
        }
        return dp[end];
    }
    int rob(vector<int>& nums) {
        if(nums.size()==0) return 0;
        if(nums.size()==1) return nums[0];
        int result1=robRange(nums,0,nums.size()-2);//情况二
        int result2=robRange(nums,1,nums.size()-1);//情况三
        return max(result1,result2);
    }
};

337.打家劫舍 III

vector<int> robTree(TreeNode* cur){
        if(cur==nullptr) return vector<int>{0,0};
        vector<int> left=robTree(cur->left);//左
        vector<int> right=robTree(cur->right);//右
        //中：
        // 偷cur，那么就不能偷左右节点。
        int val1=cur->val+left[0]+right[0];
        // 不偷cur，那么可以偷也可以不偷左右节点，则取较大的情况
        int val2=max(left[0],left[1])+max(right[0],right[1]);
        return {val2,val1};
    }
    int rob(TreeNode* root) {
        vector<int> result=robTree(root);
        return max(result[0],result[1]);
    }