1. 程式人生 > >leetcode 416 分割等和子集

leetcode 416 分割等和子集

方法一:用動態規劃

class Solution {
public:
    bool canPartition(vector<int>& nums) {

    sort(nums.begin(), nums.end());//對陣列進行排序

    int sum=accumulate(nums.begin(),nums.end(),0);//利用演算法求和;
     if(sum%2)//如何和的一半不是偶數,則肯定不能劃分成兩個等和子集,返回錯誤
         return false;
     int half=sum/2;
     auto iter=max_element(nums.begin(),nums.end());//求出最大元素,如果最大元素大於和的一半,肯定不能劃分


    if(*iter>half)
        return false;
    if(*iter==half)
        return true;//有元素值剛好等於一半,立刻返回真

//接下來利用動態規劃求解所能到達的和
    vector<int>res(half+1,0);//初始化一個數組res[i]=1;表示和等於i,,res[i]=0表示不能等於i;
    res[0] = 1;//和等於0 一定能成立,空集就行

     for(size_t i=0;i<nums.size();++i)
     {
         for(int j=half;j>=nums[i];j--)
         {   
             if(res[j-nums[i]])
             {
                 res[j]=1;
                 if(j==half)
                     return true;
             } 
             
         }
     }
        return res[half];
    }
};

方法二:利用遞迴,(DFS)此方法較動態規劃要好很多;

基本思想:舉例子說明最容易理解,例如 nums=[1,5,4,3,7];

第一步:對陣列進行排序,nums=[1,3,4,5,7],求出其和的一半 half=10;對於陣列中元素要麼在這一個子集中,要麼在另外一個子集中。

第二步:例如1如果在一個子集中,相當於在[3,4,5,7]中去找一個子集和為9的再加上1這個元素,或者1在另外一個子集中,所以就相當於在[3,4,5,7] 中找一個和依舊為10 的子集,因此我們可以獲得遞迴方程式;

recur(nums,0,10)=recur(nums,1,9)||recur(nums,1,10);

class Solution {
public:
    bool canPartition(vector<int>& nums) {
    sort(nums.begin(), nums.end());
    int sum=accumulate(nums.begin(),nums.end(),0);
     if(sum%2)
         return false;
     int half=sum/2;
     auto iter=max_element(nums.begin(),nums.end());
    if(*iter>half)
        return false;
    if(*iter==half)
        return true;
    vector<int>res(half+1,0);
    res[0] = 1;
     for(size_t i=0;i<nums.size();++i)
     {
         for(int j=half;j>=nums[i];j--)
         {   
             if(res[j-nums[i]])
             {
                 res[j]=1;
                 if(j==half)
                     return true;
             } 
             
         }
     }
        return res[half];
    }
};