DFS问题初探

今天刷洛谷的题，做到函数与递归这一块（P1036 选数）。真的很头疼，一直很难理解递归的代码，更不用说自己去写了， 但有些题目基本也只能用递归来写，不得不硬着头皮研究一下。

如果理解有不对，或者大神有比较好的理解递归程序的方法，欢迎在下面留言哦。谢谢。

先上一个全排列：

P1706 全排列问题

题目描述

输出自然数1到n所有不重复的排列，即n的全排列，要求所产生的任一数字序列中不允许出现重复的数字。

输入输出格式

输入格式：

n(1≤n≤9)

输出格式：

由1～n组成的所有不重复的数字序列，每行一个序列。每个数字保留5个常宽。

输入输出样例

输入样例#1： 复制

3

输出样例#1： 复制

    1    2    3
    1    3    2
    2    1    3
    2    3    1
    3    1    2
    3    2    1

#include <stdio.h>
#include <string.h>
#define maxn 10

int ar[maxn];    // 答案数组
int flag[maxn];  // 记号数组
int n;  // 输入数字

void dfs(int c)
{
    if(c == n+1) // 判断是否完成一组数（n个），用n+1与c比较而不是n是因为下面是dfs(c+1)
    {
        int i;
        for(i = 1; i <= n; i++) printf("%5d", ar[i]); // 题目要求的格式
        printf("\n");
    }
    else{                        // 如果还没到n个数
        int i;
        for(i = 1; i <= n; i++)   
        {
            if(!flag[i])         // 如果这个数还没有被取到
            {
                flag[i] = 1;     // 标记其已经被使用
                ar[c] = i;       // 注意的是，flag的下标即为我们当前判断是否已经被使用的数
                                 //字，而c代表我们正在找的第c个数
                dfs(c+1);         
                flag[i] = 0;     //反复这个步骤：找到一个没被使用的， 下一个没被使用的
                                 // ...直到最后一个数，所以全排的第一个结果就是顺序的
             }                   //然后从最后一个向前回溯，回溯的同时将原来标记已经使用的重置
        }
    }
}

int main(){
    scanf("%d", &n);
    memset(ar, 0, sizeof(ar));
    memset(flag, 0, sizeof(flag));
    dfs(1);

    return 0;
}

 

P1036 选数

题目描述

已知 nn 个整数 x_1,x_2,…,x_nx1​,x2​,…,xn​，以及11个整数kk(k<nk<n)。从nn个整数中任选kk个整数相加，可分别得到一系列的和。例如当n=4,k=3n=4,k=3,44个整数分别为3,7,12,193,7,12,19时，可得全部的组合与它们的和为：

3+7+12=223+7+12=22

3+7+19=293+7+19=29

7+12+19=387+12+19=38

3+12+19=343+12+19=34。

现在，要求你计算出和为素数共有多少种。

例如上例，只有一种的和为素数：3+7+19=293+7+19=29。

输入输出格式

输入格式：

键盘输入，格式为：

n,kn,k(1 \le n \le 20,k<n1≤n≤20,k<n)

x_1,x_2,…,x_n (1 \le x_i \le 5000000)x1​,x2​,…,xn​(1≤xi​≤5000000)

输出格式：

屏幕输出，格式为： 11个整数（满足条件的种数）。

输入输出样例

输入样例#1： 复制

4 3
3 7 12 19

输出样例#1： 复制

1

#include <stdio.h>
#include <string.h>
#include <math.h>
#define maxn 25
int in[maxn];
int ar[maxn];    
int flag[maxn];  // 记号数组
int n, k;  // 输入数字
int i;
int num = 0;
int sum = 0;


int is_prime(int n)
{
    int i;
    if(n == 0 || n == 1) return 0;
    for(i = 2; i < sqrt(n); i++)
    {
        if(n % i == 0) return 0;
    }
    return 1;
}

void dfs(int c)
{
    if(c == k+1) 
    {
        if(is_prime(sum)) num++;
    }
    else{                        
        int i;
        for(i = 1; i <= n; i++)
        {
            if(!flag[i] && i > ar[c-1])  // 这里我们只要按初始顺序排序的，不需要全排列  
            {
                flag[i] = 1;   
                sum = sum + in[i];
                ar[c] = i;       
                                 
                dfs(c+1);
                flag[i] = 0;    
                sum = sum - in[i];       
             }                   
        }
    }
}

int main(){
    scanf("%d%d", &n, &k);
    memset(ar, 0, sizeof(ar));
    memset(flag, 0, sizeof(flag));
    for(i=1; i<=n; i++)
    {
        scanf("%d", &in[i]);
    }
    dfs(1);
    printf("%d", num);

    return 0;
}