제출 #561787

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
561787		4fecta	Power Plant (JOI20_power)	C++17	0 / 100	4 ms	4952 KiB

power

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define int ll
#define ld long double
#define pii pair<int, int>
#define f first
#define s second
#define boost() cin.tie(0), cin.sync_with_stdio(0)

const int MN = 200005;

int n, u, v, p[MN], dp[MN], ans;
char c;
vector<int> a[MN];

void dfs(int cur, int par) {
    int sum = 0;
    for (int nxt : a[cur]) {
        if (nxt == par) continue;
        dfs(nxt, cur);
        sum += dp[nxt];
    }
    if (p[cur]) dp[cur] = max(1ll, sum - 1);
    else dp[cur] = sum;
}

int32_t main() {
    boost();

    cin >> n;
    for (int i = 1; i < n; i++) {
        cin >> u >> v;
        a[u].push_back(v);
        a[v].push_back(u);
    }
    for (int i = 1; i <= n; i++) cin >> c, p[i] = c - '0';
    for (int i = 1; i <= n; i++) {
        dfs(i, 0);
        ans = max(ans, dp[i]);
    }
    printf("%lld\n", ans);

    return 0;
}

• Subtask 1: 0 / 6
• Subtask 2: 0 / 41
• Subtask 3: 0 / 53