棋盘挑战(dfs爆搜,剪枝)

棋盘挑战

题目链接

给定一个 N×N 的棋盘，请你在上面放置 N 个棋子，要求满足：
每行每列都恰好有一个棋子
每条对角线上都最多只能有一个棋子
1 2 3 4 5 6

1 | | O | | | | |

2 | | | | O | | |

3 | | | | | | O |

4 | O | | | | | |

5 | | | O | | | |

6 | | | | | O | |

上图给出了当 N=6 时的一种解决方案，该方案可用序列 2 4 6 1 3 5 来描述，该序列按顺序给出了从第一行到第六行，每一行摆放的棋子所在的列的位置。
请你编写一个程序，给定一个 N×N 的棋盘以及 N 个棋子，请你找出所有满足上述条件的棋子放置方案。

输入格式

共一行，一个整数 N。

输出格式

共四行，前三行每行输出一个整数序列，用来描述一种可行放置方案，序列中的第 i 个数表示第 i 行的棋子应该摆放的列的位置。
这三行描述的方案应该是整数序列字典序排在第一、第二、第三的方案。
第四行输出一个整数，表示可行放置方案的总数。

数据范围

6≤N≤13

输入样例：

6

输出样例：

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

算法分析

我们这题使用暴力搜索来完成,时间复杂度顶多是O(n²)完全符合数据,每行每列,每对角线不能有两个,所以我们可以用2个数组来判断当前是否符合(因为我们遍历列或行可以保证一个),那(x,y)所在对角线就是y=x+b1和y=-x+b2所以每一个点都有x+y+b2=0这条是左上到右下的一条和y-x+b1=0这个是左下到右上的一条对角线,我们把(x,y)代入的话x+y和x-y+n就代表一个对角线你可以带几个数试一下,数组中x是纵轴,y是横轴
x-y+n是为了不出现负数.详细看代码

代码实现

#include<iostream>
#include<algorithm>
#include<math.h>
#include<cstdio>
using namespace std;
const int N=20;
bool hang[N],dui[N*2],undui[N*2];
int path[N];
int n;
int ans;
void dfs(int x)
{
    if(x>n)//如果选取列打到了就可以输出了
    {
        ans++;
        if(ans<=3)
        {
            for(int i=1;i<=n;i++)
                cout<<path[i]<<' ';
            cout<<endl;
        }
        return ;
    }
    for(int y=1;y<=n;y++)//遍历列判断
    {
        if(!hang[y]&&!dui[x+y]&&!undui[x-y+n])
        {
            path[x]=y;//存储路径
            hang[y]=dui[x+y]=undui[x-y+n]=true;//标记存在数了
            dfs(x+1);//遍历下一步
            path[x]=0;//回溯
            hang[y]=dui[x+y]=undui[x-y+n]=false;//回溯
        }
    }
}
int main()
{
    cin>>n;
    dfs(1);
    cout<<ans<<endl;
}