HDU 1016 Prime Ring Problem（dfs）

最新推荐文章于 2019-05-22 21:04:01 发布

Infinity_Izayoi

最新推荐文章于 2019-05-22 21:04:01 发布

阅读量304

点赞数

CC 4.0 BY-SA版权

分类专栏： ACM之路搜索

本文链接：https://blog.youkuaiyun.com/Infinity_Izayoi/article/details/54428276

ACM之路同时被 2 个专栏收录

86 篇文章

订阅专栏

搜索

4 篇文章

订阅专栏

本文介绍了一道经典的算法题目，使用深度优先搜索（DFS）寻找所有满足相邻两数之和为素数的数环组合，并给出了完整的AC代码实现。

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

题目链接：

http://acm.hdu.edu.cn/showproblem.php?pid=1016

题意：

输出所有满足相邻两个数的和是素数的环。

题解：

运用dfs，从1开始搜索，只要满足和上一个数互质就存到数组里，终止条件为搜索了n-1次并且最后一个数加1是质数时输出，输出数组里所有数字。

AC代码：

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
using namespace std;

int ans[20 + 10];
int n;

bool not_contain(int num, int depth)
{
    for(int i = 0; i <= depth; i++)
    {
        if(ans[i] == num)return false;
    }
    return true;
}

bool is_prime(int num)
{
    int i = 2;
    while(i <= num/2)
    {
        if(num % i == 0)
        {
            return false;
        }
        i++;
    }
    return true;
}

void dfs(int depth)
{
    if(depth >= 0 && depth == n-1 && is_prime(ans[depth] + ans[0]))
    {
        for(int i = 0; i <= depth; i++)
        {
            if(i == depth)
            {
                printf("%d\n", ans[i]);
            }
            else{
                printf("%d ", ans[i]);
            }
        }
    }
    for(int i = 1; i <= n; i++)
    {
        if(is_prime(ans[depth] + i) && not_contain(i,depth))
        {
            depth++;
            ans[depth] = i;
            dfs(depth);
            ans[depth] = 0;
            depth--;
        }
    }
}

int main()
{
    int cnt = 1;

    while(~scanf("%d", &n))
    {
        ans[0] = 1;
        printf("Case %d:\n", cnt++);
        dfs(0);
        cout << endl;
    }
    return 0;
}