C代码
地址寻址版
bool canPartition(int* nums, int numsSize){
int i,j;
bool *dp = NULL;
int sum = 0;
int hsum = 0;
bool ret = false;
if(numsSize < 2) {
return false;
}
for (i = 0; i < numsSize; i++) {
sum += nums[i];
}
if ((sum & 1) != 0)
return false;
hsum = sum >> 1;
dp = (bool *)calloc(numsSize, (hsum + 1) * sizeof(bool));
if(nums[0] < hsum) {
*(dp+(0*(hsum+1))+nums[0]) = 1;
}
for(i = 1; i < numsSize; i++) {
for (j = 1; j < hsum+1; j++) {
if (nums[i] == j) {
*(dp+(i*(hsum+1))+j) = 1;
} else {
if(j > nums[i])
*(dp+(i*(hsum+1))+j) = *(dp+((i-1)*(hsum+1))+j) | *(dp+((i-1)*(hsum+1))+(j-nums[i]));
else
*(dp+(i*(hsum+1))+j) = *(dp+((i-1)*(hsum+1))+j) ;
}
}
}
ret = *(dp+(numsSize-1)*(hsum+1)+hsum);
return ret;
}
注意点
- calloc与malloc通过一次申请出来的空间,只能通过地址访问的方式模拟二维数组访问;否则直接
dp[0][0]=false
会导致空指针访问,因为dp[0]=0
,所以0[0]就是空指针访问了; - 这道题目的思想需要参考01背包问题;
i
代表遍历原始数组的每个元素。 - 最最最重要的题解是,能否分成两个数组的先决条件
1.数组的总和必须为偶数
; - 转化背包问题的关键在于如何从数组中取出
数组总数一半即sum/2的数组来
;
结果(待优化)
数组版
bool canPartition(int* nums, int numsSize){
int hsum = 0;
int i,j;
bool ret = false;
bool **dp = NULL;
if (numsSize < 2) {
return false;
}
for (i = 0; i < numsSize; i++) {
hsum += nums[i];
}
if ((hsum & 1) != 0)
return false;
hsum >>= 1;
dp = calloc(numsSize, sizeof(bool *));
for (i = 0; i < numsSize; i++) {
dp[i] = calloc(hsum+1, sizeof(bool));
}
if(nums[0] > hsum) {
} else if (nums[0] < hsum) {
dp[0][nums[0]] = true;
} else {
return true;
}
for (i = 1; i < numsSize; i++) {
for (j = 1; j < hsum+1; j++) {
if (nums[i] > j) {
dp[i][j] = dp[i-1][j];
} else if (nums[i] < j) {
dp[i][j] = dp[i-1][j] | dp[i-1][j-nums[i]];
} else {
dp[i][j] = true;
}
}
}
ret = dp[numsSize-1][hsum];
return ret;
}
结果
题目