力扣 -- 416. 分割等和子集(01背包问题)

力扣 -- 416. 分割等和子集(01背包问题)_第1张图片

解题步骤:

力扣 -- 416. 分割等和子集(01背包问题)_第2张图片

力扣 -- 416. 分割等和子集(01背包问题)_第3张图片

力扣 -- 416. 分割等和子集(01背包问题)_第4张图片

力扣 -- 416. 分割等和子集(01背包问题)_第5张图片

力扣 -- 416. 分割等和子集(01背包问题)_第6张图片

参考代码:

未优化代码:

class Solution {
public:
    bool canPartition(vector& nums) {
        int n=nums.size();
        int sum=0;
        for(const auto& e:nums)
        {
            sum+=e;
        }
        if(sum%2==1)
        {
            return false;
        }
        int aim=sum/2;
        //多开一行,多开一列
        vector> dp(n+1,vector(aim+1));
        //初始化
        for(size_t i=0;i<=n;i++)
        {
            dp[i][0]=true;
        }
        for(size_t i=1;i<=n;i++)
        {
            for(size_t j=1;j<=aim;j++)
            {
                dp[i][j]=dp[i-1][j]||((j>=nums[i-1])&&dp[i-1][j-nums[i-1]]);
            }
        }
        return dp[n][aim];
    }
};

优化后的代码:

class Solution {
public:
    bool canPartition(vector& nums) {
        int n=nums.size();
        int sum=0;
        for(const auto& e:nums)
        {
            sum+=e;
        }
        if(sum%2==1)
        {
            return false;
        }
        int aim=sum/2;
        //多开一个位置
        vector dp(aim+1);
        //初始化
        dp[0]=true;

        for(size_t i=1;i<=n;i++)
        {
            //一定记得要从右往左填写
            for(size_t j=aim;j>=nums[i-1];j--)
            {
                dp[j]=dp[j]||dp[j-nums[i-1]];
            }
        }
        return dp[aim];
    }
};

你学会了吗???

你可能感兴趣的:(力扣动态规划,力扣经典面试题,leetcode,算法,动态规划,c++,数据结构)