动态划分数组

动态划分数组

问题描述：给定一个正整数的集合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

用一个二维数组dp[ i ][ j ]存储前i个数中有几组相加等于j（j<=sum/2)，相当于i表示输入的数据，j表示mid的值，a [ ]用于存储输入的数据，从a[1]开始存储，默认赋值dp[0][j]=0 ，dp[i][0]=1，dp[0][0]=1。

关键算法 ：如果 j<a[i] dp[i][j]=dp[i-1][j]+dp[i-1][j-a[i]]（第i个数加入集合），
如果 j>=a[i] dp[i][j]=dp[i-1][j]（第i个数不加入集合）

特殊情况：如果a[i]=0,dp[i][j]=dp[i-1][j].

找同组数：反向搜索dp[i][mid],如果dp[i][mid]!=dp[i-1][mid]时对应的a[i]可以存在于两个相等集合中的一个集合中，找第二个数是只需将mid的值变化为mid-a[i]再用类似方法查找即可。

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cmath>
using namespace std;
int main()
{
    int a[301],n,m=0,sum=0,mid,k=0,tap=0;
    cin>>n;
    a[0]=0;
    for(int i=1; i<=n; i++)
    {
        cin>>a[i];
        sum+=a[i];
        if(a[i]==0)
            m++;
    }
    if(m==n)
    {
        cout<<"0"<<endl;
        return 0;
    }
    if(sum%2!=0)
    {
        cout<<"no"<<endl;
        return 0;
    }
    int **dp;
    dp=(int **)malloc(sizeof(int *)*n);
    mid = sum/2;
    for(int i = 0; i <=n ; i++)
    {
        dp[i]=(int *)malloc(sizeof(int)*mid);
    }
    for(int j=1; j<=mid; j++)
        dp[0][j]=0;
    for(int i=1; i<=n; i++)
        dp[i][0]=1;
    dp[0][0]=1;
    for(int i=1; i<=n; i++)
        for(int j=1; j<=mid; j++)
        {
            if(j<a[i])
                dp[i][j]=dp[i-1][j];
            if(j>=a[i])
            {
                if(j-a[i]<0)
                    dp[i-1][j-a[i]]=0;
                if(a[i]==0)
                    dp[i][j]=dp[i-1][j];
                if(a[i]!=0)
                    dp[i][j]=dp[i-1][j]+dp[i-1][j-a[i]];
            }
        }
    for(int i=0;i<=mid;i++)
        cout<<i<<" ";
    cout<<endl;
    for(int i=0;i<=n;i++)
    {
        for(int j=0;j<=mid;j++)
            cout<<dp[i][j]<<" ";
        cout<<endl;
    }
    for(int i=n; i>0; i--)
    {
        if(k==0)
        {

            if((dp[i][mid]!=dp[i-1][mid])&&(mid>=0))
            {
                cout<<a[i];
                mid-=a[i];
                k++;
                tap=1;
            }
        }
        else
        {
            if((dp[i][mid]!=dp[i-1][mid])&&(mid>=0))
            {
                cout<<" "<<a[i];
                mid-=a[i];
            }
        }
    }
    if(tap==0)
        cout<<"no"<<endl;
    return 0;
}

反思：由于没有考虑输入样例后满足sum%2!=0的情况后又不存在划分为两个相等的数组情况造成提交错误，如 4 1 5 8 10，加入了tap变量判断。