結果

問題 No.439 チワワのなる木
ユーザー parukiparuki
提出日時 2016-10-29 01:20:06
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 71 ms / 5,000 ms
コード長 2,240 bytes
コンパイル時間 1,673 ms
コンパイル使用メモリ 178,864 KB
実行使用メモリ 25,132 KB
最終ジャッジ日時 2024-11-24 07:02:22
合計ジャッジ時間 3,072 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 0 ms
6,820 KB
testcase_01 AC 2 ms
6,816 KB
testcase_02 AC 2 ms
6,816 KB
testcase_03 AC 1 ms
6,820 KB
testcase_04 AC 2 ms
6,820 KB
testcase_05 AC 2 ms
6,816 KB
testcase_06 AC 2 ms
6,816 KB
testcase_07 AC 2 ms
6,820 KB
testcase_08 AC 1 ms
6,820 KB
testcase_09 AC 1 ms
6,816 KB
testcase_10 AC 2 ms
6,816 KB
testcase_11 AC 2 ms
6,820 KB
testcase_12 AC 2 ms
6,816 KB
testcase_13 AC 2 ms
6,820 KB
testcase_14 AC 2 ms
6,820 KB
testcase_15 AC 2 ms
6,816 KB
testcase_16 AC 3 ms
6,820 KB
testcase_17 AC 2 ms
6,820 KB
testcase_18 AC 42 ms
11,572 KB
testcase_19 AC 36 ms
10,916 KB
testcase_20 AC 55 ms
14,172 KB
testcase_21 AC 15 ms
6,816 KB
testcase_22 AC 16 ms
7,716 KB
testcase_23 AC 60 ms
15,772 KB
testcase_24 AC 71 ms
23,368 KB
testcase_25 AC 43 ms
13,568 KB
testcase_26 AC 41 ms
13,488 KB
testcase_27 AC 52 ms
25,132 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"
using namespace std;
#define FOR(i,j,k) for(int (i)=(j);(i)<(int)(k);++(i))
#define rep(i,j) FOR(i,0,j)
#define each(x,y) for(auto &(x):(y))
#define mp make_pair
#define all(x) (x).begin(),(x).end()
#define debug(x) cout<<#x<<": "<<(x)<<endl
#define smax(x,y) (x)=max((x),(y))
#define smin(x,y) (x)=min((x),(y))
#define MEM(x,y) memset((x),(y),sizeof (x))
#define sz(x) (int)(x).size()
typedef long long ll;
typedef pair<int, int> pii;
typedef vector<int> vi;
typedef vector<ll> vll;

vector<vector<int> > makeTree(const vector<int> &u, const vector<int> &v) {
    int V = (int)u.size() + 1;
    vector<vector<int> > tree(V), graph(V);
    for(int i = 0; i < V - 1; ++i) {
        graph[u[i]].push_back(v[i]);
        graph[v[i]].push_back(u[i]);
    }
    queue<pair<int, int> > Q;
    Q.push(make_pair(0, -1));
    while(!Q.empty()) {
        int cur, pre;
        tie(cur, pre) = Q.front();
        Q.pop();
        for(auto &to : graph[cur]) if(to != pre) {
            tree[cur].push_back(to);
            Q.push(make_pair(to, cur));
        }
    }
    return tree;
}

const int MA = 100010;
int N;
char S[100010];
vector<vi> G;
ll dp[100010][4], ww[MA], wc[MA], nw[MA], nc[MA];

ll dfs(int u) {
    if(S[u] == 'w')nw[u]++;
    if(S[u] == 'c')nc[u]++;

    ll res = 0;
    each(v, G[u]) {
        res += dfs(v);
        nw[u] += nw[v];
        nc[u] += nc[v];
        ww[u] += ww[v];
        wc[u] += wc[v];
        if(S[u] == 'w') {
            ww[u] += nw[v];
            wc[u] += nc[v];
        }
    }

    each(v, G[u]) {
        if(S[u]=='c') res += ww[v];
        else if(S[u] == 'w')res += wc[v];
    }

    ll sumW = nw[u] - !!(S[u] == 'w');
    ll sumC = nc[u] - !!(S[u] == 'c');
    if(S[u]=='w')each(v, G[u]) {
        ll numW = sumW - nw[v], numC = nc[v];
        res += numW*numC;
    }
    each(v, G[u]) {
        res += (sumW - nw[v])*wc[v];
    }
    each(v, G[u]) {
        res += ww[v] * (sumC-nc[v]);
    }
    return res;
}

int main(){
    cin >> N;
    scanf("%s", S);
    vi A(N - 1), B(N - 1);
    rep(i, N - 1) {
        int a, b;
        scanf("%d%d", &a, &b);
        A[i] = a - 1;
        B[i] = b - 1;
    }
    G = makeTree(A, B);
    ll ans = dfs(0);
    cout << ans << endl;
}
0