結果

問題 No.439 チワワのなる木
ユーザー te-shte-sh
提出日時 2017-10-30 16:02:46
言語 D
(dmd 2.106.1)
結果
WA  
実行時間 -
コード長 1,232 bytes
コンパイル時間 726 ms
コンパイル使用メモリ 106,172 KB
実行使用メモリ 13,500 KB
最終ジャッジ日時 2024-06-12 22:16:25
合計ジャッジ時間 2,530 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
6,816 KB
testcase_01 AC 1 ms
6,940 KB
testcase_02 AC 1 ms
6,940 KB
testcase_03 AC 1 ms
6,944 KB
testcase_04 AC 1 ms
6,940 KB
testcase_05 AC 1 ms
6,940 KB
testcase_06 AC 1 ms
6,940 KB
testcase_07 AC 1 ms
6,944 KB
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 AC 90 ms
12,792 KB
testcase_25 AC 82 ms
13,500 KB
testcase_26 WA -
testcase_27 AC 67 ms
12,652 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import std.algorithm, std.conv, std.range, std.stdio, std.string;
import std.container; // SList, DList, BinaryHeap

void main()
{
  auto n = readln.chomp.to!int;
  auto s = readln.chomp;

  auto children = new int[][](n);
  foreach (_; 0..n-1) {
    auto rd = readln.splitter;
    auto a = rd.front.to!int-1; rd.popFront();
    auto b = rd.front.to!int-1;
    children[a] ~= b;
    children[b] ~= a;
  }

  auto parent = new int[](n);
  parent[0] = -1;

  auto q = DList!int(0), st = SList!int();
  while (!q.empty) {
    auto u = q.front; q.removeFront();
    st.insertFront(u);
    foreach (v; children[u])
      if (v != parent[u]) {
        parent[v] = u;
        q.insertBack(v);
      }
  }

  auto dc = new int[](n), dw = new int[](n);

  while (!st.empty) {
    auto u = st.front; st.removeFront();
    dc[u] = s[u] == 'c';
    dw[u] = s[u] == 'w';
    foreach (v; children[u])
      if (v != parent[u]) {
        dc[u] += dc[v];
        dw[u] += dw[v];
      }
  }

  auto ans = 0L;
  foreach (u; 0..n)
    if (s[u] == 'w') {
      foreach (v; children[u])
        if (v != parent[u]) {
          ans += long(dc[v]) * (dw[0] - dw[v] - 1);
          ans += long(dw[v]) * (dc[0] - dc[v]);
        }
    }

  writeln(ans);
}
0