結果

問題 No.439 チワワのなる木
ユーザー H3PO4H3PO4
提出日時 2021-07-10 09:35:47
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
AC  
実行時間 652 ms / 5,000 ms
コード長 984 bytes
コンパイル時間 105 ms
コンパイル使用メモリ 12,928 KB
実行使用メモリ 37,632 KB
最終ジャッジ日時 2024-07-02 01:45:44
合計ジャッジ時間 6,693 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 31 ms
10,880 KB
testcase_01 AC 31 ms
10,880 KB
testcase_02 AC 30 ms
10,624 KB
testcase_03 AC 31 ms
10,752 KB
testcase_04 AC 31 ms
10,752 KB
testcase_05 AC 30 ms
10,624 KB
testcase_06 AC 31 ms
10,880 KB
testcase_07 AC 31 ms
10,752 KB
testcase_08 AC 32 ms
10,752 KB
testcase_09 AC 31 ms
10,752 KB
testcase_10 AC 31 ms
10,752 KB
testcase_11 AC 31 ms
10,880 KB
testcase_12 AC 31 ms
10,624 KB
testcase_13 AC 31 ms
10,752 KB
testcase_14 AC 33 ms
10,752 KB
testcase_15 AC 35 ms
10,880 KB
testcase_16 AC 39 ms
11,008 KB
testcase_17 AC 38 ms
10,880 KB
testcase_18 AC 448 ms
25,600 KB
testcase_19 AC 411 ms
25,216 KB
testcase_20 AC 590 ms
30,336 KB
testcase_21 AC 176 ms
17,152 KB
testcase_22 AC 165 ms
16,000 KB
testcase_23 AC 652 ms
32,128 KB
testcase_24 AC 609 ms
34,048 KB
testcase_25 AC 613 ms
30,976 KB
testcase_26 AC 532 ms
29,952 KB
testcase_27 AC 449 ms
37,632 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
from collections import deque

input = sys.stdin.readline

N = int(input())
S = input()
T = [[] for _ in range(N)]
for _ in range(N - 1):
    a, b = (int(x) - 1 for x in input().split())
    T[a].append(b)
    T[b].append(a)

# dfs
dfs_order = []
parent = [None] * N
d = deque([0])
nonvisited = [True] * N
nonvisited[0] = False
while d:
    v = d.pop()
    dfs_order.append(v)
    for x in T[v]:
        if nonvisited[x]:
            d.append(x)
            parent[x] = v
            nonvisited[x] = False

Csum, Wsum = S.count('c'), S.count('w')

# 子から親へ
C, W = [0] * N, [0] * N
ans = 0
for v in reversed(dfs_order):
    if S[v] == 'c':
        C[v] += 1
    else:  # S[v] == 'w'
        W[v] += 1
    for child in T[v]:
        if child == parent[v]:
            continue
        C[v] += C[child]
        W[v] += W[child]
        if S[v] == 'w':
            ans += W[child] * (Csum - C[child])
    if S[v] == 'w':
        ans += C[v] * (Wsum - W[v])
print(ans)
0