poj1016素数环

题意 : 给一个n 给出由1到n的书组成的环两两相加是素数
用dfs搜索 i表示当前数字 len表示深度 成环还要满足数组收尾相加是素数
代码：

#include<bits/stdc++.h>
using namespace std;

int n;
int a[21];
int vis[21];

bool is_prime(int a){
    if(a == 1)
        return false;
    for(int i = 2; i*i <= a; i++)
        if(a % i == 0)
            return false;
    return true;
}
void dfs(int i, int len){
    a[len] = i;
    if(len == n){
        if(is_prime(a[1] + a[n]))
            for(int i = 1; i <= n; i++){
                cout << a[i] ;
                if(i != n)
                    cout <<' ';
                else
                    cout << endl;
            }
        return;
    }

    for(int j = 2; j <= n; j++){
        if(is_prime(a[len] + j) && vis[j] == 0){
            //cout <<a[len] <<' ' <<  j <<endl;
            vis[j] = 1;
            dfs(j, len+1);
            vis[j] = 0;
        }
    }
}

int main(){
    int k = 1;
    while(cin >> n){
        cout <<"Case " << k++ <<':' <<endl;
        memset(vis, 0, sizeof(vis));
        dfs(1, 1);
        cout <<endl;
    }
    return 0;
}