oj.uz

Submission #307026

# Submission time Handle Problem Language Result Execution time Memory
307026 2020-09-26T19:15:09 Z jovan_b Power Plant (JOI20_power) C++17
0 / 100
 4 ms 4992 KB 
#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
typedef long double ld;

string s = "0";
int res;
int dp[200005];
vector <int> graf[200005];

void dfs(int v, int par){
    int mx = 0;
    for(auto c : graf[v]){
        if(c == par) continue;
        dfs(c, v);
        dp[v] += dp[c];
        mx = max(mx, dp[c]);
    }
    if(s[v] == '1') res = max(res, 1+mx);
    else res = max(res, dp[v]);
    dp[v] -= s[v]-'0';
    if(dp[v] <= 0) dp[v] = 1;
    //cout << v << " " << dp[v] << endl;
    res = max(res, dp[v]);
}

int main(){
    ios_base::sync_with_stdio(false), cin.tie(0);
    cout.precision(10);
    cout << fixed;

    int n;
    cin >> n;
    for(int i=1; i<n; i++){
        int a, b;
        cin >> a >> b;
        graf[a].push_back(b);
        graf[b].push_back(a);
    }
    string g;
    cin >> g;
    s += g;
    dfs(1, 0);
    cout << res;
    return 0;
}

Subtask #1
0.0 / 6.0

# Verdict Execution time Memory Grader output
1 Correct 4 ms 4992 KB Output is correct
2 Incorrect 3 ms 4992 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #2
0.0 / 41.0

# Verdict Execution time Memory Grader output
1 Correct 4 ms 4992 KB Output is correct
2 Incorrect 3 ms 4992 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #3
0.0 / 53.0

# Verdict Execution time Memory Grader output
1 Correct 4 ms 4992 KB Output is correct
2 Incorrect 3 ms 4992 KB Output isn't correct
3 Halted 0 ms 0 KB -