划分问题——动态规划

划分问题——动态规划

思路以及图片来源

Description
给定一个正整数的集合A={a1,a2,….,an},是否可以将其分割成两个子集合，使两个子集合的数加起来的和相等。例A = { 1, 3, 8, 4, 10} 可以分割：{1, 8, 4} 及 {3, 10}
Input
第一行集合元素个数n　　n <=300 第二行n个整数
Output
如果能划分成两个集合，输出任意一个子集，否则输出“no”
Sample Input
5
1 3 8 4 10
Sample Output
3 10

主要是建一个二维表格**t[i][j]**来存储相关信息。例如要求集合A={3,4,8,1,10}的解，就要构建如下表格

表格含义
i1-i5 分别对应集合A中的五个元素，横坐标j从0到i1-i5求和除2.t[i][j]含义是集合A1-Ai元素（我们假设集合A下标从1开始）所有子集中有没有子集和是j的，有的话为true，没有就是false。例如：**t[4][11]**意思是集合A1-A4(即3、4、8、1)所有子集中有没有子集和为11的，如图所示，子集{3，8}和为11，所以t[4][11]为true.

---

因此我们要求出**t[i][sum/2]**为真就满足条件，因为能找出构成一半和，剩下的肯定能构成另外一半的和。
动态规划的子问题之间都是有联系的，我么表格第i层的值都是借助第i-1层的值求出的（每一层都借助上一层的值）。
表格规律每一层的第一个元素（即第0列）一定为真，因为空集是任何集合的子集。还有第i层元素本身那一列也为真。再就是当集合中多余两个元素时，我们只要在上一层的所有为true基础上加上元素本身值即可（但不要越界）。例如第2层t[2][4]和t[2][7]都是在第一层t[1][0]和t[1][3]基础上加4得到的。

---

在一个问题是求构成元素和一半的元素有哪些？
我么从表格右下角t[5]13开始看，只有本身为true且上方为false才是我们要找的(只有这样的true才是新的，刚创建的)，所以t[4][13]是一个，此时我们减去找到的这个元素sum/2-Ai，不停重复此操作直到sum/2减到0就全部找到了。

注意：因为不知道sum/2的值为多少，所以要动态分配空间。

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

vector<int> Partition( int *arr, int sumHalf, int n ) {
    int **t;
    t = new int*[n + 5];//行个数

    for( int i = 0; i <= n; i++ ) {
        t[i] = new int[sumHalf + 5]; //每一行空间为 sumHalf+5 个
    }

    //上方都是构建二维表格，因为我们不知道sum/2的值到底时多大，所以用动态分配空间


    vector<int>v;
    t[1][0] = 1;	//我是从第1行开始赋值的，没有用第0行

    for( int i = 1; i <= sumHalf; i++ ) {
        t[1][i] = 0;
    }

    t[1][arr[0]] = 1; //将t[1][0]和t[1][arr[0]]置1，其余置0，后面行都是根据上一行来的

    for( int i = 2; i <= n; i++ ) {		//建表格
        for( int j = 0; j <= sumHalf; j++ ) {
            if( t[i][j] != 1 ) { //将上方元素赋值给下边，但不能完全复制，不能将更新后为1的再置0
                t[i][j] = t[i - 1][j];
            }

            //新增的1都是在上一层为1的基础上加的     并且   不能越界
            if( ( t[i - 1][j] == 1 ) && ( j + arr[i - 1] <= sumHalf ) ) { //在第i行基础上加新增
                t[i][j + arr[i - 1]] = 1;
            }

        }
    }

    if( t[n][sumHalf] != 1 ) { //没有找到
        for( int i = 0; i <= n; i++ ) {
            delete [] t[i];
        }

        delete t;
        return v;
    }

    for( int j = n; j >= 1 && sumHalf; j-- ) { //具体包含哪些元素
        if( t[j - 1][sumHalf] != 1 ) {
            v.push_back( arr[j - 1] ); //找到就放到容器中
            sumHalf -= arr[j - 1];
        }
    }

    return v;
}

int main() {
    int n;
    cin >> n;
    int arr[301];
    int sum = 0;

    for( int i = 0; i < n; i++ ) {
        cin >> arr[i];
        sum += arr[i];
    }

    if( sum % 2 ) { //当不能整除2时肯定无解
        cout << "no" << endl;
        return 0;
    }

    vector<int>v = Partition( arr, sum / 2, n ); //我用了vector容器存结果

    if( v.size() ) { //两种情况，找到和没找到，我们根据vector中元素个数判断，没找到就是0个
        for( int i = v.size() - 1; i >= 0; i-- ) {
            if( i == 0 ) {
                cout << v[i];
            } else {
                cout << v[i] << " ";
            }
        }
    } else {
        cout << "no";
    }

    cout << endl;
    return 0;
}