結果

問題 No.1928 Make a Binary Tree
ユーザー tnodino
提出日時 2022-05-06 22:45:44
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 757 bytes
コンパイル時間 242 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 78,208 KB
最終ジャッジ日時 2024-07-06 00:09:31
合計ジャッジ時間 18,466 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 4 WA * 29 TLE * 1 -- * 23
権限があれば一括ダウンロードができます

ソースコード

diff # 

from sys import setrecursionlimit
setrecursionlimit(10**6)

def dfs1(pos):
    ret = 1
    for nxt in G1[pos]:
        ret += dfs1(nxt)
    Dist[pos] = ret
    return ret

N = int(input())
G1 = [[] for _ in range(N)]
G2 = [-1] * N
for _ in range(N-1):
    x,y = map(int,input().split())
    x -= 1
    y -= 1
    G1[x].append(y)
    G2[y] = x
Dist = [-1] * N
dfs1(0)
for i in range(N):
    Dist[i] = [Dist[i], i]
Dist.sort(reverse=True)
Cnt = [0] * N
Flg = [1] * N
for i in range(1,N):
    pos = G2[i]
    Queue = [pos]
    while pos != -1:
        if Cnt[pos] < 2:
            Cnt[pos] += 1
            break
        pos = G2[pos]
        Queue.append(pos)
    if pos == -1:
        for k in Queue:
            G2[k] = -1
        Flg[i] = 0
print(sum(Flg))
0