《编程之美》第2.18节:数组分割
问题:有一个无序,元素个数为2N的正整数数组,要求:如何能把这个数组分割为元素个数为N的两个数组,并使者两个子数组的和最接近?
解法:动态规划,开辟一个N*Sum/2的数组,isOK[i][v]表示是否可以找到i个数,使得它们的和是v。然后对于每一个在数组中的数,不断的更新这个数组,最后在isOK[N][j]中找出一个最大的值就是所求了。
代码:
#include<iostream>
#include<numeric>
#include<algorithm>
#include<string>
using namespace std;
int arrayCut(int *A,int N,int Sum)
{
bool **isOK=new bool*[N+1];
for(int i=0;i<N+1;i++)
isOK[i]=new bool[Sum/2+1];
for(int i=0;i<N+1;i++)
for(int j=0;j<Sum/2+1;j++)
isOK[i][j]=false;
isOK[0][0]=true;
for(int k=1;k<=2*N;k++)
{
for(int i=min(k,N);i>=1;i--)
for(int v=1;v<=Sum/2;v++)//注意此处必须从大数开始,否则有可能导致当前这个数加两遍的情况
if(v>=A[k-1] && isOK[i-1][v-A[k-1]])
isOK[i][v]=true;
}
int result=0;
for(int i=Sum/2;i>=1;i--)
{
if(isOK[N][i])
{
result=i;
break;
}
}
return result;
}
int main()
{
int A[10]={1,5,7,8,9,6,3,11,20,17};
int result=arrayCut(A,5,accumulate(A,A+10,0));
cout<<result<<endl;
system("pause");
return 0;
}
注:打印路径的代码没有写,可以参考:
http://blog.sina.com.cn/s/blog_66223402010164ux.html