【HDOJ 1016】Prime Ring Problem

【HDOJ 1016】Prime Ring Problem

暴力dfs 搜到最后一位后输出 输出前注意判断最后一位与首位(1)加和是否为素数

代码如下:

#include <iostream>
#include <cstdlib>
#include <cstdio>
#include <cstring>
#include <queue>

using namespace std;

int num[22];
bool IsPrim[388],vis[22];

void SetP()
{
    memset(vis,0,sizeof(vis));
    memset(IsPrim,0,sizeof(IsPrim));
    int i,j;
    for(i = 3; i <= 380; ++i)
    {
        for(j = 2; j < i; ++j)
        {
            if(!(i%j)) IsPrim[i] = true;
        }
    }
}

void dfs(int p,int n)
{
    int i;
    if(p == n)
    {
        if(IsPrim[num[p-1]+1]) return;
        for(i = 0; i < n; ++i)
        {
            if(i) printf(" ");
            printf("%d",num[i]);
        }
        puts("");
        return;
    }

    for(i = 2; i <= n; ++i)
    {
        if(!vis[i] && !IsPrim[num[p-1]+i])
        {
            num[p] = i;
            vis[i] = 1;
            dfs(p+1,n);
            vis[i] = 0;
        }
    }
}

int main()
{
    int n,z = 0;
    SetP();

    while(~scanf("%d",&n))
    {
        printf("Case %d:\n",++z);
        num[0] = 1;
        dfs(1,n);
        puts("");
    }

    return 0;
}