从N个数选取k个数的组合--不降原则（DFS）

从N个数中选取k个数的组合–不降原则（DFS）

原理 ：不降原则（看代码前先看一下原理吧）

举个例子：
比如说在6里面随便选5个数，那么选法都是什么呢？
瞎枚举？
12345
12346
前两个还不会弄混
然后很可能就乱了
少点数可能不会乱
但是多了就不好整了
比如说在100里随便选50个数。
1 2 3 4 5 6 7 8 9 10 11 12…

所以我们可以运用不降原则：
保证枚举的这些数是升序排列
其实真正的不降原则还可以平
比如 1 2 2 3 3 4…
但是这里要说的“不降原则”不能平哦！
对于这道题也不能平
否则就有重复数字了

拿6个里面选3个举例子
1 2 3
1 2 4
1 2 5
1 2 6
第一轮枚举完毕。
第二个数加一
1 3 ？
这个“？”应该是4，因为是升序排列
1 3 4
1 3 5
1 3 6
接着，就是这样
1 4 5
1 4 6
1 5 6
第一位是1枚举完毕
第一位是2呢?
2 3 4
2 3 5
2 3 6
2 4 5
2 4 6
2 5 6
就是这样的，枚举十分清晰，对吗？
以此类推…
3 4 5
3 4 6
3 5 6
4 5 6
然后就枚举不了了，结束。
所以说，这样就可以避免判重了。

代码

#include<iostream>
#include<cstring>

using namespace std;
int n,k;   //全局变量：从n个数的集合中选取k个数
int a[25]; //存放n个数的集合数据
int vis[25];//在dfs中记录数据是否被访问过
int re[25];//存放被选取的数字


void dfs(int step,int start)//参数step代表选取第几个数字，参数start代表从集合的第几个开始选
{
    if(step==k)//如果选够了k个就输出
    {
        for(int i=0;i<k;i++)
        {
            cout<<re[i]<<" ";
        }
        cout<<endl;
    }
    for(int i=start;i<n;i++)//不降原则的核心步骤1：从第i+1个开始选取数字(避免重选)
    {
        if(vis[i]==1)
            continue;
        vis[i]=1;
        re[step]=a[i];
        dfs(step+1,i+1);   //不降原则的核心步骤2：从第i+1个开始选取数字(避免重选)
        vis[i]=0;
    }
    return;
}

int main()
{

    while(cin>>n>>k)
    {
        memset(a,0,sizeof(a));
        memset(re,0,sizeof(re));
        memset(vis,0,sizeof(vis));
        for(int i=0;i<n;i++)
        {
            cin>>a[i];
        }
        dfs(0,0);
    }
    return 0;
}

运行结果

变形——从N个数中选取k个数求和（举一反三）

代码

#include<iostream>
#include<cstring>

using namespace std;
int n,k;   //全局变量：从n个数的集合中选取k个数
int a[25]; //存放n个数的集合数据
int vis[25];//在dfs中记录数据是否被访问过
int re[25];//存放被选取的数字


void dfs(int step,int sum,int start)//参数step代表选取第几个数字，参数sum代表从选取前step-1个数时的总数，参数start代表从集合的第几个开始选
{
    if(step==k)//如果选够了k个就输出
    {
        cout<<re[0];
        for(int i=1;i<k;i++)
        {
            cout<<'+'<<re[i];
        }
        cout<<'='<<sum<<endl;
    }
    for(int i=start;i<n;i++)//不降原则的核心步骤1：从第i+1个开始选取数字(避免重选)
    {
        if(vis[i]==1)
            continue;
        vis[i]=1;
        re[step]=a[i];
        dfs(step+1,sum+a[i],i+1);   //不降原则的核心步骤2：从第i+1个开始选取数字(避免重选)
        vis[i]=0;
    }
    return;
}

int main()
{

    while(cin>>n>>k)
    {
        memset(a,0,sizeof(a));
        memset(re,0,sizeof(re));
        memset(vis,0,sizeof(vis));
        for(int i=0;i<n;i++)
        {
            cin>>a[i];
        }
        dfs(0,0,0);
    }
    return 0;
}

运行结果

变形——从N个数中选取k个数求积（举一反三）

代码

#include<iostream>
#include<cstring>

using namespace std;
int n,k;   //全局变量：从n个数的集合中选取k个数
int a[25]; //存放n个数的集合数据
int vis[25];//在dfs中记录数据是否被访问过
int re[25];//存放被选取的数字


void dfs(int step,int sum,int start)//参数step代表选取第几个数字，参数start代表从集合的第几个开始选
{
    if(step==k)//如果选够了k个就输出
    {
        cout<<re[0];
        for(int i=1;i<k;i++)
        {
            cout<<'*'<<re[i];
        }
        cout<<'='<<sum<<endl;
    }
    for(int i=start;i<n;i++)//不降原则的核心步骤1：从第i+1个开始选取数字(避免重选)
    {
        if(vis[i]==1)
            continue;
        vis[i]=1;
        re[step]=a[i];
        dfs(step+1,sum*a[i],i+1);   //不降原则的核心步骤2：从第i+1个开始选取数字(避免重选)
        vis[i]=0;
    }
    return;
}

int main()
{

    while(cin>>n>>k)
    {
        memset(a,0,sizeof(a));
        memset(re,0,sizeof(re));
        memset(vis,0,sizeof(vis));
        for(int i=0;i<n;i++)
        {
            cin>>a[i];
        }
        dfs(0,1,0);
    }
    return 0;
}

运行结果