题目描述
一个如下的 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
题解
这题不难,但最后卡的一个点令人很难办。
这次我的记录题解主要涉及对那个点的研究解决。
有些方法虽不成,但亦有其参考价值。
初解
又一道dfs。主要就是共存判断要注意。
代码
#include<bits/stdc++.h>
using namespace std;
int n,ans=0;
int a[100]; //栈存储结果
void dfs(int top)
{
if(top==n+1) 基线条件
{
ans++;
if(ans<=3)
{
for(int i=1;i<=n;i++)
{
if(i==1) cout<<a[i];
else cout<<" "<<a[i];
}
cout<<endl;
}
return;
}
for(int i=1;i<=n;i++) //循环这一行的n列
{
//对应点为(top,i)
//判断共存
int flag=1;
for(int j=1;j<=top-1;j++) //与之前栈中已存点(j,a[j])进行共存判断
{
if(j!=top&&a[j]!=i&&abs(a[j]-i)!=abs(j-top))
{
; //行标不等+列标不等+斜率绝对值不为1(斜线)
}
else
{
flag=0;
break;
}
}
if(flag==1)
{
a[top] = i;
dfs(top+1);
}
}
}
int main()
{
cin>>n;
dfs(1);
cout<<ans<<endl;
return 0;
}
斜共存判断:两点斜率不为正负1
不过,结果是
于是,我开始了尝试。
尝试1:吸氧(o^2)
这个过了,时间瞬间降下来了。但是比赛似乎不让用诶0.0
尝试2: 寄存器+改为C的输入输出
这是企图偷鸡的办法,想看看能不能把时间降一点(毕竟就一个点了)
但还是不行
代码
#include<bits/stdc++.h>
using namespace std;
int n,ans=0;
int a[100]; //栈
void dfs(int top)
{
if(top==n+1)
{
ans++;
if(ans<=3)
{
for(register int i=1;i<=n;i++) //听说寄存器会更快哦
{
if(i==1) printf("%d",a[i]);
else printf(" %d",a[i]);
}
printf("\n");
}
return;
}
for(register int i=1;i<=n;i++)
{
//点为(top,i)
//判断共存
bool flag=true;
for(register int j=1;j<=top-1;j++) //两层
{
if(j!=top&&a[j]!=i&&abs(a[j]-i)!=abs(j-top))
{
;
}
else
{
flag=0;
break;
}
}
if(flag)
{
a[top] = i;
dfs(top+1);
}
}
}
int main()
{
scanf("%d",&n);
dfs(1);
printf("%d\n",ans); //改用了C
return 0;
}
尝试3:打表
关键时刻确实可以。毕竟题中给了6≤n≤13。工作量也不大。
过了。
尝试4:重新处理共存情况
研究了一下题解区大佬的代码。我用了两层循环才办到的事,他们一层就办到了。这O(n^2)和O(n)可是天差地别。
遂学习了解。
通过数组标记,避免了一层循环。
其实这贴近于找规律了。
看同一行、列、斜的坐标的共性,只要避免了这些共性,也就找到结果了。
首先行的移动本身就是我那dfs的深层参数top控制,是不用管的。
然后列标要存一个bool数组,用过的列标标记为true。
斜度分为左斜还是右斜。
看看这些点吧。
同左斜一线:(2,4) (3,3) (4,2) (5,1)
同右斜一线:(2,4) (3,5) (4,6)
显然,同一左斜,行列标之和相同。同一右斜,(列标-行标)相同。
这就是共性,也就是左右斜数组要标记的因素。
ac代码:
#include<bits/stdc++.h>
using namespace std;
int n,ans=0;
int a[100]; //栈
bool b[100],c[100],d[100]; //标记数组(找规律)
void dfs(int top)
{
if(top==n+1)
{
ans++;
if(ans<=3)
{
for(int i=1;i<=n;i++)
{
if(i==1) cout<<a[i];
else cout<<" "<<a[i];
}
cout<<endl;
}
return;
}
for(int i=1;i<=n;i++) //列标
{
if(!b[i] && !c[top+i] && !d[top-i+n]) //为了安全
{
a[top] = i;
b[i]=true; //不能赋值1
c[top+i]=true;
d[top-i+n]=true;
dfs(top+1);
b[i]=false; //回溯
c[top+i]=false;
d[top-i+n]=false;
}
}
}
int main()
{
cin>>n;
dfs(1);
cout<<ans<<endl;
return 0;
}
细节:
1.代码中出现了d[top-i+n],其实按推导规律,只要top-i相等就行了。但这里出现了减法。为了安全,要保证数组下标不能为负数,遂+n。其实你愿意加多少加多少。
2.对于bool变量的赋值
只能赋值true/false。
我本来的代码是
a[top] = i;
b[i]=1;
c[top+i]=1;
d[top-i]=1;
dfs(top+1);
b[i]=0; //回溯
c[top+i]=0;
d[top-i]=0;
但提交的时候,洛谷告诉我"Call to a member function row_array() on bool"。
不太懂,但想来也是bool赋值问题,也许我这企图让它自动类型转化的写法不太好吧。
以后记住:bool类型的东西,赋值便只赋值true/false.