• 欢迎光临~

K皇后

开发技术 开发技术 2022-07-31 次浏览

https://www.luogu.com.cn/problem/P2105

  • 遍历行 ,如果发现有标记的就直接下一个
  • 答案先加上这一行所有不可能在逐个减去
  • 遍历所有皇后
  • lie[queens[q].y]!=i说明此时在同一行
  • int y = queens[q].x + queens[q].y - i左下45°上有皇后
  • y = i - queens[q].x + queens[q].y;右下45°有皇后
  • 将lie中的值赋值为i一方面可以标记,这样就不会因为多个皇后同时管理到一个点而导致这个点被多次删去,另一方面因为i代表的是行数,所以到下一行时所有的都相当于没有标记,就不用memset了
#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define MAX 10000001
struct node
{
    int x, y;
} queens[MAX];
int hang[MAX], lie[MAX];
int n, m, k, ans;
void input()
{
    cin >> n >> m >> k;
    for (int i = 1; i <= k; i++)
    {
        scanf("%d%d", &queens[i].x, &queens[i].y);
        hang[queens[i].x] = 1;
    }
}
void work()
{
    for (int i = 1; i <= n; i++)
    {
        if (hang[i])
            continue;
        ans += m;
        for (int q = 1; q <= k; q++)
        {
            if (lie[queens[q].y] != i)
                ans--;
            lie[queens[q].y] = i;
            // zuoxia
            int y = queens[q].x + queens[q].y - i;
            if (y >= 1 && y <= m)
            {
                if (lie[y] != i)
                {
                    ans--;
                }
                lie[y] = i;
            }
            y = i - queens[q].x + queens[q].y;
            if (y >= 1 && y <= m)
            {
                if (lie[y] != i)
                {
                    ans--;
                }
                lie[y] = i;
            }
        }
    }
    cout << ans;
}
int main()
{
    input();
    work();
}

 

程序员灯塔
转载请注明原文链接:K皇后
喜欢 (0)