結果

問題 No.3339 Tree on Checkerboard
コンテスト
ユーザー ikatakos
提出日時 2025-11-08 01:52:30
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 2,208 bytes
コンパイル時間 278 ms
コンパイル使用メモリ 12,416 KB
実行使用メモリ 75,164 KB
最終ジャッジ日時 2025-11-08 01:53:29
合計ジャッジ時間 56,571 ms
ジャッジサーバーID
(参考情報)
judge4 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 28 WA * 70
権限があれば一括ダウンロードができます

ソースコード

diff #

class UnionFind:
    def __init__(self, n):
        self.table = [-1] * n

    def root(self, x):
        stack = []
        tbl = self.table
        while tbl[x] >= 0:
            stack.append(x)
            x = tbl[x]
        for y in stack:
            tbl[y] = x
        return x

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

    def unite(self, x, y):
        r1 = self.root(x)
        r2 = self.root(y)
        if r1 == r2:
            return False
        d1 = self.table[r1]
        d2 = self.table[r2]
        if d1 <= d2:
            self.table[r2] = r1
            self.table[r1] += d2
        else:
            self.table[r1] = r2
            self.table[r2] += d1
        return True

    def get_size(self, x):
        return -self.table[self.root(x)]


def solve(n):
    if n == 2:
        return [['#', '.'], ['.', '.']]

    if n % 2 == 0:
        ans2 = solve(n - 1)
        for i in range(n - 1):
            if ans2[0][i] == '#' or ans2[i][0] == '#':
                return []
        ans = [['#', '.'] * (n // 2)]
        for i in range(n - 1):
            ans.append(['.' if i % 2 == 0 else '#'] + ans2[i])
        return ans

    w = n // 2

    def get_key(i, j):
        return i // 2 * w + j // 2

    uft = UnionFind(w ** 2)
    ans = []
    for i in range(n):
        if i % 2 == 0:
            ans.append(['.'] * n)
        else:
            ans.append(['.', '#'] * w + ['.'])
    t = 2
    while t < n - 1:
        for i in range(t, n - 1, t * 2):
            for j in range(t, n - 1, t * 2):
                ans[i][j] = '#'
                uft.unite(get_key(i - 1, j - 1), get_key(i - 1, j + 1))
                uft.unite(get_key(i - 1, j - 1), get_key(i + 1, j - 1))
                uft.unite(get_key(i - 1, j - 1), get_key(i + 1, j + 1))
        t <<= 1

    ans[-1][-1] = '#'
    lb = get_key(n - 2, n - 2)
    for j in range(n - 4, 0, -2):
        key = get_key(n - 2, j)
        if uft.find(key, lb):
            continue
        ans[-1][j - 1] = '#'
        ans[j - 1][-1] = '#'
        uft.unite(key - 1, lb)

    return ans


n = int(input())
ans = solve(n)
if len(ans) == 0:
    print(-1)
else:
    for row in ans:
        print(''.join(row))
0