結果

問題 No.439 チワワのなる木
コンテスト
ユーザー nebukuro09
提出日時 2016-10-29 14:13:58
言語 D
(dmd 2.112.0)
コンパイル:
dmd -fPIE -m64 -w -wi -O -release -inline -I/opt/dmd/src/druntime/import/ -I/opt/dmd/src/phobos -L-L/opt/dmd/linux/lib64/ -fPIC _filename_
実行:
./Main
結果
TLE  
実行時間 -
コード長 1,029 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 577 ms
コンパイル使用メモリ 85,288 KB
実行使用メモリ 30,160 KB
最終ジャッジ日時 2026-03-05 08:10:40
合計ジャッジ時間 55,942 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 19 TLE * 9
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

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