leetcode 416.分割等和子集

问题描述

思路

本题可以看做能否将n个物体放入容量恰好为sum/2的背包中，是一个典型的背包问题
状态转移方程为
如果当前物体（第i个）可以放得下，也就是if(sums[i]<=c)
dp[i][c]=(dp[i-1][c]||dp[i-1][c-sums[i]]);
即考虑0到i个物品能不能恰好放入容量为c的背包，就要考虑第i个物品放还是不放，如果放，那就是考虑0到i-1个物品能不能恰好放入容量为c-sums[i]的背包，，如果不放，就考虑能否将0到i-1个物品能不能恰好放入容量为c的背包。
如果当前物体（第i个）可以放不下 那么
dp[i][c]=dp[i-1][c];

代码

 	class Solution {
public:
    bool canPartition(vector<int>& nums) {
        int n=nums.size(),sum=0;
        if(n==0||n==1) return false;
        for(int i=0;i<n;i++) {
            sum+=nums[i];
        } 
        if(sum%2==1) return false;
        sum/=2;
        vector <vector <int> > dp(n,vector <int> (sum+1,0));
        for(int j=0;j<=sum;j++) {//初始化，如果第一个物体重量恰好为j，说明可以放下
            if(nums[0]==j) {
                dp[0][j]=1; 
                break;
            }
        }
        for(int i=1;i<n;i++) {
            for(int j=0;j<=sum;j++) {
                if(j-nums[i]>=0)
                    dp[i][j]=(dp[i-1][j]||dp[i-1][j-nums[i]]);
                else dp[i][j]=dp[i-1][j];
            }
        }
        if(dp[n-1][sum]==0) return false;
        return true;
    }
};