算法实验：划分问题

算法实验：划分问题

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

源代码

#include <iostream>
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<math.h>
#include<malloc.h>
using namespace std;
int a[301];
int c[301];
int main()
{
    int n;
    long long int sum=0;
    memset(c,0,sizeof(c));
    scanf("%d",&n);
    for(int i=1; i<=n; i++)
    {
        scanf("%d",&a[i]);
        sum+=a[i];
    }
    if(sum%2==1)
    {
        printf("no\n");
        return 0;
    }
    int **b;
    b=(int **)malloc(sizeof(int *)*(n+1));
    for(int i=0;i<=n;i++)
    {
        b[i]=(int *)malloc(sizeof(int)*(sum/2+1));
    }
    for(int i=0;i<=n;i++)
    {
        for(int j=0;j<=sum/2;j++)
        {
            b[i][j]=0;
        }
    }
    b[0][0]=1;
    for(int i=1; i<=n; i++)
    {
        for(int j=0; j<=sum/2; j++)
        {
            if(b[i-1][j]==1)
            {
                b[i][j]=1;
                if(a[i]+j<=sum/2)
                {
                    b[i][a[i]+j]=1;
                }
            }
        }
    }
    sum=sum/2;
    int t=0;
    for(int i=1;i<=n;i++)
    {
        for(int j=1; j<=n; j++)
        {
            if(b[j][sum]==1)
            {
                c[t++]=a[j];
                sum=sum-a[j];
                break;
            }
        }
        if(sum==0)
            break;
    }
    if(b[n][sum]==0)
    {
        printf("no\n");
        return 0;
    }
    for(int i=t-1; i>0; i--)
    {
        printf("%d ",c[i]);
    }
    printf("%d\n",c[0]);
    return 0;
}