提出 #17968851

---

ソースコード 拡げる

import sys
import numpy as np
import numba
from numba import njit, b1, i4, i8, f8

read = sys.stdin.buffer.read
readline = sys.stdin.buffer.readline
readlines = sys.stdin.buffer.readlines

def from_read(dtype=np.int64):
    return np.fromstring(read().decode(), dtype=dtype, sep=' ')


def from_readline(dtype=np.int64):
    return np.fromstring(readline().decode(), dtype=dtype, sep=' ')

@njit((i8, i8, i8[:, :], i8[:, :]), cache=True)
def main(H, W, AB, CD):
    G = np.zeros((H, W), np.int64)
    for i in range(len(AB)):
        a, b = AB[i]
        G[a - 1, b - 1] = 1
    for i in range(len(CD)):
        c, d = CD[i]
        G[c - 1, d - 1] = -1
    res = np.zeros((H, W), np.bool_)  # 光ったマス
    for _ in range(4):
        res, G = res.T[::-1], G.T[::-1]
        # 下方向に解く
        A = np.zeros_like(res)
        H, W = A.shape
        for h in range(H):
            if h >= 1:
                A[h] |= A[h - 1]
            A[h][G[h] == 1] = 1
            A[h][G[h] == -1] = 0
        res |= A
    return np.sum(res)

H, W, N, M = map(int, readline().split())
nums = from_read()
AB = nums[:N + N].reshape(N, 2)
CD = nums[N + N:].reshape(M, 2)

print(main(H, W, AB, CD))

提出情報

ジャッジ結果