回溯法之素数环问题

把1到20这重新排列，使得排列后的序列A满足：
a. 任意相邻两个数之和是素数
b. 不存在满足条件a的序列B使得：A和B的前k（0 <= k <= 19）项相同且B的第k+1项比A的第k+1项小。

#include <iostream>
#include <cmath>
/*
    1-20 20个元素排成一个环，要求相邻的两个元素的和都为素数
*/

using namespace std;

int sum = 0;
int isUsed[20] = { 0 }; //记录值是否被使用
void output(int* A, int n);
bool isPrimeNumber(int t);
/*
    m索引，从1开始寻找
    P用来查询是否为素数
*/
void search(int* A, int n, int m,int* P) {

    if (m == n && P[A[0] + A[n-1]]) {
        output(A,n);
        exit(-1);
        sum += 1;
    }   
    else { 
        //穷举所有全排列
        for (int i = 2; i <= n; i++) {
            if (!isUsed[i] && P[i + A[m - 1]] == 1){ //当前值i没被使用,且与前一个选定值之和为素数  
                A[m] = i;//选i为当前项值  
                isUsed[i] = 1;//状态从没被使用改为被使用  
                search(A,n,m+1,P);//进入下一层,若cur+1<n则求下一个有效值,否则执行输出语句  
                //递归后面的语句在从n-1层到第1层回调时执行  
                isUsed[i] = 0;//状态还原,使重新求下一个有效串时不被干扰  
            }

        }
    }

}


bool isPrimeNumber(int t) {
    if (t<2)
        return false;
    int len = (int)sqrt(t + 0.0);
    for (int i = 2; i <= len; i++){
        if (t%i == 0)
            return false;
    }
    return true;
}
void output(int* A, int n) {
    for (int i = 0; i < n; i++) {
        if (i == n - 1) {
            break;
        }
        cout << A[i] << " ";
    }
    cout << A[n-1] <<endl;
}

const int n = 20;
int main() {

    int p[40] =
    { 0, 0, 1, 1, 0,
    1, 0, 1, 0, 0,
    0, 1, 0, 1, 0,
    0, 0, 1, 0, 1,
    0, 0, 0, 1, 0,
    0, 0, 0, 0, 1,
    0, 1, 0, 0, 0,
    0, 0, 1, 0, 0 };

    int A[n] = { 0 };
    for (int i = 0; i < n; i++) {
        A[i] = i+1;
    }
    search(A, n,1, p);
//  output(A, n);
//  cout << sum << endl;
}