2021-dfsTRN-K

最新推荐文章于 2021-03-06 17:58:57 发布

原创最新推荐文章于 2021-03-06 17:58:57 发布 · 140 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种解决素数环问题的方法，通过DFS算法找出所有符合条件的素数环，并按字典序输出。代码实现中使用了素数筛选、状态标记等技巧。

https://vjudge.net/contest/425321#problem/K

题目概述

素数环问题。要按照字典序输出素数环。
（吐槽一下，那个题目中的顺时针逆时针太误导人了，和那毫无关系）

思路

可能我还不太熟悉输出所有情况的题，导致做的有点卡。。不过不要紧。
用数组记录已经dfs假设的答案，如果答案确实符合条件，就顺势在dfs内输出并回溯。
最后要注意输出行末没有空格。PE了一下

代码

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <math.h>

using namespace std;
int n,prime[50],used[50],ans[50],vis[50];

void primelist()
{
    int i,j;
    for(i=2;i<=50;i++)
    {
        if(!used[i])
        {
            used[i]=1;
            prime[i]=1;
            for(j=i;j<=50;j+=i)
            {
                used[j]=1;
            }
        }
    }
}

void dfs(int step)
{
    if(step==n&&prime[1+ans[n]])
    {
        int i;
        printf("1");///注意输出格式
        for(i=2;i<=n;i++)
        {
            printf(" %d",ans[i]);///在递归函数内部合适的地方输出结果
        }
        printf("\n");
    }
    if(step==n) return;
    int i;
    for(i=1;i<=n;i++)
    {
        if(!vis[i]&&prime[i+ans[step]])
        {
            vis[i]=1;
            ans[step+1]=i;
            dfs(step+1);
            vis[i]=0;
        }
    }
}
int main()
{
    int cas=0;
    primelist();
    while(~scanf("%d",&n))
    {
        memset(vis,0,sizeof(vis));
        cas++;
        printf("Case %d:\n",cas);
        ans[1]=1,vis[1]=1;
        dfs(1);
        printf("\n");
    }
    return 0;
}