计算组合的算法

来源:互联网 发布:单片机与传感器 编辑:程序博客网 时间:2024/05/21 17:26

长度为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));}