題目要求:
分析:
這是一個0,1揹包問題。(更多揹包問題,請看這裏:傳送門:經典算法總結——揹包問題(java實現)【已完結】)
這裏要注意,當sum爲奇數的時候,就直接不用做了,因爲sum/2是一個小數。
具體代碼如下:
class Solution {
public boolean canPartition(int[] nums) {
int len = nums.length;
int sum = 0;
for(int i = 0; i < len; i ++) {
sum += nums[i];
}
boolean[][] dp = new boolean[len + 1][sum / 2 + 1];
if(sum % 2 != 0) {
return false;
}
//這裏也可以將初始條件設置爲dp[0][0] = 0,接下來等於多少計算機就會自己推算了
for (int i = 1; i < sum / 2 + 1; i++) {
if (nums[0] == i) {
dp[0][i] = true;
}
}
for(int i = 1; i < len; i ++) {
for(int j = 0; j < sum / 2 + 1; j ++) {
if(nums[i] > j) {
dp[i][j] = dp[i - 1][j];
} else {
dp[i][j] = dp[i - 1][j - nums[i]] || dp[i - 1][j];
}
}
}
return dp[len - 1][sum / 2];
}
}
上面的方法是看別人的,我覺得其實按照我的智商,是想不到直接new一個boolean的dp是怎麼去做的,所以我又寫了一個按照自己這種菜雞程度才能想到的代碼,具體如下:
class Solution {
public boolean canPartition(int[] nums) {
int len = nums.length;
int sum = 0;
for(int i = 0; i < len; i ++) {
sum += nums[i];
}
if(sum % 2 != 0)
return false;
int target = sum / 2;
int[][] dp = new int[len + 1][target + 1];
for(int i = 0; i <= target; i ++) {
dp[0][i] = 0;
}
for(int i = 1; i <= len; i ++) {
for(int j = 0; j <= target; j ++) {
if(j < nums[i - 1]) {
dp[i][j] = dp[i - 1][j];
} else {
dp[i][j] = Math.max(dp[i - 1][j], dp[i - 1][j - nums[i-1]] + nums[i - 1]);
}
}
}
if(dp[len][target] == target) {
return true;
}
return false;
}
}
性能是不太好,但是勝在自己能想到。