結果

問題 No.1605 Matrix Shape
ユーザー hir355hir355
提出日時 2021-07-16 22:07:12
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,551 bytes
コンパイル時間 170 ms
コンパイル使用メモリ 82,428 KB
実行使用メモリ 127,700 KB
最終ジャッジ日時 2024-07-06 09:21:59
合計ジャッジ時間 10,527 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 33 WA * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

class UnionFind:
    def __init__(self, n):
        self.par = [i for i in range(n+1)]
        self.rank = [0] * (n + 1)
        self.size = [1] * (n + 1)
 
    def find(self, x):
        if self.par[x] == x:
            return x
        else:
            self.par[x] = self.find(self.par[x])
            return self.par[x]

    def unite(self, x, y):
        x = self.find(x)
        y = self.find(y)
        if x == y:
            return
        if self.rank[x] < self.rank[y]:
            self.par[x] = y
            self.size[y] += self.size[x]
        else:
            self.par[y] = x
            self.size[x] += self.size[y]
            if self.rank[x] == self.rank[y]:
                self.rank[x] += 1
 
    def same_check(self, x, y):
        return self.find(x) == self.find(y)

MAX = 2 * 10 ** 5
n = int(input())
g = [[] for _ in range(MAX + 1)]
inv = [0] * (MAX + 1)
out = [0] * (MAX + 1)
s = set()
uf = UnionFind(MAX + 1)
for i in range(n):
    h, w = map(int, input().split())
    uf.unite(h, w)
    s.add(h)
    s.add(w)
    g[h].append(w)
    inv[w] += 1
    out[h] += 1
f1 = 0
f2 = 0
v = [0] * (MAX + 1)
cnt = 0
for i in range(MAX + 1):
    j = uf.find(i)
    if uf.size[j] >= 2 and v[j] == 0:
        v[j] = 1
        cnt += 1
    if inv[i] + 1 == out[i]:
        f1 += 1
    elif out[i] + 1 == inv[i]:
        f2 += 1
    elif inv[i] == out[i]:
        continue
    else:
        print(0)
        exit(0)
if cnt >= 2:
    print(0)
    exit(0)
if f1 == f2 == 0:
    print(len(s))
elif f1 == f2 == 1:
    print(1)
else:
    print(0)
0