BNUOJ 4299 God Save the i-th Queen（想法题）

题目链接==》    http://www.bnuoj.com/v3/problem_show.php?pid=4299

 

题意：一个X*Y棋盘上放了N个queens，再放一个queen，有几个空格可以放，前提是和已经在棋盘上的queen不同行不同列不同斜线

 

思路：每个点所在四条直线上，不能再放queen，所以将有点的直线标记，再遍历所有点，如果点所在直线已被标记，则不选。（完了）

 

代码：

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<iostream>

using namespace std;

const int T=4e4+9;

int h[T],l[T],xie_1[T],xie_2[T];///xie_1表示斜率大于0的直线
int sum,n,m;

void begin1( )
{
    sum=0;
    memset(h, 0 , sizeof( int )*T);
    memset(l, 0 , sizeof( int )*T);
    memset(xie_1, 0 , sizeof( int )*T);
    memset(xie_2, 0 , sizeof( int )*T);
}

void judge( int x, int y)
{
    int w;
    h[x]=1; l[y]=1;
    xie_1[x+y+1]=1;
    w=m+1-y;  /// 标记斜率为负的棋盘，则将列标号倒置，从m-->1,所以这个点所在直线也是L(x+w+1)
    xie_2[x+w+1]=1;
}

void judge_two( int x, int y )
{
    int w=m+1-y;
    if( h[x] || l[y] || xie_1[x+y+1] || xie_2[x+w+1] )
        return ;
    sum++;

}




int main( )
{
    int k,a,b;
    while(scanf("%d%d%d",&n,&m,&k),(n+m+k))
    {
        begin1( );
        for(int i=1;i<=k;i++)
        {
            scanf("%d%d",&a,&b);
            judge( a, b );
        }
        for(int i=1;i<=n;i++)
            for(int j=1;j<=m;j++)
                judge_two( i, j );
        printf("%d\n",sum);
    }
    return 0;
}