結果

問題 No.3373 Partial Complement Tree
コンテスト
ユーザー 👑 loop0919
提出日時 2025-10-21 12:59:43
言語 PyPy3
(7.3.15)
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 856 bytes
コンパイル時間 675 ms
コンパイル使用メモリ 81,396 KB
実行使用メモリ 82,160 KB
最終ジャッジ日時 2025-11-21 20:51:39
合計ジャッジ時間 10,110 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 4 TLE * 2 -- * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import deque

INF = 10**9
MOD = 998244353


def solve():
    N = int(input())
    graph = [[] for _ in range(N)]

    for _ in range(N - 1):
        U, V = [int(x) - 1 for x in input().split()]
        graph[U].append(V)
        graph[V].append(U)

    def bfs(start):
        dist = [INF] * N
        dist[start] = 0

        que = deque([start])

        while len(que) > 0:
            curr = que.popleft()

            for to in graph[curr]:
                if dist[to] < INF:
                    continue
                dist[to] = dist[curr] + 1
                que.append(to)

        return len([i for i in range(N) if dist[i] == 3])

    ans = 0
    for i in range(N):
        ans += bfs(i)
        ans %= MOD

    ans //= 2

    print(ans)


if __name__ == "__main__":
    T = int(input())
    for _ in range(T):
        solve()
0