結果

問題 No.2892 Lime and Karin
ユーザー noiminoimi
提出日時 2024-09-13 23:50:48
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 140 ms / 8,000 ms
コード長 3,574 bytes
コンパイル時間 2,405 ms
コンパイル使用メモリ 210,400 KB
実行使用メモリ 27,048 KB
最終ジャッジ日時 2024-09-13 23:50:56
合計ジャッジ時間 7,489 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge6
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,812 KB
testcase_01 AC 2 ms
6,816 KB
testcase_02 AC 2 ms
6,812 KB
testcase_03 AC 2 ms
6,816 KB
testcase_04 AC 2 ms
6,940 KB
testcase_05 AC 2 ms
6,940 KB
testcase_06 AC 2 ms
6,940 KB
testcase_07 AC 2 ms
6,940 KB
testcase_08 AC 2 ms
6,940 KB
testcase_09 AC 2 ms
6,940 KB
testcase_10 AC 2 ms
6,944 KB
testcase_11 AC 2 ms
6,944 KB
testcase_12 AC 2 ms
6,940 KB
testcase_13 AC 2 ms
6,944 KB
testcase_14 AC 45 ms
8,320 KB
testcase_15 AC 41 ms
8,064 KB
testcase_16 AC 74 ms
11,008 KB
testcase_17 AC 17 ms
6,944 KB
testcase_18 AC 73 ms
10,880 KB
testcase_19 AC 105 ms
13,184 KB
testcase_20 AC 7 ms
6,944 KB
testcase_21 AC 63 ms
9,088 KB
testcase_22 AC 69 ms
10,880 KB
testcase_23 AC 27 ms
6,944 KB
testcase_24 AC 105 ms
13,440 KB
testcase_25 AC 140 ms
13,440 KB
testcase_26 AC 111 ms
13,312 KB
testcase_27 AC 113 ms
13,440 KB
testcase_28 AC 110 ms
13,440 KB
testcase_29 AC 118 ms
13,464 KB
testcase_30 AC 109 ms
13,420 KB
testcase_31 AC 113 ms
13,400 KB
testcase_32 AC 131 ms
13,440 KB
testcase_33 AC 112 ms
13,440 KB
testcase_34 AC 44 ms
13,844 KB
testcase_35 AC 44 ms
13,964 KB
testcase_36 AC 45 ms
13,872 KB
testcase_37 AC 48 ms
13,844 KB
testcase_38 AC 46 ms
13,836 KB
testcase_39 AC 88 ms
20,480 KB
testcase_40 AC 90 ms
26,380 KB
testcase_41 AC 85 ms
21,376 KB
testcase_42 AC 90 ms
27,048 KB
testcase_43 AC 86 ms
23,168 KB
testcase_44 AC 37 ms
7,296 KB
testcase_45 AC 119 ms
12,396 KB
testcase_46 AC 44 ms
7,936 KB
testcase_47 AC 59 ms
9,344 KB
testcase_48 AC 25 ms
6,944 KB
testcase_49 AC 75 ms
10,368 KB
testcase_50 AC 78 ms
13,560 KB
testcase_51 AC 76 ms
13,696 KB
testcase_52 AC 75 ms
13,696 KB
testcase_53 AC 75 ms
13,616 KB
testcase_54 AC 76 ms
13,696 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define pii pair<int, int>
#define pll pair<ll, ll>
#define vi vector<int>
#define vl vector<ll>
#define ov4(a, b, c, d, name, ...) name
#define rep3(i, a, b, c) for(ll i = (a); i < (b); i += (c))
#define rep2(i, a, b) rep3(i, a, b, 1)
#define rep1(i, n) rep2(i, 0, n)
#define rep0(n) rep1(aaaaa, n)
#define rep(...) ov4(__VA_ARGS__, rep3, rep2, rep1, rep0)(__VA_ARGS__)
#define per(i, a, b) for(ll i = (a) - 1; i >= (b); i--)
#define fore(e, v) for(auto &&e : v)
#define all(a) begin(a), end(a)
#define si(a) (int)(size(a))
#define lb(v, x) (lower_bound(all(v), x) - begin(v))
#define eb emplace_back

template <typename T, typename S> bool chmin(T &a, const S &b) { return a > b ? a = b, 1 : 0; }
template <typename T, typename S> bool chmax(T &a, const S &b) { return a < b ? a = b, 1 : 0; }

const int INF = 1e9 + 100;
const ll INFL = 3e18 + 100;

#define i128 __int128_t

struct _ {
    _() { cin.tie(0)->sync_with_stdio(0), cout.tie(0); }
} __;

struct BIT {
    vi a;
    BIT(int n) : a(n + 1) {}
    void add(int i, int x) {
        i++;
        while(i < si(a)) a[i] += x, i += i & -i;
    }
    int sum(int r) {
        int s = 0;
        while(r) s += a[r], r -= r & -r;
        return s;
    }
    int sum(int l, int r) { return sum(r) - sum(l); }
};

int main() {
    int n;
    cin >> n;
    vector<vi> g(n);
    rep(n - 1) {
        int x, y;
        cin >> x >> y;
        --x, --y;
        g[x].eb(y), g[y].eb(x);
    }

    string s;
    cin >> s;
    vi a(n);
    rep(i, n) a[i] = (s[i] == '1' ? 1 : -1);

    if(n == 1) {
        cout << count(all(a), 1) << endl;
        exit(0);
    }

    vi sub(n), v(n);
    auto dfs = [&](auto &&f, int x, int p) -> void {
        v[x] += a[x];
        sub[x]++;
        if(g[x][0] == p) swap(g[x][0], g[x].back());
        fore(e, g[x]) {
            if(e == p) continue;
            v[e] = v[x];
            f(f, e, x);
            sub[x] += sub[e];
            if(sub[e] > sub[g[x][0]]) swap(e, g[x][0]);
        }
        return;
    };
    dfs(dfs, 0, -1);

    vi in(n), out(n), tour(n);
    int idx = 0;
    {
        auto dfs = [&](auto &&f, int x, int p) -> void {
            tour[idx] = x;
            in[x] = idx++;
            out[x] = in[x] + sub[x];
            fore(e, g[x]) if(e != p) f(f, e, x);
        };
        dfs(dfs, 0, -1);
    }

    ll ans = count(all(a), 1);
    BIT bit(n * 4 + 1);

    int car = n * 2;

    vi tmp;
    // rep(i, n) cout << v[i] << " ";
    // cout << endl;

    int S = 0;
    auto add = [&](int i) {
        S++;
        bit.add(i + car, 1);
        tmp.eb(i + car);
    };
    auto dfs2 = [&](auto &&f, int x, int p, bool clear) -> void {
        rep(i, 1, si(g[x])) {
            int e = g[x][i];
            if(e == p) continue;
            f(f, e, x, true);
        }
        if(g[x][0] != p) f(f, g[x][0], x, false);

        ll tans = 0;
        tans += S - bit.sum(v[x] - a[x] + car + 1);
        add(v[x]);
        rep(i, 1, si(g[x])) {
            int e = g[x][i];
            if(e == p) continue;
            rep(j, in[e], out[e]) {
                int t = tour[j];
                int tar = 2 * v[x] - a[x] - v[t];
                tans += S - bit.sum(tar + car + 1);
            }
            rep(j, in[e], out[e]) add(v[tour[j]]);
        }
        // cout << x << ": " << tans << endl;

        ans += tans;
        if(clear) {
            fore(i, tmp) bit.add(i, -1);
            tmp.clear();
            S = 0;
        }
    };
    dfs2(dfs2, 0, -1, true);
    cout << ans << endl;
}
0