結果

問題 No.439 チワワのなる木
ユーザー nebukuro09nebukuro09
提出日時 2016-10-29 14:13:58
言語 D
(dmd 2.106.1)
結果
TLE  
実行時間 -
コード長 1,029 bytes
コンパイル時間 1,530 ms
コンパイル使用メモリ 87,220 KB
実行使用メモリ 18,496 KB
最終ジャッジ日時 2023-09-02 22:37:36
合計ジャッジ時間 8,198 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge11
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,372 KB
testcase_01 AC 1 ms
4,368 KB
testcase_02 AC 1 ms
4,368 KB
testcase_03 AC 2 ms
4,372 KB
testcase_04 AC 1 ms
4,372 KB
testcase_05 AC 2 ms
4,368 KB
testcase_06 AC 1 ms
4,372 KB
testcase_07 AC 2 ms
4,368 KB
testcase_08 AC 2 ms
4,372 KB
testcase_09 AC 1 ms
4,368 KB
testcase_10 AC 2 ms
4,368 KB
testcase_11 AC 1 ms
4,372 KB
testcase_12 AC 1 ms
4,372 KB
testcase_13 AC 2 ms
4,368 KB
testcase_14 AC 4 ms
4,368 KB
testcase_15 AC 25 ms
4,368 KB
testcase_16 AC 107 ms
4,368 KB
testcase_17 AC 59 ms
4,368 KB
testcase_18 TLE -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

import std.stdio;
import std.array;
import std.string;
import std.conv;
import std.algorithm;
import std.typecons;
import std.range;

string S;
int[][] rinsetsu;

long dfs(char c, int n, int prev) {
  long ret = 0;
  foreach (nn; rinsetsu[n])
    if (nn != prev)
      ret += dfs(c, nn, n);
  return S[n] == c ? ret+1 : ret;
}

void main() {
  int N = readln().chomp.to!int;
  S = readln().chomp;
  rinsetsu = new int[][](N);
  foreach (i; iota(N-1)) {
    auto input = readln().split.map!(to!int);
    rinsetsu[input[0]-1] ~= input[1]-1;
    rinsetsu[input[1]-1] ~= input[0]-1;
  }

  long ans = 0;
  foreach (root; iota(N)) {
    if (S[root] != 'w')
      continue;
    int len = rinsetsu[root].length.to!int;
    long[] c = new long[](len); 
    long[] w = new long[](len);
    foreach (i; iota(len)) {
      c[i] = dfs('c', rinsetsu[root][i], root);
      w[i] = dfs('w', rinsetsu[root][i], root);
    }
    foreach (i; iota(len))
      foreach (j; iota(i+1, len))
        ans += c[i]*w[j] + c[j]*w[i];
  }
  writeln(ans);
}
0