天天写算法之Prime Ring Problem

点击打开链接

一个纯粹的DFS的题，主要是看用什么写法会简单，并且一次写对。用一个num，存储数列，并且因为要输出所有满足要求的数据，因此需要判断何时输出这个数据，并且还需要考虑如何考虑最后一个数据与开头数据加和仍然为质数的问题。

因此需要两个数组，一个记录访问，一个记录每个位置的数据。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<queue>
#define inf 0x3f3f3f3f
#define MAX 22
#define repf(i,from ,to) for(int i =from ; i <to ; i++)
using namespace std;


int visit[MAX];
int n;
int num[MAX];

bool isPrime(int a )
{
  if(a==1)
    return false ;
  for(int i = 2;i<=sqrt(a);i++)
  {
      if(a%i==0)
      {
          return false ;
      }
  }
  return true ;
}
//迭代深搜
void dfs(int step)
{
   if(step>n&&isPrime(num[n]+num[1]))
   {
       for(int i = 1 ; i <n ; i ++)
        cout << num[i]<<" ";
        cout << num[n]<<endl;
   }
   for(int i = 2 ; i <=n ; i ++)
   {
       num[step]=i;
       if(isPrime(num[step]+num[step-1])&&!visit[i])
       {
           visit[i]=1;
           dfs(step+1);
           visit[i]=0;
       }
   }
}
int main(){
 int c = 1 ;
  while(  cin >> n )
   {
    cout<<"Case "<<c++<<":"<<endl;
    memset(num,0,sizeof(num));
    num[1] = 1;
    dfs(2);
    cout <<endl;
   }
   return 0 ;

}