1008 素数环问题

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

输入
没有输入。

输出
输出A，两个数字之间用一个空格隔开，第一个数字前面和最后一个数字后面没有空格。

输入样例

输出样例

提示
本题不提供输出样例。

#include <iostream>
#include <math.h>
#include <stdlib.h>

using namespace std;
int a[21];   //存放结果的数组
int sum=0;
void print()
{
     sum++;
    if(sum==2)
   {
        for(int i=1;i<20;i++)
            cout<<a[i]<<" ";
        cout<<a[20]<<endl;
        exit(0);
    }

}
void swap(int m,int i)
{
    int t;
    t=a[m];
    a[m]=a[i];
    a[i]=t;

}
int isprime(int num)
{
    int k=sqrt(num);
    for(int i=2;i<=k;i++)
    {
        if(num%i==0)
            return 0;
    }
    return 1;
}
void search(int m)
{
    if(m>20)
    {
        if(isprime(a[1]+a[20]))
            print();
            return;
    }
    else
    {
        for(int i=m;i<=20;i++)
        {
            swap(m,i);     //交换a[m]和a[i]
            if(isprime(a[m]+a[m-1])) //交换了之后，如果a[m]与前面元素之和是素数
                search(m+1);
            swap(m,i);     //若不是素数再交换回来
        }
    }
}
int main()
{
    int i;
    for(i=1;i<=20;i++)
        a[i]=i;
    search(2);
    return 0;
}

分析：利用回溯算法，假设前m个数已经排好的情况下，安排第m+1个数，其中，每个数的安排都要通过检测其与前面一个数的和是否为素数，如果是，则继续安排下一个数。
初始化的时候，先讲a[i]位对应i，在search()函数中再将m位的元素与可能的元素进行交换。
尚未解决的问题：在编程的过程中，发现输出结果是无限的数组并且每次结果不一样，在借鉴了别人的答案之后，在print()函数里加了计数变量sum，sum==2时输出，便可以只输出一组结果，但作者现在还不明白是为什么。