一、分割数组
- 给定一个只包含正整数的非空数组nums。请你判断是否可以将这两个数组分割成两个子集,
使得两个子集的元素和相等
二、思路分析
- 可以转化为01背包问题,只不过不是求最大价值,而是求有没有可能装满背包,
只需要把所有情况列举出来就好
三、代码实现
1、使用二维数组
public static boolean splitarray(int[] nums){
int n = nums.length;
int sum = 0;
for (int i = 0; i < n; i++){
sum += nums[i];
}
if (sum % 2 != 0) return false; //若为偶数,肯定不能等分
int target = sum / 2;
boolean dp[][] = new boolean[n][target + 1];
//dp[i][j]表示前i个数能否凑够j值
dp[0][0] = true;
if (nums[0] < target){
dp[0][nums[0]] = true;
}
for (int i = 1; i < n; i++){
for (int j = 0; j <= target; j++){
if (j - nums[i] >= 0){
//放或者不放有一个为true就行
dp[i][j] = dp[i-1][j] || dp[i-1][j-nums[i]];
} else {
//选择不放,上一个为true就为true
dp[i][j] = dp[i-1][j];
}
}
}
return dp[n-1][target];
}
2、使用一维数组
package OneDaySuanFa;
public class SplitArray {
//转为一维数组
public static boolean splitarray1(int[] nums){
int n = nums.length;
int sum = 0;
for (int i = 0; i < n; i++){
sum += nums[i];
}
if (sum % 2 != 0) return false;
int target = sum / 2;
boolean[] dp = new boolean[target+1]; //dp[i]表示是否能够凑够i,如果可以,返回true;如果不可以,返回false
dp[0] = true;
if (nums[0] < target) dp[nums[0]] = true;
for (int i = 1; i < n; i++){
for (int j = target; j >= 0; j--){
if (j >= nums[i]){
dp[j] = dp[j] || dp[j-nums[i]];
}
}
}
return dp[target];
}
public static void main(String[] args) {
int[] n1 = new int[]{1,11,5,5};
System.out.println(splitarray1(n1));
int[] n2 = new int[]{2,2,2,4};
System.out.println(splitarray1(n2));
int[] n3 = new int[]{1,3,5};
System.out.println(splitarray1(n3));
int[] n4 = new int[]{1,1,10,4,12};
System.out.println(splitarray1(n4));
}
}