結果

問題 No.439 チワワのなる木
ユーザー mio_hmio_h
提出日時 2016-10-29 00:10:39
言語 C++11
(gcc 11.4.0)
結果
WA  
実行時間 -
コード長 2,228 bytes
コンパイル時間 1,736 ms
コンパイル使用メモリ 152,004 KB
実行使用メモリ 24,232 KB
最終ジャッジ日時 2023-08-15 21:55:28
合計ジャッジ時間 3,319 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge13
このコードへのチャレンジ
（要ログイン）

テストケース

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

ソースコード

diff # 

class in{struct myIterator{int it;const bool rev;explicit constexpr myIterator(int it_, bool rev=false):it(it_),rev(rev){}int operator*(){return it;}bool operator!=(myIterator& r){return it!=r.it;}void operator++(){rev?--it:++it;}};const myIterator i,n;public:explicit constexpr in(int n):i(0),n(n){}explicit constexpr in(int i,int n):i(i,n<i),n(n){}const myIterator& begin(){return i;}const myIterator& end(){return n;}};

#include <bits/stdc++.h>
using namespace std;
// input
int n;
string s;
vector<vector<int>> adj;
// end input
using i64 = long long;
struct data {
    i64 c, w, ww, wc, cww, wwc;
    explicit data() : c(0LL), w(0LL), ww(0LL), wc(0LL), cww(0LL), wwc(0LL){} // 下から上
    void print() { cout << c << ' ' << w << ' ' << ww << ' ' << wc << ' ' << cww << ' ' << wwc << endl;}
};

inline void merge(bool c, data& l, const data& r) {
    l.c += r.c;
    l.w += r.w;
    l.ww += r.ww + (!c ? r.w : 0LL);
    l.wc += r.wc + (!c ? r.c : 0LL);
    l.cww += r.cww + (c ? r.ww: 0LL);
    l.wwc += r.wwc + (!c ? r.wc : 0LL);
}
inline void add(data& l, const data& r) {
    l.c += r.c;
    l.w += r.w;
    l.ww += r.ww;
    l.wc += r.wc;
    l.cww += r.cww;
    l.wwc += r.wwc;
}
inline data sa(const data& l_, const data& r) {
    data l = l_;
    l.c -= r.c;
    l.w -= r.w;
    l.ww -= r.ww;
    l.wc -= r.wc;
    l.cww -= r.cww;
    l.wwc -= r.wwc;
    return l;
}
data ans;
data dfs(int cur, int par) {
    data res, sum;
    bool c = s[cur] == 'c';
    if(c) res.c++; else res.w++;
    add(sum, res);
    vector<data> dat;
    for(int nxt : adj[cur]) {
        if(nxt == par) continue;
        auto ret = dfs(nxt, cur);
        merge(c, res, ret);
        dat.emplace_back(ret);
        add(sum, ret);
    }
    /*
    for(auto d : dat) {
        data r = sa(sum, d);
        ans.cww += d.c * r.ww;
        ans.cww += d.cw 
    }
      */
    return res;
}

int main() {
    cin.tie(nullptr);
    ios::sync_with_stdio(false);
    cin >> n;
    cin >> s;
    adj.resize(n);
    for(int _ : in(n - 1)) {
        int a, b;
        cin >> a >> b;
        --a; --b;
        adj[a].emplace_back(b);
        adj[b].emplace_back(a);
    }
    auto result = dfs(0, -1);
    cout << result.cww + result.wwc << endl;
}
0