数组划分求划分结果的差值

最新推荐文章于 2023-05-05 22:36:00 发布

cjack23

最新推荐文章于 2023-05-05 22:36:00 发布

阅读量571

点赞数 1

分类专栏：算法文章标签： delete 扩展

算法专栏收录该内容

2 篇文章 0 订阅

订阅专栏

 
 
  
  问题描述：给定一个正整数数组a，大小为n，数组的每个元素取值于[0,1,...,K],K>0，将这个数组划分为两个集合A1和A2，使得这两部分元素和最小，S1为A1所有元素求和，S2为A2所有元素求和，即|S1-S2|最小。
 
 
 
 
  
  
  
  
   
   分析：由于数组中的元素都是正数，我们用SUM来表示数组a的所有元素的和，那么我们只要求一个集合，使得这个集合所有元素的和尽量接近SUM/2即可，我们用Q[i,j]表示用数组中任意的i个元素求和，和是否为j，如果是，那么Q[i,j]=1,否则Q[i,j]=0. 那么状态转移方程为：
  
  
  
  
   
   对于数组中的每个元素a[k]，如果j>=a[k]并且Q[i-1,j-a[k]]=1,那么Q[i,j]=1. 初始条件为Q[0,0]=1,其它都为0，最终我们只要从后（i=n,j=SUM/2）往前逐层遍历这个矩阵Q,找到第一个不为0的像，那么两个数组的差值为SUM-j-j (min|S1-S2|)即为所求.代码如下：（数组下标从1开始计算）

#include <iostream>
using namespace std;

void array_partition_solve_noBounds(int *a, int n)
{
	int i=0,j=0,k=0;
	int sum=0;
	//求和
	for(i=1; i<=n; i++)
	{
		sum += a[i];
	}

	//申请一个二维数组Q
	bool ** Q = new bool*[n+1];
	for(i=0; i<=n; i++)
	{
		Q[i] = new bool[sum/2+1];
	}
	for(i=0; i<=n; i++)
	{
		for(j=0; j<=sum/2; j++)
			Q[i][j] = false;
	}
	Q[0][0] = true;

	//给Q数组按照规矩赋值
	for(k=1; k<=n; k++)
	{
		for(i=k; i>=1; i--)//Q中，将a[k]加入K列之前的所有列，看是否能够成功
			for (j=a[k]; j<=sum/2; j++)//a[k]是否能够加入第i列，基于Q[i-1][j-a[k]]
			{
				if(Q[i-1][j-a[k]])
				{
					Q[i][j] = true;
				}
			}
	}

	//寻找满足条件的一个值
	int index_i = 0;
	int index_j = 0;
	int flag = false;
	for(j=sum/2; j>0; j--)
	{
		if(flag)
			break;
		for(i=n; i>0; i--)
		{
			if(Q[i][j])
			{
				index_i = i;
				index_j = j;
				flag = true;
			}
		}
	}

	cout<<"一个序列的和为："<<index_j<<endl;
	cout<<"两个序列的差值为："<<sum - index_j -index_j<<endl;

	bool *arry = new bool[n+1];//分别两个序列的数组
	for(i=0; i<=n; i++)
	{
		arry[i] = false;
	}
	//寻找满足条件的一个序列
	while (index_j>0)
	{
		for(k=1; k<=n; k++)
		{//在第index_j行横向查找满足条件的一个值
			if(index_j>=a[k] && Q[index_i-1][index_j-a[k]])
			{
				arry[k] = true;//在数组中记录该序列的值
				index_j -=a[k];
				index_i -=1;
				break;
			}
		}
	}

	cout<<"两个序列分别为："<<endl;
	for(i=1; i<=n; i++)
	{
		if(arry[i])
		{
			cout<<a[i]<<" ";
		}
	}
	cout<<endl;
	for(i=1; i<=n; i++)
	{
		if(!arry[i])
		{
			cout<<a[i]<<" ";
		}
	}
	cout<<endl;

	for(i=1; i<=n; i++)
	{
		delete [] Q[i];
	}
	delete [] Q;
}

int main()
{
	int a[] = {0,1,2,4,5,6,7,8};
	array_partition_solve_noBounds(a,7);
	return 0;
}