Day36||1049. 最后一块石头的重量 II ● 494. 目标和 ● 474.一和零

1049. 最后一块石头的重量

这道题很像分割加和相等的子集的问题，把石头分成尽可能相同的两堆。

class Solution {
public:
    int lastStoneWeightII(vector<int>& stones) {
        vector<int> dp(15001,0);
        int sum=0;
        for(int a:stones)
        {
            sum+=a;
        }
        int target=sum/2;

        for(int i=0;i<stones.size();i++)
        {
            for(int j=target;j>=stones[i];j--)
            {
                dp[j]=max(dp[j],dp[j-stones[i]]+stones[i]);
            }
        }
        int result=0;
        result=sum-dp[target]-dp[target];
        return result;

    }
};

● 494. 目标和 

二维dp：

二维dp的初始化较为麻烦，我们需要将dp[0][0]和dp[0][nums[0]] 都设为1，这样才能得到正确结果，所以应该做滚动数组的优化更加方便！         

 class Solution {
public:
    int findTargetSumWays(vector<int>& nums, int target) {
        int sum=0;
        for(int a:nums)
        {
            sum+=a;
        }
        if(abs(target)>sum)
        return 0;
        if((target+sum)%2==1)
        return 0;
        
        int bage=(target+sum)/2;
        
    vector<vector<int> > dp(nums.size(),vector<int>(bage+1,0));
        
        if(nums[0]<=bage)
        dp[0][nums[0]]=1;
        dp[0][0]++;
        for(int i=1;i<nums.size();i++)
        {
            for(int j=0;j<=bage;j++)
            {
                if(j<nums[i])
                dp[i][j]=dp[i-1][j];
                else{
                    dp[i][j]=dp[i-1][j]+dp[i-1][j-nums[i]];
                    
                }
                cout<<dp[i][j]<<" ";
            }
            cout<<endl;
        }
        return dp[nums.size()-1][bage];

    }
};

滚动数组：

 class Solution {
public:
    int findTargetSumWays(vector<int>& nums, int target) {
        int sum=0;
        for(int a:nums)
        {
            sum+=a;
        }
        if(abs(target)>sum)
        return 0;
        if((target+sum)%2==1)
        return 0;
        
        int bage=(target+sum)/2;
        
    vector<int> dp(bage+1,0);
    dp[0]=1;
    for(int i=0;i<nums.size();i++)
    {
        for(int j=bage;j>=nums[i];j--)
        {
            dp[j]+=dp[j-nums[i]];
        }
    }
    return dp[bage];
    }
};

● 474.一和零 

这道题其实很简单，只是把背包问题单一的重量的维度改编成了两个维度。

class Solution {
public:
    int findMaxForm(vector<string>& strs, int m, int n) {
        vector<vector<int> > dp(m+1,vector(n+1,0));
        for(string str:strs)
        {
            int one=0;
            int zero=0;
            for(int i=0;i<str.size();i++)
            {
                if(str[i]=='0')
                zero ++;
                else 
                one++;
            }
            for(int i=m;i>=zero;i--)
            {
                for(int j=n;j>=one;j--)
                {
                    dp[i][j]=max(dp[i][j],dp[i-zero][j-one]+1);
                }
            }
        
        }
        return dp[m][n];

    }
};