HDU 1016 Prime Ring Problem

最新推荐文章于 2021-06-01 18:39:59 发布

原创最新推荐文章于 2021-06-01 18:39:59 发布 · 448 阅读

0 ·

CC 4.0 BY-SA版权

DFS 专栏收录该内容

5 篇文章

订阅专栏

本文介绍了一个使用深度优先搜索（DFS）算法解决特定环状排列问题的方法。该问题要求找到一种排列方式，使得环上任意相邻两个数字之和为素数。通过预处理素数并利用DFS进行回溯搜索，最终找到了所有符合条件的排列。

这题我用的是DFS。不多说，看代码和注释。

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

//因为n<20，所以最大的素数也就是40左右，因此用一个数组来记录45以内的素数，用于之后的判定。
bool isprime[45]={0,1,1,1,0,1,0,1,0,0,0,1,0,1,0,0,0,1,0,1,0,0,0,1,0,0,0,0,0,1,0,1,0,0,0,0,0,1,0,0,0,1,0,1,0};
int n;
bool visited[24];//用于标记是否被访问过
void dfs(int cnt,int id,int ans[])
{
    if(cnt==n+1)
    {
        printf("%d",ans[1]);
        for(int i=2;i<=n;i++)
            printf(" %d",ans[i]);
        printf("\n");
        return;
    }
    for(int i=2;i<=n;i++)
    {
        if(visited[i]||i==id)
            continue;
        visited[i]=true;
        if(cnt==n)//这里注意要区分是不是到达第n个数，如果到达因为是圆环，所以还要判定和1相加是不是素数。
        {
            int t1=ans[cnt-1]+i;
            int t2=i+1;
            if(isprime[t1]&&isprime[t2])
            {
                ans[cnt]=i;
                dfs(cnt+1,i,ans);
            }
        }
        else//否则就判断和前一个数相加是不是素数
        {
            int t=ans[cnt-1]+i;
            if(isprime[t])
            {
                ans[cnt]=i;
                dfs(cnt+1,i,ans);
            }
        }
        visited[i]=false;//记得恢复
    }
}
int main()
{
    int k=1;
    while(scanf("%d",&n)!=EOF)
    {
        int ans[25];
        ans[1]=1;
        memset(visited,false,sizeof(visited));
        visited[1]=true;
        printf("Case %d:\n",k++);
        dfs(2,1,ans);
        printf("\n");
    }
    return 0;
}