编程之美2.18 数组分割

来自《编程之美》中的一道题：有一个没有排序、元素个数为2n的正整数数组，要求：把这个数组分割为元素个数为n的两个数组，并使两个数组的和最接近。
---------------------------------------------------------------------------------------------------------------------------------
本题与01背包问题有些类似，只不过最优化条件变成了：“必须挑选出n个数”与"n个数的和要最接近2n个数和的一半"。
比较理想的情况是遍历一次原始数组就能求解出答案，假设2n个数的和为sum，用数学归纳法来描述：
归纳假设：已知如何在m个数中挑选1到min(m,n)个数，并使其和小于等于sum/2；
以此对m+1进行归纳，终止条件为m=2n，然后检查n个数的所有可能的和，并找到最接近sum/2的结果即可，当然还需要用某种方法记录生成此和的n个数。

以m=3为例，我们已知3个数的所有排列组合的和，那么就有3种1个数的组合、3种2个数的组合、1种3个数的组合，我们需要把这些组合及其和保存下来，以方便将问题规模归纳扩展到m+1。

#include <iostream>
#include <vector>
#include <assert.h>
using namespace std;
typedef unsigned int UINT;
vector<UINT> arraySplit(const vector<UINT>& a)
{
    UINT sum =0;
    for(UINT i=0; i < a.size(); ++i){sum += (a[i]);}
    sum /= 2;
    
    size_t cnt = a.size()/2;
    vector<vector<UINT> > ele(sum+1,vector<UINT>());
    vector<vector<vector<UINT> > > table(cnt,ele);
    for (UINT i=0; i < cnt*2; ++i)
    {
        UINT begin = (i < cnt) ? i : cnt-1;
        for(UINT k = begin; k > 0; --k)
        {
            for (UINT j=0; j <= sum; ++j)
            {   
                if (table[k-1][j].size() == 0){continue;}
                UINT val = j+ a[i];
                if (val > sum){continue;}
                if (table[k][val].size() == 0)
                {
                    table[k][val].insert(table[k][val].end(),table[k-1][j].begin(),table[k-1][j].end()); 
                    table[k][val].push_back(a[i]);
                }
            }
        }
        if (table[0][a[i]].size() == 0){table[0][a[i]].push_back(a[i]);}
    }
    
    for(int i=sum; i >=0; --i)
    {
        if (table[cnt-1][i].size() > 0)
        {
            return table[cnt-1][i];
        }
    }
    assert(false && "error");
    return vector<UINT>();
}

void doTestArraySplit()
{
    vector<UINT>a;
    for (int i=0; i < 10; ++i)
    {
        a.push_back(UINT(double(rand())/RAND_MAX*10.0));
    }
    
    vector<UINT> result = arraySplit(a);
    assert(result.size() == a.size()/2 && "invalid cnt");
    for (size_t i=0; i < a.size(); ++i){cout<<a[i]<<",";}
    UINT sum =0;
    for(UINT i=0; i < a.size(); ++i){sum += (a[i]);}
    cout<<"sum="<<sum<<endl;
    for(size_t j=0; j < result.size(); ++j){cout<<result[j]<<",";}

    sum=0;
    for(UINT i=0; i < result.size(); ++i){sum += (result[i]);}
    cout<<"sum="<<sum<<endl;
}

void testArraySplit()
{
    srand(2);
    doTestArraySplit();
	srand(4);
    doTestArraySplit();
    srand(7);
    doTestArraySplit();
}

void main()
{
	testArraySplit();
}

测试结果：

1,5,7,8,9,6,3,11,20,17,sum=87
1,5,8,9,20,sum=43

参考文献： http://poplars.blog.163.com/blog/static/139422174201155762242/