結果

問題 No.439 チワワのなる木
コンテスト
ユーザー nebukuro09
提出日時 2017-04-04 17:17:17
言語 D
(dmd 2.109.1)
結果
AC  
実行時間 115 ms / 5,000 ms
コード長 1,409 bytes
コンパイル時間 625 ms
コンパイル使用メモリ 121,020 KB
実行使用メモリ 26,308 KB
最終ジャッジ日時 2024-06-12 18:35:33
合計ジャッジ時間 2,736 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

import std.stdio, std.array, std.string, std.conv, std.algorithm;
import std.typecons, std.range, std.random, std.math, std.container;
import std.numeric, std.bigint, core.bitop;

int N;
string S;
int[][] adj;
long allC, allW;
long[] subC, subW;


Tuple!(long, long) dfs1(int n, int prev) {
    subC[n] += (S[n] == 'c');
    subW[n] += (S[n] == 'w');
    foreach (m; adj[n]) {
        if (m == prev) continue;
        auto t = dfs1(m, n);
        subC[n] += t[0];
        subW[n] += t[1];
    }
    return tuple(subC[n], subW[n]);
}

long dfs2(int n, int prev) {
    long ret = 0;
    if (S[n] == 'w') {
        foreach (m; adj[n]) {
            if (m == prev) {
                auto c = allC - subC[n];
                auto w = subW[n] - 1;
                ret += c * w;
            }
            else {
                ret += subC[m] * (allW-subW[m]-1);
            }
        }
    }

    foreach (m; adj[n]) {
        if (m == prev) continue;
        ret += dfs2(m, n);
    }

    return ret;
}


void main() {
    N = readln.chomp.to!int;
    S = readln.chomp;
    adj = new int[][](N);
    foreach (i; 0..N-1) {
        auto s = readln.split.map!(to!int);
        adj[s[0]-1] ~= s[1] - 1;
        adj[s[1]-1] ~= s[0] - 1;
    }
    subC = new long[](N);
    subW = new long[](N);
    fill(subC, 0);
    fill(subW, 0);

    dfs1(0, -1);
    allC = subC[0];
    allW = subW[0];

    dfs2(0, -1).writeln;
}
0