从数组分割到背包问题(1)留下一个问题,时间复杂度和空间复杂度还可以进一步优化么?
从状态方程
F[i][j][k] = max(F[i-1][j][k],F[i-1][j][k-A[i]]+A[i]),其中1<=i<=2n,1<=j<=min(i,n),1<=k<=sum/2
可以看出来,F[i][j][k]只与F[i-1][][]有关,那我们是否能够去掉i那一维呢?因为F[i][j][k]只与F[i-1][][]有关,所以我们用二维数组来代替的时候应该对F[i][j][k]的“j”维进行逆序遍历。为什么?因为只有这样才能保证计算F[i][j][k]时利用的F[i-1][j][]和F[i-1][j-1][]是真正i-1这个状态的值,如果正序遍历,那么当计算F[][j][]时,F[][j-1][]已经变化,那么计算的结果就是错误的。这样,就将空间复杂度优化到了O(n/2*sum),时间复杂度没变。
好了,上代码:
int dp[50][1000];
int GetSplitArraySumImproved(int *arr, int n)
{
int sum = 0;
for (int i=1; i<=n; ++i)
sum += arr[i];
int val = sum>>1;
for (int i=1; i<=n; ++i) {
int limit = min(i, n>>1);
for (int j=limit; j>=1; --j) {
for (int k=arr[i]; k<=val; ++k) {
if (dp[j][k] < dp[j-1][k-arr[i]]+arr[i]) {
dp[j][k]=dp[j-1][k-arr[i]]+arr[i] ;
P[i][j][k] = 1;
}
}
}
}
return dp[n>>1][val];
}
这个代码是不是有点眼熟呢?不错,正是《编程之美》上的解法三~
好了,接下来要详细讲解背包问题了~