K皇后


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();
}

 

原创文章,作者:3628473679,如若转载,请注明出处:https://blog.ytso.com/277998.html

(0)
上一篇 2022年7月31日
下一篇 2022年7月31日

相关推荐

发表回复

登录后才能评论