結果

問題 No.1605 Matrix Shape
ユーザー rin204
提出日時 2021-07-16 22:04:43
言語 PyPy3
(7.3.15)
結果
RE  
実行時間 -
コード長 1,742 bytes
コンパイル時間 210 ms
コンパイル使用メモリ 82,424 KB
実行使用メモリ 102,812 KB
最終ジャッジ日時 2024-07-06 09:16:57
合計ジャッジ時間 6,086 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1 RE * 2
other AC * 12 WA * 17 RE * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

n = int(input())
max_ = 2 * 10 ** 5
edges = [[] for _ in range(max_)]
in_cnt = [0] * max_
out_cnt = [0] * max_
for _ in range(n):
    h, w = map(int, input().split())
    h -= 1
    w -= 1
    edges[h].append(w)
    in_cnt[h] += 1
    out_cnt[w] += 1
    
s = -1
g = -1

for j, (i, o) in enumerate(zip(in_cnt, out_cnt)):
    if i == o:
        continue
    if i == o + 1:
        if s == -1:
            s = j
        else:
            print(0)
            exit()
    elif i + 1 == o:
        if g == -1:
            g = j
        else:
            print(0)
            exit()
    else:
        print(0)
        exit()

stack = [s]
used = [False] * n
used[s] = True
while stack:
    pos = stack.pop()
    for npos in edges[pos]:
        if used[npos]:
            continue
        used[npos] = True
        
for i in range(max_):
    if not used[i] and in_cnt[i] != 0:
        print(0)
        exit()

if s == g == -1:
    stack = [h]
    used = [False] * n
    used[h] = True
    while stack:
        pos = stack.pop()
        for npos in edges[pos]:
            if used[npos]:
                continue
            used[npos] = True
            
    for i in range(max_):
        if not used[i] and in_cnt[i] != 0:
            print(0)
            exit()
    
    ans = 0
    for e in edges:
        ans += len(set(e))
    print(ans)
    exit()

cnt = 0
used = [False] * n
stack = [s]
used[s] = True
while stack:
    pos = stack.pop()
    for npos in edges[pos]:
        if used[npos]:
            continue
        if npos == g:
            cnt += 1
            continue
        used[npos] = True
        stack.append(npos)

l = 0
for e in edges:
    if g in e: l += 1

if cnt == 1:
    print(max(1, l - 1))
else:
    print(l)
0