洛谷P1219八皇后

典型深搜, 直接上代码:

#include
#include 
using namespace std;
int n, cnt;
int r[20], check[5][50];
void eight_queen(int now, int row);
int main()
{
 scanf("%d", &n);
 eight_queen(0, 0);
 printf("%d", cnt);
 return 0;
}
void eight_queen(int now, int row){
 r[now] = row;
 if(now == n){
  cnt++;
  if(cnt < 4){
   for(int i = 1; i <= n; i++)printf("%d ", r[i]);
   printf("\n");
  } 
  
 }
 else{
  for(int i = 1; i <= n; i++){
   if(!check[0][i] && !check[1][now + 1 + i] && !check[2][n + now + 1 - i]){//坑: 数组下标不能是负数!! 
    check[0][i] = 1; check[1][now + 1 + i] = 1; check[2][n + now + 1 - i] = 1;
    eight_queen(now + 1, i);
    check[0][i] = 0; check[1][now + 1 + i] = 0; check[2][n + now + 1 - i] = 0;
   }
  }
 }
}

值得注意的地方:

  1. 用check记录皇后放置的位置,在具有线性关系的地图(棋盘)上很实用.(如果用遍历会TLE.
  2. 数组下标不能为负数!!!(被坑好几遍了都
  3. 深搜标记后记得清除标记

你可能感兴趣的:(ACM,dfs,洛谷)