C++组合问题的简单实现

简述

arr= [2,3,5,6]
reference= [0,0,1,1]
若待组合序列为arr[4]，要完成C(4,2)的排序
将辅助数组reference，后2个元素置1，调用next_permutation()后0011会递增到0101,对应reference变为[0,1,0,1]
reference的第i个元素为1，表明选中arr中第i个元素。所以每种reference对应一种组合

• 其中next_permutation()是增序生成排列的函数，包含在algorithm头文件中

代码

/*
arr=        [2,3,5,6]
reference=  [0,0,1,1]
若待组合序列为arr[4]，要完成C(4,2)的排序
将辅助数组reference，后2个元素置1，调用next_permutation()后0011会递增到0101,对应reference变为[0,1,0,1]
reference的第i个元素为1，表明选中arr中第i个元素。所以每种reference对应一种组合
*/

#include<iostream>
#include<algorithm>
#include<memory.h>
using namespace std;
// 带组合的所有数
long* arr=NULL;
// 辅助数组
bool* reference=NULL;
int main(){
    int n,t;
    cin>>n>>t;
//  输入数组
    arr=new long[n];
    reference=new bool[n];
    memset(reference,0,n*sizeof(bool));
    for(int i=0;i<n;i++){
        cin>>arr[i];
    }
    for(int i=0;i<t;i++){
        reference[n-i-1]=true;
    }

    do{
        for(int i=0;i<n;i++){
        if(reference[i]){
            cout<<arr[i]<<" ";
        }
        }
        cout<<endl;
        // 对referen的n个元素排序
    }while(next_permutation(reference,reference+n));
    return 0;
}

其他

next_permulation()函数之后再研究研究