結果

問題 No.439 チワワのなる木
ユーザー hos.lyric
提出日時 2019-05-24 23:57:46
言語 D
(dmd 2.083.0)
結果
AC  
実行時間 291 ms
コード長 3,015 Byte
コンパイル時間 2,136 ms
使用メモリ 49,768 KB
最終ジャッジ日時 2019-05-24 23:57:55

テストケース

テストケース表示
入力 結果 実行時間
使用メモリ
in01.txt AC 2 ms
6,868 KB
in02.txt AC 2 ms
6,872 KB
in03.txt AC 3 ms
6,868 KB
in04.txt AC 2 ms
6,872 KB
in05.txt AC 2 ms
6,872 KB
in06.txt AC 2 ms
6,872 KB
in07.txt AC 2 ms
6,872 KB
in08.txt AC 2 ms
6,872 KB
r01.txt AC 3 ms
6,872 KB
r02.txt AC 3 ms
6,868 KB
r03.txt AC 2 ms
6,868 KB
r04.txt AC 2 ms
6,872 KB
r05.txt AC 3 ms
6,872 KB
r06.txt AC 2 ms
6,872 KB
r07.txt AC 2 ms
6,872 KB
r08.txt AC 5 ms
6,868 KB
r09.txt AC 7 ms
6,872 KB
r10.txt AC 7 ms
6,872 KB
r11.txt AC 192 ms
17,640 KB
r12.txt AC 205 ms
14,608 KB
r13.txt AC 267 ms
26,356 KB
r14.txt AC 69 ms
11,436 KB
r15.txt AC 62 ms
10,892 KB
z01.txt AC 281 ms
30,764 KB
z02.txt AC 291 ms
44,032 KB
z03.txt AC 282 ms
27,416 KB
z04.txt AC 226 ms
28,668 KB
z05.txt AC 178 ms
49,768 KB
テストケース一括ダウンロード

ソースコード

diff #
import std.conv, std.functional, std.range, std.stdio, std.string;
import std.algorithm, std.array, std.bigint, std.complex, std.container, std.math, std.numeric, std.regex, std.typecons;
import core.bitop;

class EOFException : Throwable { this() { super("EOF"); } }
string[] tokens;
string readToken() { for (; tokens.empty; ) { if (stdin.eof) { throw new EOFException; } tokens = readln.split; } auto token = tokens.front; tokens.popFront; return token; }
int readInt() { return readToken.to!int; }
long readLong() { return readToken.to!long; }
real readReal() { return readToken.to!real; }

bool chmin(T)(ref T t, in T f) { if (t > f) { t = f; return true; } else { return false; } }
bool chmax(T)(ref T t, in T f) { if (t < f) { t = f; return true; } else { return false; } }

int binarySearch(alias pred, T)(in T[] as) { int lo = -1, hi = cast(int)(as.length); for (; lo + 1 < hi; ) { const mid = (lo + hi) >> 1; (unaryFun!pred(as[mid]) ? hi : lo) = mid; } return hi; }
int lowerBound(T)(in T[] as, T val) { return as.binarySearch!(a => (a >= val)); }
int upperBound(T)(in T[] as, T val) { return as.binarySearch!(a => (a > val)); }


int N;
string S;
int[] A, B;

int[][] G;
int[] par;
long[][] dp, DP;

void dfs0(int u, int p) {
  par[u] = p;
  foreach (v; G[u]) {
    if (v != p) {
      dfs0(v, u);
    }
  }
  dp[u] = [0, 0];
  ++dp[u]["cw".indexOf(S[u])];
  foreach (v; G[u]) {
    if (v != p) {
      dp[u][] += dp[v][];
    }
  }
}

void dfs1(int u, int p) {
  long[] sum = [0, 0];
  ++sum["cw".indexOf(S[u])];
  if (p != -1) {
    sum[] += DP[u][];
  }
  foreach (v; G[u]) {
    if (v != p) {
      sum[] += dp[v][];
    }
  }
  foreach (v; G[u]) {
    if (v != p) {
      DP[v][] = sum[] - dp[v][];
    }
  }
  foreach (v; G[u]) {
    if (v != p) {
      dfs1(v, u);
    }
  }
}

void main() {
  try {
    for (; ; ) {
      N = readInt();
      S = readToken();
      A = new int[N - 1];
      B = new int[N - 1];
      foreach (i; 0 .. N - 1) {
        A[i] = readInt() - 1;
        B[i] = readInt() - 1;
      }
      
      G = new int[][N];
      foreach (i; 0 .. N - 1) {
        G[A[i]] ~= B[i];
        G[B[i]] ~= A[i];
      }
      par = new int[N];
      dp = new long[][](N, 2);
      DP = new long[][](N, 2);
      const rt = 0;
      dfs0(rt, -1);
      dfs1(rt, -1);
      debug {
        writeln("rt = ", rt);
        writeln("dp = ", dp);
        writeln("DP = ", DP);
      }
      long ans;
      foreach (u; 0 .. N) {
        if (S[u] == 'w') {
          long[] sum = [0, 0];
          if (par[u] != -1) {
            sum[] += DP[u][];
          }
          foreach (v; G[u]) {
            if (v != par[u]) {
              sum[] += dp[v][];
            }
          }
          if (par[u] != -1) {
            ans += DP[u][0] * (sum[1] - DP[u][1]);
          }
          foreach (v; G[u]) {
            if (v != par[u]) {
              ans += dp[v][0] * (sum[1] - dp[v][1]);
            }
          }
        }
      }
      writeln(ans);
    }
  } catch (EOFException e) {
  }
}
0