洛谷 P2105 K皇后
题目描述
小Z最近捡到了一个棋盘,他想在棋盘上摆放K个皇后。他想知道在他摆完这K个皇后之后,棋盘上还有多少了格子是不会被攻击到的。
(Ps:一个皇后会攻击到这个皇后所在的那一行,那一列,以及两条对角线)
输入输出格式
输入格式:
第一行三个正整数 n,m,K,表示棋盘的行列,以及小Z摆放了K个皇后。
接下来K行,每行两个正整数x,y,表示这个皇后被摆在了第x行,第y列,数据保证没有任何两个皇后会被摆在同一个格子里。
输出格式:
一行一个整数,表示棋盘上还有多少了格子是不会被攻击到的。
这与用递归解决的n皇后问题有点像,又不完全相同,不适合用递归,看了篇没什么注释的题解整理了下.
感觉c数组的含义有点抽象,不确定表示什么.
#include <iostream>
#include <cstring>
using namespace std;
int x[505],y[505],c[20005];
//c用来标记是否会被攻击,不代表行或列
int main(){
int n,m,k;
cin>>n>>m>>k;
int ans=0,num;
memset(c,0,sizeof(c));
for(int i=1;i<=k;i++)
{
cin>>x[i]>>y[i];
}
for(int i=1;i<=n;i++)
{
num=m;
for(int j=1;j<=k;j++)
{
if(i==x[j])//这行有皇后直接break
{
num=0;
break;
}
else{
if(c[y[j]]!=i)// 判断列
{
c[y[j]]=i;//因为i一直在变,不用更新数组,可以判断到每一行
num--;
}
if((y[j]+(x[j]-i))>0 && (y[j]+(x[j]-i))<=m && c[(y[j]+(x[j]-i))]!=i)
{
c[y[j]+(x[j]-i)]=i;
num--;
}
if((y[j]-(x[j]-i))>0 && (y[j]-(x[j]-i))<=m && c[y[j]-(x[j]-i)]!=i)
{
c[y[j]-(x[j]-i)]=i;
num--;
}
//会被皇后斜着攻击的点的横纵坐标之差与皇后的相等
//x[j]-i为皇后与这行横坐标之差,再用y[j]加减则是每个会被攻击的点的纵坐标
}
}
//cout<<i<<" "<<num<<endl;
ans+=num;
}
cout<<ans;
return 0;
}