洛谷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=;
 const int K=;

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

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

 int main()
 {
     read(n);read(m);read(k);
     for(int i=;i<=k;++i)
       read(X[i]),read(Y[i]);
     for(int i=;i<=n;++i)
     {
         int tmp=m;
         for(int j=;j<=k;++j)
         {//枚举m列(20000)会超时，那就枚举k个棋子(500)
             if(X[j]==i)
             {//棋子j能吃到整一行，这一行不被攻击的棋子为0，退出循环
                 tmp=;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> && Y[j]-X[j]+i<=m && !vis[Y[j]-X[j]+i])//or vis[XXX]!=i 可以不需要重设0
               vis[Y[j]-X[j]+i]=,--tmp;//printf("tmp:%d-- on line%d,colu%d\n",tmp,i,X[j]-Y[j]+i);
             if(X[j]+Y[j]-i> && X[j]+Y[j]-i<=m && !vis[X[j]+Y[j]-i])
               vis[X[j]+Y[j]-i]=,--tmp;//printf("tmp:%d-- on line%d,colu%d\n",tmp,i,X[j]+Y[j]-i);
             //列
             if(!vis[Y[j]])
               vis[Y[j]]=,--tmp;//printf("tmp:%d-- on line%d,colu%d\n",tmp,i,Y[j]);
             if(!tmp)
               break;
         }
         memset(vis,,sizeof vis);
         Ans+= tmp;
 //        printf("On line%d Ans:%d\n",i,Ans);
     }
     printf("%d",Ans);
     return ;
 }