HDUOJ 1016 Prime Ring Problem（dfs）

链接:http://acm.hdu.edu.cn/showproblem.php?pid=1016
思路：这一道题第一次写是很久远的事情，那时候刚学c++两三天？当时直接next_permutation，妥妥TLE，然后想了一下 20!，天哪！不超时就奇怪了，然后就观察，比如n为4的时候，1324各种排列在3的时候已经不合法了，就不用再考虑1324，1342这两种情况了，优化了好多，n越大越明显。上代码！

---

AC代码

#include<bits/stdc++.h>
#define ll long long
#define lowbit(x) x&(-x)
#define eps 1e-8
#define lson l,m,td<<1
#define rson m+1,r,td<<1|1
#define INF 0x3f3f3f3f
#define clear(a) memset(a,0,sizeof a)
#define clear_1(a) memset(a,-1,sizeof a)
#define line puts("");
#define LOCALR freopen("C:\\Users\\Administrator\\Desktop\\in.txt","r",stdin);
#define LOCALW freopen("C:\\Users\\Administrator\\Desktop\\out.txt","w",stdout);
using namespace std;
/*******************************************************************/
/*******************************************************************/
template<class T>
inline int read(T &x){
    char ch;
    bool flag=false;
    if((ch=getchar())==EOF)return -1;
    for(;!isdigit(ch);ch=getchar())if(ch=='-')flag=true;
    for(x=0;isdigit(ch);x=x*10+ch-'0',ch=getchar());
    x=flag?-x:x;
    return 1;
}
template<class T>
inline void write(T x,bool isnewline=false){
    static const int maxlen=100;
    static char s[maxlen];
    if(x<0){putchar('-');x=-x;}
    if(!x){putchar('0');return;}
    int len=0;
    for(;x;x/=10)s[len++]=x%10+'0';
    if(isnewline)puts(s);
    else for(int i=len-1;i>=0;i--)putchar(s[i]);
}
/*******************************************************************/
/*******************************************************************/
bool prime[40];
void get_prime(){
    memset(prime,true,sizeof prime);
    prime[0]=prime[1]=false;
    for(int i=2;i<40;i++)
        if(prime[i])for(int j=2*i;j<40;j+=i)prime[j]=false;
}
bool vis[22];
int a[22],n;
void print(){
    for(int i=0;i<n;i++){
        write(a[i]);
        if(i!=n-1)putchar(' ');
        else line
    }
}
void dfs(int cur){
    if(cur==n){
        if(prime[a[0]+a[n-1]])print();
    }
    else{
        for(int i=1;i<=n;i++){
            if(vis[i]||!prime[a[cur-1]+i])continue;
            vis[i]=1;a[cur]=i;
            dfs(cur+1);
            vis[i]=0;
        }
    }
}
int main(){
    get_prime();
    int cnt=0;
    while(~scanf("%d",&n)){
        memset(vis,0,sizeof vis);
        printf("Case %d:\n",++cnt);
        if(n==1)puts("1");
        else if(!(n&1)){
            a[0]=1;vis[1]=1;
            dfs(1);
            vis[1]=0;
        }
        line
    }
    return 0;
}