Submission #1296803

#TimeUsernameProblemLanguageResultExecution timeMemory
1296803SzymonKrzywdaSvjetlo (COCI20_svjetlo)C++20
0 / 110
2096 ms52944 KiB
#include <iostream>
#include <vector>
#include <cmath>
#include <algorithm>

using namespace std;


const int MAXN = 5e5 + 7;
vector<int> graf[MAXN];
int dp[MAXN][2][2]; //zmiana parent, koncze w v
bool color[MAXN];


void dfs(int v, int p) {

    if (v != p && graf[v].size() == 1) {
        if (color[v]) {
            dp[v][0][0] = 0;
            dp[v][0][1] = 0;
            dp[v][1][1] = 0;
            return;
        }

        dp[v][0][0] = 1;
        dp[v][0][1] = 1;
        dp[v][1][1] = 3;
        return;
    }

    int xorr = (graf[v].size()) % 2, optsuma = graf[v].size() + (v == p);
    if (v == p) xorr ^= 1;
    xorr ^= color[v];


    for (int s : graf[v]) {
        if (s != p) {
            dfs(s, v);
            if (dp[s][0][1] > dp[s][1][1]) xorr ^= 1;
            optsuma += min(dp[s][0][1], dp[s][1][1]);
        }
    }
    dp[v][0][0] = 1e9;
    dp[v][0][1] = 1e9;
    dp[v][1][1] = 1e9;

    for (int s : graf[v]) {
        if (s == p) continue;
        int oh = dp[s][0][1] > dp[s][1][1];
        int optsuma2 = optsuma - min(dp[s][0][1], dp[s][1][1]);
        //cout << "O: " << v << ' ' << ' ' <<  min(dp[s][0][1], dp[s][1][1]) <<  ' ' << optsuma << ' ' << optsuma2 << '\n';
        dp[v][0][0] = min(dp[v][0][0], optsuma2 + dp[s][0][0] + (xorr ^ oh) * (int)1e7 - 1);
    }
    if (!xorr) dp[v][1][1] = optsuma + 3;
    else dp[v][0][1] = optsuma;
    dp[v][0][0] = min(dp[v][0][0], dp[v][0][1]);

    //cout << v << ": " << dp[v][0][0] << ' ' << dp[v][0][1] << ' ' << dp[v][1][1] << ' ' << optsuma << ' ' << xorr << '\n';

}

int main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    int n, a, b;
    cin >> n;

    char c;
    for (int i = 1; i <= n; i++) {
        cin >> c;
        color[i] = c - '0';
    }

    for (int i = 0; i < n - 1; i++) {
        cin >> a >> b;
        graf[a].push_back(b);
        graf[b].push_back(a);
    }

    
    int ans = 1e9;
    for (int i = 1; i <= n; i++) {
        dfs(i, i);
       // cout << dp[i][0][0] << ' ' << dp[i][0][1] << ' ' << dp[i][1][1] << '\n';
        ans = min({ans, dp[i][0][0], dp[i][0][1], dp[i][1][1]});
    } 


    cout << ans << '\n';



    return 0;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...