技術標籤：專題一 簡單搜尋DFS洛谷演算法

我是大一新生，剛入ACM的坑，寒假閒來無事，白天訓練晚上就發一發今天寫的題目鞏固一下知識，希望可以拿到ACM金牌!

題目描述:
一個如下的6×6 的跳棋棋盤，有六個棋子被放置在棋盤上，使得每行、每列有且只有一個，每條對角線（包括兩條主對角線的所有平行線）上至多有一個棋子。

上面的佈局可以用序列 2 4 6 1 3 5 來描述，第 i 個數字表示在第 i 行的相應位置有一個棋子，如下：
行號
1 2 3 4 5 6
列號
2 4 6 1 3 5
這只是棋子放置的一個解。請編一個程式找出所有棋子放置的解。

一行一個正整數 n，表示棋盤是 n×n 大小的。

前三行為前三個解，每個解的兩個數字之間用一個空格隔開。第四行只有一個數字，表示解的總數。

6

2 4 6 1 3 5
3 6 2 5 1 4
4 1 5 2 6 3
4

對於 100% 的資料，6≤n≤13。

這是一道典型的題目，每行每列每條對角線上只能有一個皇后，利用深度優先搜尋就可以解決。廢話不多說，上程式碼

#include<iostream>
 

#include<cstdio>
#include<cstdlib>
using namespace std;
int a[100],b[100]={0},c[100]={0},d[100]={0};
//陣列a記錄結果的列號，陣列b記錄列佔領，
//陣列c記錄左下到右上到對角線佔領，陣列d記錄右下到左上的對角線佔領。
int total=0,n;
void printf(){//輸出函式
    if(total<=2)
    {
    for(int k=1;k<=n;k++)
    printf("%d ",a[k]);
    printf("\n"
 
);
    }
    total++;
}
void dfs(int i){
    if(i>n){
    printf();
    return;
    }
    else{
        for(int j=1;j<=n;j++){
            if((!b[j])&&(!c[i+j])&&(!d[i-j+n]))
            {
                a[i]=j;//記錄當前行所對應的列
                b[j]=1;//宣佈佔領此列
                c[i+j]=1;//宣佈佔領兩條對角線
                d[i-j+n]=1;
                dfs(i+1);//進行下一輪深搜，放置下一個皇后
                b[j]=0;
                c[i+j]=0;
                d[i-j+n]=0;
                //回溯
            }
        }
    }
}
int main(){
    scanf("%d",&n);
    dfs(1);
    printf("%d",total);
}