N皇后

2018-06-23  本文已影响0人  AdmondGuo

回溯法核心代码:

void generateP(index)
{
  if(符合边界条件){
    计数,返回上一级;
  }
  for(枚举情况){
    if(能进行到下一代){
      flag=true;  //初始化
      generateP(index+1);
      flag=false;
    }  
  }
}

n皇后问题回溯法

/*n皇后问题回溯法*/
#include "stdlib.h"
#include "cstdio"
#define MAX 1000

bool hash[MAX];
int p[MAX];
int count=0;
int n=0;

void print(){
    int i,j;
    for(i=1;i<=n;i++){
        for(j=1;j<=n;j++){
            if(p[i]==j){
                printf("%d",1);printf(" ");
            }
            else{
                printf("%d",0);printf(" ");
        }
            printf("\n");
        }
    }
}
int generate(int index){//一个generate是一行
    if(index==n+1){       //最后,count++
        //print();
        count++;
        return 0;
    }
    for(int i=1;i<=n;i++){
        if(hash[i]==false){
            bool flag=true;
            for(int pre=1;pre<index;pre++){
                if(abs(index-pre)==abs(i-p[pre])){
                    flag=false;
                    break;
                }
            }
        if(flag){
            p[index]=i;
            hash[i]=true;
            generate(index+1);
            hash[i]=false;
        }
    }
}
    return 0;
}

int main()
{
    scanf("%d",&n);
    generate(1);
    printf("%d",count);
    system("pause");
    return 0;
}


上一篇 下一篇

猜你喜欢

热点阅读