5973 Problem B 【递归入门】组合的输出

最新推荐文章于 2022-09-22 21:00:03 发布

TBACAMILLE

最新推荐文章于 2022-09-22 21:00:03 发布

阅读量361

点赞数

CC 4.0 BY-SA版权

分类专栏： ACM刷题 # DFS（深度优先搜索）

本文链接：https://blog.csdn.net/m0_37632283/article/details/104357892

ACM刷题同时被 2 个专栏收录

22 篇文章

订阅专栏

DFS（深度优先搜索）

5 篇文章

订阅专栏

探讨如何使用递归算法解决从n个整数中选取k个整数相加，其和为素数的问题。提供了三种递归方法的代码实现，包括使用递归进行全排列、基于进制的方法以及使用visited数组避免重复计算。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

问题 C: 【递归入门】组合+判断素数

时间限制 : 1.000 sec 内存限制 : 128 MB

解决: 711 提交: 711 统计

题目描述

已知 n 个整数b1,b2,…,bn

以及一个整数 k（k＜n）。

从 n 个整数中任选 k 个整数相加，可分别得到一系列的和。

例如当 n=4，k＝3，4 个整数分别为 3，7，12，19 时，可得全部的组合与它们的和为：
　　　　3＋7＋12=22　　3＋7＋19＝29　　7＋12＋19＝38　　3＋12＋19＝34。
　　现在，要求你计算出和为素数共有多少种。

例如上例，只有一种的和为素数：3＋7＋19＝29。

输入

第一行两个整数：n , k （1<=n<=20，k＜n）
第二行n个整数：x1,x2，…,xn （1<=xi<=5000000）

输出

一个整数（满足条件的方案数）。

样例输入 Copy

4 3
3 7 12 19

样例输出 Copy

一、AC代码

方法一：使用递归，与全排列不同的是每一次递归进入的for循环中的起始位都是比当前数字大的数字。

#include<cstdio>
#include<iostream>
#include<cstdlib>
#include<vector>
#include<cmath>
#include<algorithm>
using namespace std;

int n,r;
int ans[21+5];
int visited[21+5];

void Print()
{
    for(int i=0; i<r-1; i++)
        printf("%d ",ans[i]);
    printf("%d\n", ans[r-1]);
}

void dfs(int index, int count)
{
    if(count==r)
    {
        Print();
        return;
    }
    else if(count>r)
        return;

    for( int i=index+1; i<=n; i++ )
    {
            ans[count] = i;
            dfs(i,count+1);
    }
}

int main(){
    while(cin>>n>>r)
    {
        dfs(0,0);
    }
}

方法二：与方法一不同的地方在于传入dfs()中的变量只有index(代表答案数组中的下标，每次填入当前位的数字为前一数字+1，+2，...+k)，并且我们的ans[0] = 1是用来帮助for循环内形成一个统一规律的初始值。为了不重复加入使用过的数字，我们还要设置一个visited[]数组做标记

#include<cstdio>
#include<iostream>
#include<cstdlib>
#include<vector>
#include<cmath>
#include<algorithm>
using namespace std;

int n,r;
int ans[21+5];
int visited[21+5];

void init_vi()
{
    for(int i=0; i<n; i++)
        visited[i] = 0;
}

void Print()
{
    for(int i=1; i<r; i++)
        printf("%d ",ans[i]);
    printf("%d\n", ans[r]);
}

void dfs(int index)
{
    if(index==r+1)
    {
        Print();
        return;
    }
    else if(index>r+1)
        return;

    for( int i=ans[index-1]; i<=n; i++ )
    {
        if(!visited[i])
        {
            visited[i] = 1;
            ans[index] = i;
            dfs(index+1);

            visited[i] = 0;
        }
    }
}

int main(){
    while(cin>>n>>r)
    {
        init_vi();
        ans[0]=1;
        dfs(1);
    }
}

方法三：可以使用类似于进制的方法，此法暂不考虑（Big Integer 再去写）