k皇后

洛谷 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;
}