HDOJ(HDU).1016 Prime Ring Problem (DFS) [从零开始DFS(3)]

从零开始DFS

HDOJ.1342 Lotto [从零开始DFS(0)] — DFS思想与框架/双重DFS

HDOJ.1010 Tempter of the Bone [从零开始DFS(1)] —DFS四向搜索/奇偶剪枝

HDOJ(HDU).1015 Safecracker [从零开始DFS(2)] —DFS四向搜索变种

HDOJ(HDU).1016 Prime Ring Problem (DFS) [从零开始DFS(3)] —小结：做DFS题目的关注点

HDOJ(HDU).1035 Robot Motion [从零开始DFS(4)]—DFS题目练习

HDOJ(HDU).1241 Oil Deposits(DFS) [从零开始DFS(5)] —DFS八向搜索/双重for循环遍历

HDOJ(HDU).1258 Sum It Up (DFS) [从零开始DFS(6)] —DFS双重搜索/去重技巧

HDOJ(HDU).1045 Fire Net [从零开始DFS(７)]—DFS练习/check函数的思想

题意分析

给出数字n，要求将1-n的数字填成素数环，即相邻2个数字的和为素数，按字典序依次输出所有可能的组合。并且题目说过所有的组合开头均为1。

哎呀这题太熟悉了，又是填数字的题目，似曾相识的感觉。

讨论过的填数字的题目，传送门：

HDOJ(HDU).1342 Lotto [从零开始DFS(0)]

HDOJ(HDU).1015 Safecracker [从零开始DFS(2)]

如果独立完成了几道dfs的题目，就会发现：其实dfs只是工具，真正考察思维的，是什么时候进行dfs，怎样进行dfs

1.什么时候进行dfs：即递归边界。满足何种情况就不进行搜索了，或者何种情况进行一个输出，亦或是利用条件判断去掉重复的情况。

2.怎样进行dfs：是二重搜索（HDOJ.1342),还是四向搜索（HDOJ.1010),还是在数组中找遍所有的元素（HDOJ.1015）。也许以后还有八向搜索，全部搜索等等方式。

不难发现本题要求的是，两个相邻的数字和为素数，那么也就是在每次搜索的时候，都判断一下前2个数字的和是否为素数，若是的话继续进行搜索，否则终止。

需要注意的是，最后还需要判断一下，最后一个数字和第一个数字的和是否为素数，因为题目的要求是素数环嘛。否则会出现多解。

为了方便判断素数，最好在初始化的时候进行素数筛。规模在50即可（n上限是19，最大就是19+18=37）。

上代码。

代码总览

/*

    HDOJ.1016

    Author:pengwill

    Date:2017-2-5

*/

#include <iostream>

#include <cstdio>

#include <cstring>

#include <cmath>

#include <algorithm>

using namespace std;

bool visit[21],prime[51];

int b[21],n;

void init()

{

    // prime 0 & not prime 1

    for(int i = 2; i<=sqrt(50) ;++ i)

        if(prime[i] == 0){

            for(int j = 2;i*j<=50;++j)

                prime[i*j] = 1;

        }

    prime[1] = 0; visit[1] = true;b[1] = 1;

}

bool check(int depth)

{

    if(depth == n+1)//对于最后要判断首位数字的和是否为素数

        if(prime[b[1]+b[depth-1]] == 0 && prime[b[depth-2]+b[depth-1]] == 0) return true;

        else return false;

    else if(prime[b[depth-2]+b[depth-1]] == 0) return true;//若不是最后就直接判断前2个即可

    else return false;

}

void print()

{

    for(int i = 1;i<=n; ++i)

        if(i == 1) printf("%d",b[i]);

        else printf(" %d",b[i]);

    printf("\n");

}

void dfs(int depth)

{

    if(false == check(depth)) return;

    if(depth == n+1){

        //输出

        print();

        return;

    }

    for(int i = 2; i<=n ;++i){

        if(!visit[i]){

            visit[i] = 1;

            b[depth] = i;

            dfs(depth+1);

            visit[i] = 0;

        }

    }

}

int main()

{

    int t = 1;

    init();

    while(scanf("%d",&n) != EOF){

        printf("Case %d:\n",t++);

        if(1==n) printf("1\n");

        else dfs(2);//第一位是1，故从深度为2开始dfs

        printf("\n");

    }

    return 0;

}

对n为1的时候进行特判。

init函数打50规模的素数表，然后把1置为访问过。若n不为1，对深度为2进行dfs。

每次在递归调用dfs之前，首先检查一下前边2个数的和（depth-1和depth-2）是否为素数。（因为b[0]为0，当depth为2的时候也可以直接调用check函数，不用特判）。需要注意的是，当depth为n+1的时候，check需要检查两项内容：一是刚才说的前两个数的和是否为素数，二是最后一个数和第一个数的和是否为素数。这样就能保证是素数环了。

本题还有一个坑点，就是输出格式。输出可能组合的时候注意是每个数字之间有一个空格，也就是在行末尾只有一个换行符。题目还说了在每种case之后输出个空行，也就是说不是每组数据之间（原文表述是 Print a blank line after each case. 是after，不是between）。 所以最后还是有一个空行的。

此题不难，dfs活学活用才是王道啊！！