oj.uz

Submission #404641

# Submission time Handle Problem Language Result Execution time Memory
404641 2021-05-14T19:11:50 Z nicolaalexandra The Xana coup (BOI21_xanadu) C++14
10 / 100
 140 ms 15300 KB 
#include <bits/stdc++.h>
#define DIM 100010
#define INF 2000000000
using namespace std;

int dp[DIM][2][2],v[DIM];
vector <int> L[DIM];
int n,i,x,y;

void dfs (int nod, int tata){
    int ok = 0;
    for (auto vecin : L[nod])
        if (vecin != tata){
            ok = 1;
            dfs (vecin,nod);
        }
    if (!ok){
        dp[nod][v[nod]][0] = 0;
        dp[nod][1-v[nod]][1] = 1;
    } else {

        /// caz 1: nu dau switch in nod
        int sum = 0, cnt = 0;
        for (auto vecin : L[nod]){
            if (vecin == tata)
                continue;
            if (dp[vecin][0][0] < dp[vecin][0][1])
                sum += dp[vecin][0][0];
            else {
                sum += dp[vecin][0][1];
                cnt++;
            }
        }

        if (cnt % 2 == 0)
            dp[nod][v[nod]][0] = sum;
        else dp[nod][1-v[nod]][0] = sum;

        /// incerc sa schimb paritatea lui cnt
        int mini = INF;
        for (auto vecin : L[nod]){
            if (vecin == tata)
                continue;
            if (dp[vecin][0][0] < dp[vecin][0][1])
                mini = min (mini,sum - dp[vecin][0][0] + dp[vecin][0][1]);
            else mini = min (mini,sum - dp[vecin][0][1] + dp[vecin][0][0]);
        }

        if (cnt % 2 == 0)
            dp[nod][1-v[nod]][0] = mini;
        else dp[nod][v[nod]][0] = mini;


        /// caz 2: dau switch in nod
        cnt = 1, sum = 0;
        for (auto vecin : L[nod]){
            if (vecin == tata)
                continue;
            if (dp[vecin][1][0] < dp[vecin][1][1])
                sum += dp[vecin][1][0];
            else {
                sum += dp[vecin][1][1];
                cnt++;
            }
        }

        if (cnt % 2 == 0)
            dp[nod][v[nod]][1] = 1 + sum;
        else dp[nod][1-v[nod]][1] = 1 + sum;

        /// schimb starea din nod
        mini = INF;
        for (auto vecin : L[nod]){
            if (vecin == tata)
                continue;
            if (dp[vecin][1][0] < dp[vecin][1][1])
                mini = min (mini,sum - dp[vecin][1][0] + dp[vecin][1][1]);
            else mini = min (mini, sum - dp[vecin][1][1] + dp[vecin][1][0]);
        }

        if (cnt % 2 == 0)
            dp[nod][1-v[nod]][1] = 1 + mini;
        else dp[nod][v[nod]][1] = 1 + mini;

    }
}

int main (){

    //ifstream cin ("date.in");
    //ofstream cout ("date.out");

    cin>>n;
    for (i=1;i<n;i++){
        cin>>x>>y;
        L[x].push_back(y);
        L[y].push_back(x);
    }

    for (i=1;i<=n;i++){
        cin>>v[i];
        dp[i][0][0] = dp[i][0][1] = dp[i][1][0] = dp[i][1][1] = INF;
    }

    dfs (1,0);

    int sol = min(dp[1][0][0],dp[1][0][1]);
    if (sol == INF)
        cout<<"impossible";
    else cout<<sol;


    return 0;
}

Subtask #1
0 / 5.0

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

Subtask #2
0 / 15.0

# Verdict Execution time Memory Grader output
1 Incorrect 2 ms 2636 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 128 ms 15236 KB Output is correct
2 Correct 124 ms 15116 KB Output is correct
3 Correct 124 ms 15300 KB Output is correct

Subtask #4
0 / 40.0

# Verdict Execution time Memory Grader output
1 Correct 120 ms 15184 KB Output is correct
2 Correct 119 ms 15008 KB Output is correct
3 Correct 129 ms 15300 KB Output is correct
4 Incorrect 140 ms 8348 KB Output isn't correct
5 Halted 0 ms 0 KB -

Subtask #5
0 / 30.0

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