oj.uz

Submission #847433

# Submission time Handle Problem Language Result Execution time Memory
847433 2023-09-09T16:12:50 Z MinaRagy06 The Xana coup (BOI21_xanadu) C++17
10 / 100
 77 ms 28012 KB 
#include <bits/stdc++.h>
using namespace std;
typedef int64_t ll;

const int N = 100'005;
int n, dp[N][2][2], a[N];
vector<int> adj[N];
int dfs(int i, int par, int parf, int curf) {
    if (~dp[i][parf][curf]) {
        return dp[i][parf][curf];
    }
    vector<int> newadj;
    for (auto nxt : adj[i]) {
        if (nxt == par) continue;
        newadj.push_back(nxt);
        dfs(nxt, i, curf, 0);
        dfs(nxt, i, curf, 1);
    }
    adj[i] = newadj;
    int m = adj[i].size();
    int dp2[m + 1][2]{};
    memset(dp2, '?', sizeof dp2);
    dp2[m][a[i] ^ parf ^ curf] = curf;
    for (int j = m - 1; j >= 0; j--) {
        for (int k = 0; k < 2; k++) {
            dp2[j][k] = min(dp2[j + 1][k ^ 1] + dp[adj[i][j]][curf][1], dp2[j + 1][k] + dp[adj[i][j]][curf][0]);
        }
    }
    return dp[i][parf][curf] = dp2[0][0];
}
int main() {
    ios_base::sync_with_stdio(0), cin.tie(0);
    cin >> n;
    for (int i = 1, u, v; i < n; i++) {
        cin >> u >> v;
        adj[u].push_back(v);
        adj[v].push_back(u);
    }
    for (int i = 1; i <= n; i++) {
        cin >> a[i];
    }
    memset(dp, -1, sizeof dp);
    int ans1 = dfs(1, 0, 0, 0);
    int ans2 = dfs(1, 0, 0, 1);
    if (min(ans1, ans2) >= int(1e9)) {
        cout << "impossible\n";
    } else {
        cout << min(ans1, ans2) << '\n';
    }
    return 0;
}

Subtask #1
0 / 5.0

# Verdict Execution time Memory Grader output
1 Incorrect 1 ms 4696 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #2
0 / 15.0

# Verdict Execution time Memory Grader output
1 Incorrect 1 ms 4696 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #3
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 77 ms 28012 KB Output is correct
2 Correct 64 ms 27696 KB Output is correct
3 Correct 61 ms 27988 KB Output is correct

Subtask #4
0 / 40.0

# Verdict Execution time Memory Grader output
1 Correct 61 ms 27992 KB Output is correct
2 Correct 62 ms 27816 KB Output is correct
3 Correct 56 ms 27996 KB Output is correct
4 Incorrect 49 ms 7500 KB Output isn't correct
5 Halted 0 ms 0 KB -

Subtask #5
0 / 30.0

# Verdict Execution time Memory Grader output
1 Incorrect 1 ms 4696 KB Output isn't correct
2 Halted 0 ms 0 KB -