Bootstrap

输入两个整数值n和m,求出整数1到n之间的和为m的所有组合

输入两个整数值n和m,求出整数1到n之间的和为m的所有组合
例:
输入:n=10 m=10
输出:
1 2 3 4
1 2 7
1 3 6
1 4 5
1 9
2 3 5
2 8
3 7
4 6
10

思路:

1.输出1-n的全排列
2.限制条件:前i项等于m,则输出前i项,否则不输出
3.搜索完数值j后,不再后续搜索不再包含j(搜索完包含1的元素后,后续搜索不再搜索元素1;搜索完包含数值2的元素后不再搜索元素2)

代码如下:

#include <stdio.h>
#include <iostream>
using namespace std;
//打印结果
void print_res(int *a,int n)
{
    for(int i=1;i<n;i++)
        printf("%d ",a[i]);
    printf("\n");
}
//递归搜索
void dfs(int *data,int *index,int step,int n,int m,int sum)
{
    int i;
    if( sum > m)//不再查找
        return;
    else if(sum == m)//输出
    {
        print_res(data,step);
        return;
    }
    for(i=data[step-1]+1;i<=n;i++)//关键
    {
        if(index[i]==0)//该元素没有使用过
        {
            data[step]=i;
            index[i]=1;
            dfs(data,index,step+1,n,m,sum+i);
            index[i]=0;
        }
    }
    return;
}

int main() {
    int m,n;
    cin>>n>>m;
    int *index=new int[n+1];
    int *data=new int[n+1];
    for(int i=0;i<n+1;i++)
    {
        index[i]=0;
        data[i]=0;
    }
    dfs(data,index,1,n,m,0);
    delete [] index;
    delete [] data;
    return 0;
}

;