您的位置:首页 > 其它

416. Partition Equal Subset Sum

2016-11-09 19:59 381 查看
Given a non-empty array containing only positive integers, find if the array can be partitioned into two subsets such that the sum of elements in both subsets is equal.

Note:

Each of the array element will not exceed 100.
The array size will not exceed 200.

Example 1:
Input: [1, 5, 11, 5]

Output: true

Explanation: The array can be partitioned as [1, 5, 5] and [11].


Example 2:
Input: [1, 2, 3, 5]

Output: false

Explanation: The array cannot be partitioned into equal sum subsets.


需要分成和相等的两部分,数组所有元素的和为sum,那么两部分的和就分别是sum/2
只要在数组中抽出一些元素出来使得和达到sum/2就可以了,dfs解决,为了提高搜索效率和方便剪枝可以先排序

public class Solution {
public boolean canPartition(int[] nums)
{
int sum=0;
for(int num :nums)
sum+=num;
if(sum%2!=0)
return false;
sum/=2;
Arrays.sort(nums);
for(int i=0,j=nums.length-1;i<j;i++,j--)
{
int temp=nums[i];
nums[i]=nums[j];
nums[j]=temp;
}
return dfs(0, sum, 0, nums, -1);
}

public boolean dfs(int index,int target,int sum_now,int[] nums,int last_index)
{
if(last_index>=nums.length-1||sum_now>target)
return false;
if(sum_now==target)
return true;
for(int i=last_index+1;i<nums.length;i++)
if(dfs(i, target, sum_now+nums[i], nums, i))
return true;
return false;
}
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: