hdu 1016

来源:互联网 发布:步惊云软件教学系统 编辑:程序博客网 时间:2024/06/06 10:01
题目大意:

输入一个数n,然后在一个有n个节点的圆环上放1~n这n个数,求满足相邻两个数相加是素数的情况,并输出;

样例:

6

8

样例输出:

Case 1:1 4 3 2 5 61 6 5 2 3 4Case 2:1 2 3 8 5 6 7 41 2 5 8 3 4 7 61 4 7 6 5 8 3 21 6 7 4 3 8 5 2
(注意格式,两组之间有一个空行)


基本思路:

一看n的范围,必然是用dfs,所以就上吧,但是很快发现不行啊,时超限,那怎么办,剪枝啊,这个还是很好剪的,就是判断一下当前最新的两个是不是相加等于素数,如果不是,就直接return;

代码如下:

#include<iostream>
#include<iomanip>
#include<algorithm>
#include<queue>
#include<stack>
#include<list>
#include<set>
#include<map>
#include<vector>
#include<cstring>
#include<cstdio>
#include<cmath>
using namespace std;
int prime[100+10];
int vis[30],temp[30];
int n;
void init()
{
    memset(prime,0,sizeof(prime));
    prime[1]=1;
    int m=sqrt(50+0.5);
    for(int i=2;i<=m;i++)
    {
        if(!prime[i])
        {
            for(int j=i+i;j<=50;j+=i)
            {
                prime[j]=1;
            }
        }
    }
    return;
}
void judge()
{
    if(prime[temp[n]+temp[1]]) return;
    for(int i=1;i<=n-1;i++)
    {
        if(prime[temp[i]+temp[i+1]])
            return;
    }
    printf("1");
    for(int i=2;i<=n;i++)
    {
        printf(" %d",temp[i]);
    }
    printf("\n");
}
void dfs(int cur)
{
    if(cur==n+1)
    {
        judge();
        return;
    }
    if(cur>2)
    {
        if(prime[temp[cur-1]+temp[cur-2]]) return;//剪枝
    }
    for(int i=2;i<=n;i++)//这个循环才是这个数组能取到每一种可能的原因,而不是下面的注释,那个注释那个地方会使程序爆炸,因为会一直存在某次递归到达不了cur+1;
    {
        if(!vis[i])
        {
            vis[i]=1;
            temp[cur]=i;
            dfs(cur+1);
            vis[i]=0;

//     dfs(cur);这个写法是错误的,之前没有弄明白;
        }
    }
    return;
}
int main()
{
    init();
    int cnt=1;
    while(scanf("%d",&n)==1)
    {
        printf("Case %d:\n",cnt++);
        memset(vis,0,sizeof(vis));
        temp[1]=1;
        dfs(2);
        printf("\n");
    }
    return 0;
}

原创粉丝点击