题目描述
一个如下的 6×6 的跳棋棋盘,有六个棋子被放置在棋盘上,使得每行、每列有且只有一个,每条对角线(包括两条主对角线的所有平行线)上至多有一个棋子。
上面的布局可以用序列 2 4 6 1 3 5来描述,第 i 个数字表示在第 i 行的相应位置有一个棋子,如下:
行号 1 2 3 4 5 6
列号 2 4 6 1 3 5
这只是棋子放置的一个解。请编一个程序找出所有棋子放置的解。
并把它们以上面的序列方法输出,解按字典顺序排列。
请输出前 3 个解。最后一行是解的总个数。
输入格式
一行一个正整数 n,表示棋盘是n×n 大小的。
输出格式
前三行为前三个解,每个解的两个数字之间用一个空格隔开。第四行只有一个数字,表示解的总数。
输入输出样例
输入 #1复制
6
输出 #1复制
2 4 6 1 3 5 3 6 2 5 1 4 4 1 5 2 6 3 4
思路:
判断对角线唯一的理论:
题解:
代码如下:
#include<bits/stdc++.h>
using namespace std;
int number[15] = {1,2,3,4,5,6,7,8,9,10,11,12,13};
bool vis[15],a[30],b[30]; //有没有访问过、左上角、左下角
int ans[15];
int cnt = 0;
bool isOK(int r,int c,int n) // 判断对角线
{
if(a[r - c + n - 1] || b[r + c] ) return false;
return true;
}
void dfs(int s,int n)
{
if(s == n)
{
cnt++;
if(cnt <= 3)
{
for(int i = 0; i < n; i++)
{
cout<<ans[i];
if(i < n - 1) cout<<" ";
else cout<<endl;
}
}
return ;
}
for(int i = 0; i < n; i++)
{
int r = s,c = number[i] - 1; //当前行数就是s,列数是number[i] - 1
if(!vis[i] && isOK(r,c,n)) //用isOK()剪枝
{
ans[s] = number[i];
a[r - c + n - 1] = b[r + c] = true;
vis[i] = true;
dfs(s + 1,n);
vis[i] = false;
a[r - c + n - 1] = b[r + c] = false;
}
}
}
int main()
{
int n; cin>>n;
dfs(0,n);
cout<<cnt;
return 0;
}
这种全排列解法就是回溯,只不过将进行回溯的方式改变了,本质是一样的。经典的回溯是:先将当前状态进队列,再dfs下一层,dfs完了再将当前状态出队列(回溯为原来状态),最后用for进行下一次横向搜索;我的全排列是用vis[]将当前状态标记为访问过,再dfs下一层,最后回溯再将当前vis[]回溯为未访问。
在我用全排列的思想在洛谷提交且AC之后,我发现有人说全排列解法开了O2优化才能过,不开O2优化会超时,于是我将我的代码在没开O2优化的状态下再次进行了提交,结果依然是AC的。所以只要进行了合理的剪枝且算法没有问题,都是AC的。