从数组分割到背包问题（2）

从数组分割到背包问题（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];
}

这个代码是不是有点眼熟呢？不错，正是《编程之美》上的解法三~

好了，接下来要详细讲解背包问题了~