回溯法之素数环问题

把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;
}
    原文作者:回溯法
    原文地址: https://blog.csdn.net/u011040361/article/details/45156357
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞