Prime Ring Problem

Posted 2021-12-01 kasenbob

题目链接：Prime Ring Problem 

思路：

类似八皇后问题，做递归处理即深搜，注意状态变化即可

代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int maxn=200;
int prime[maxn];
int vis[20];
int ans[20];
int n;

void init()
{
    memset(prime,0,sizeof(prime));
    prime[0]=prime[1]=1;
    for(int i=2;i<=maxn;i++)
    {
        if(prime[i]==0)
        {
            for(int j=2;i*j<=maxn;j++)
            {
                prime[i*j]=1;
            }
        }
    }
}

void ring(int ce)
{
    if(ce==n)
    {
        for(int i=0;i<n;i++)
        {
            printf("%d",ans[i]);
            i==n-1?cout<<endl:cout<<" ";
        }
        return;
    }
    else if(ce==0)
    {
        vis[1]=1;
        ans[ce]=1;
        ring(ce+1);
    }
    else if(ce==n-1)
    {
        for(int i=1;i<=n;i++)
        {
            if(vis[i]==0)
            {
                if(prime[i+ans[ce-1]]==0&&prime[i+ans[0]]==0)
                {
                    vis[i]=1;
                    ans[ce]=i;
                    ring(ce+1);
                    vis[i]=0;
                }
                else
                {
                    return;
                }
            }
        }
    }    
    else
    {
        for(int i=1;i<=n;i++)
        {
            if(vis[i]==0)
            {
                if(prime[i+ans[ce-1]]==0)
                {
                    vis[i]=1;
                    ans[ce]=i;
                    ring(ce+1);
                    vis[i]=0;
                }
            }
        }
    }
}

int main()
{
    init();
    int Case=1;
    while(~scanf("%d",&n))
    {
        memset(ans,0,sizeof(ans));
        memset(vis,0,sizeof(vis));
        if(Case!=1)cout<<endl;
        printf("Case %d:\n",Case++);
        ring(0);
        
    }
}