长度为nNum的整数数组pGroups,计算从pGroups中选取nSelNum个元素的组合,放到pRes中
不同组合个数TotalNum由函数CalCombNum(a,b)计算,pRes长度为TotalNum*nSelNum
void CalCombination(int *pGroups, int nNum, int nSelNum,int *pRes)
{
nSelNum=(nSelNum>nNum) ? nNum:nSelNum;
int* pTemp = new int[nSelNum+1];
for(int i=0; i<=nSelNum; i++)
pTemp[i] = i-1;//这里pTemp[0]=-1用来作为循环判断标识
int k = nSelNum;
bool flag = true;// 标志找到一个有效组合
while(pTemp[0] == -1)
{
if(flag)//输出符合要求的组合
{
for(int i=1; i<=nSelNum; i++)
{
*pRes=pGroups[pTemp[i]];
pRes++;
}
flag = false;
}
pTemp[k]++;//在当前位置选择新的数字
if(pTemp[k] == nNum)// 当前位置已无数字可选,回溯
{
pTemp[k--] = 0;
continue;
}
if(k < nSelNum)// 更新当前位置的下一位置的数字
{
pTemp[++k] = pTemp[k-1];
continue;
}
if(k == nSelNum)
flag = true;
}
delete[] pTemp;
}
计算阶乘
int CalFactorial(int nNum)
{
if(nNum<=0) return 1;
int nRes=1;
for(int i=1;i<=nNum;i++)
nRes*=i;
return nRes;
}
计算从a个元素中选取b个元素的组合种类:
int CalCombNum(int a, int b)
{
return CalFactorial(a)/(CalFactorial(b)*CalFactorial(a-b));
}