算法练习：排列组合之组合和

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

typedef vector<int> IntArray;
 

//结果集
typedef vector<vector<int>> ResultSet; 
ResultSet gResultSet;

//原始序列中不含相同的值
void GetResultSet( const IntArray& mSrcArray, int nTarget, 
				  IntArray& mDstArray, int iStart )
{
	if ( nTarget < 0 ) return;
	if ( nTarget == 0 )
	{
		//找到一个结果
		gResultSet.push_back( mDstArray );
	}
	else
	{
		for( int i = iStart; i < mSrcArray.size(); ++i )
		{
			//后面更大的数不可能满足条件
			if ( mSrcArray[i] > nTarget ) break;

			//加入当前元素
			mDstArray.push_back( mSrcArray[i] );

			//递归处理，因为元素可以重复使用，所以从当前位置继续递归
			GetResultSet( mSrcArray, nTarget-mSrcArray[i], mDstArray, i );

			//重置
			mDstArray.pop_back();
		}
	}
}

//序列中可能有相同的元素，并且每个元素最多只能使用一次，不含重复组合
void GetResultSetEx( const IntArray& mSrcArray, int nTarget, 
				  IntArray& mDstArray, int iStart )
{
	if ( nTarget < 0 ) return;
	if ( nTarget == 0 )
	{
		//找到一个结果
		gResultSet.push_back( mDstArray );
	}
	else
	{
		for( int i = iStart; i < mSrcArray.size(); ++i )
		{
			//后面更大的数不可能满足条件
			if ( mSrcArray[i] > nTarget ) break;

			//避免结果集重复，只取第一个相同值加入结果中
			if ( i != iStart && mSrcArray[i] == mSrcArray[i-1] ) continue;

			//加入当前元素
			mDstArray.push_back( mSrcArray[i] );

			////递归处理，因为元素可以重复使用，所以从当前位置继续递归
			//GetResultSet( mSrcArray, nTarget-mSrcArray[i], mDstArray, i );

			//递归处理，因为元素不可以重复使用，所以从下一位置继续递归
			GetResultSetEx( mSrcArray, nTarget-mSrcArray[i], mDstArray, i+1 );

			//重置
			mDstArray.pop_back();
		}
	}
}


//输出结果集
void OutPutResultSet()
{
	if ( gResultSet.size() <= 20 )
	{
		for( ResultSet::iterator it = gResultSet.begin(); 
			it != gResultSet.end(); ++it )
		{
			for( IntArray::iterator itTemp = it->begin(); 
				itTemp != it->end(); ++itTemp )
			{
				cout << *itTemp << " ";
			}
			cout << endl;
		}
		
	}
	cout << "总共结果数：" << gResultSet.size() << endl;
	cout << "---------------------------------------" << endl;
}


int main()
{
	IntArray mSrcArray;
	IntArray mDstArrayTemp;
	int nTarget = 0;
	
	while( true )
	{
		//构造源数据
		int nTemp = 0;
		mSrcArray.clear();
		while( cin >> nTemp )
		{
			if ( nTemp == 0 ) break;
			mSrcArray.push_back( nTemp );
		}
		cin >> nTarget;

		//从小到大排序
		sort( mSrcArray.begin(), mSrcArray.end() );

		mDstArrayTemp.clear();
		gResultSet.clear();
		//GetResultSet( mSrcArray, nTarget, mDstArrayTemp, 0 );
		GetResultSetEx( mSrcArray, nTarget, mDstArrayTemp, 0 );

		//输出结果
		OutPutResultSet();
	}
	return 0;
}

版权声明：本文为博主原创文章，未经博主允许不得转载。