結果

問題 No.2696 Sign Creation
ユーザー mo124121mo124121
提出日時 2024-02-04 10:06:15
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 2,854 bytes
コンパイル時間 396 ms
コンパイル使用メモリ 82,148 KB
実行使用メモリ 122,560 KB
最終ジャッジ日時 2024-12-20 11:53:24
合計ジャッジ時間 20,553 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 34 WA * 1 TLE * 3
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys

sys.setrecursionlimit(2 * 10**3)

if sys.implementation.name == "pypy":
    import pypyjit

    pypyjit.set_param("max_unroll_recursion=-1")


from collections import Counter, defaultdict


class UnionFind:
    def __init__(self, N):
        self.N = N
        self.parent_or_size = [-1] * N
        self.group_count = N

    def root(self, x):
        if self.parent_or_size[x] < 0:
            return x
        else:
            self.parent_or_size[x] = self.root(self.parent_or_size[x])
            return self.parent_or_size[x]

    def unite(self, x, y):
        root_x = self.root(x)
        root_y = self.root(y)
        if root_x == root_y:
            return
        if -self.parent_or_size[root_x] < -self.parent_or_size[root_y]:
            root_x, root_y = root_y, root_x
        self.parent_or_size[root_x] += self.parent_or_size[root_y]
        self.parent_or_size[root_y] = root_x
        self.group_count -= 1

    def same(self, x, y):
        return self.root(x) == self.root(y)

    def size(self, x):
        return -self.parent_or_size[self.root(x)]

    def groups(self):
        ret = defaultdict(list)
        for i in range(self.N):
            ret[self.root(i)].append(i)
        return [group for group in ret.values()]


input = lambda: sys.stdin.readline().rstrip()

SHIFT = 20
MASK = (1 << SHIFT) - 1


def conv(x, y):
    return x << SHIFT | y


def rev(pos):
    return pos >> SHIFT, pos & MASK


H, W, N, D = map(int, input().split())

star = {}
XY = []
for i in range(N):
    x, y = map(int, input().split())
    XY.append((x, y))
    star[conv(x, y)] = i

MOVES = []
for xd in range(-D, D + 1):
    for yd in range(-D, D + 1):
        d = abs(xd) + abs(yd)
        if d > D:
            continue
        MOVES.append((xd, yd))


uf = UnionFind(N)

cand = Counter()
for i, (x, y) in enumerate(XY):
    for xd, yd in MOVES:
        d = abs(xd) + abs(yd)
        if d > D:
            continue
        xn = x + xd
        yn = y + yd
        if xn < 0 or yn < 0:
            continue
        posn = conv(xn, yn)
        cand[posn] += 1
        if posn in star:
            uf.unite(i, star[posn])

stella = uf.groups()
is_seiza = [0] * N


count = 0
flag = 0
for block in stella:
    if len(block) > 1:
        count += 1
        for i in block:
            is_seiza[i] = 1
    else:
        flag = 1
mi = count
ma = count + flag


for pos in cand:
    x, y = rev(pos)
    seen = set()
    if cand[pos] == 1:
        continue
    for xd, yd in MOVES:
        d = abs(xd) + abs(yd)
        if d > D:
            continue
        xn = x + xd
        yn = y + yd
        posn = conv(xn, yn)
        if posn not in star:
            continue
        seen.add(uf.root(star[posn]))
    c = count + 1
    for s in seen:
        if is_seiza[s]:
            c -= 1
    mi = min(mi, c)
    ma = max(ma, c)
print(mi, ma)
0