洛谷P2105 K皇后

To 洛谷.2105 K皇后

题目描述

小Z最近捡到了一个棋盘，他想在棋盘上摆放K个皇后。他想知道在他摆完这K个皇后之后，棋盘上还有多少了格子是不会被攻击到的。

（Ps：一个皇后会攻击到这个皇后所在的那一行，那一列，以及两条对角线）

输入输出格式

输入格式：

 

第一行三个正整数 n，m，K，表示棋盘的行列，以及小Z摆放了K个皇后。

接下来K行，每行两个正整数x，y，表示这个皇后被摆在了第x行，第y列，数据保证没有任何两个皇后会被摆在同一个格子里。

 

输出格式：

 

一行一个整数，表示棋盘上还有多少了格子是不会被攻击到的。

 

输入输出样例

输入样例#1：

12 13 6
10 4
12 10
1 1
2 3
3 2
2 6

输出样例#1：

25

说明

【数据规模和约定】

对于 30%的数据，1 ≤ n，m ≤ 5000，1 ≤ k ≤ 500。

对于另外 10%的数据，k =1。

对于 100%的数据，1 ≤ n，m ≤ 20000，1 ≤ k ≤ 500。

【时空限制】

1s/16M

代码：

#include<cstdio>
#include<cstring>
using namespace std;
const int N=20005;
const int K=505;

int n,m,k,Ans,X[K],Y[K];
bool vis[N];

void read(int &now)
{
    now=0;char c=getchar();
    while(c>'9'||c<'0')c=getchar();
    while(c>='0'&&c<='9')now=(now<<3)+(now<<1)+c-'0',c=getchar();
}

int main()
{
    read(n);read(m);read(k);
    for(int i=1;i<=k;++i)
      read(X[i]),read(Y[i]);
    for(int i=1;i<=n;++i)
    {
        int tmp=m;
        for(int j=1;j<=k;++j)
        {//枚举m列(20000)会超时，那就枚举k个棋子(500) 
            if(X[j]==i)
            {//棋子j能吃到整一行，这一行不被攻击的棋子为0，退出循环 
                tmp=0;break;
            }
//            int t1,t2;
//            if(X[j]<Y[j])
//              t1=Y[j],t2=X[j];
//            else
//              t1=X[j],t2=Y[j];
            //以下是两条对角线判断。注意同一行同一个列只能判一次 
            if(Y[j]-X[j]+i>0 && Y[j]-X[j]+i<=m && !vis[Y[j]-X[j]+i])//or vis[XXX]!=i 可以不需要重设0 
              vis[Y[j]-X[j]+i]=1,--tmp;//printf("tmp:%d-- on line%d,colu%d\n",tmp,i,X[j]-Y[j]+i);
            if(X[j]+Y[j]-i>0 && X[j]+Y[j]-i<=m && !vis[X[j]+Y[j]-i])
              vis[X[j]+Y[j]-i]=1,--tmp;//printf("tmp:%d-- on line%d,colu%d\n",tmp,i,X[j]+Y[j]-i);
            //列 
            if(!vis[Y[j]])
              vis[Y[j]]=1,--tmp;//printf("tmp:%d-- on line%d,colu%d\n",tmp,i,Y[j]);
            if(!tmp)
              break;
        }
        memset(vis,0,sizeof vis);
        Ans+= tmp;
//        printf("On line%d Ans:%d\n",i,Ans);
    }
    printf("%d",Ans);
    return 0;
}