Submission #95893

#	Time	Username	Problem	Language	Result	Execution time	Memory
95893		onjo0127	Cats or Dogs (JOI18_catdog)	C++11	38 / 100	3026 ms	9448 KiB

catdog

#include "catdog.h"
#include <bits/stdc++.h>
using namespace std;

int A[100009], D[2][100009];
vector<int> adj[100009], tree[100009];

void dfs(int x, int p) {
    for(auto& it: adj[x]) {
        if(it != p) {
            tree[x].push_back(it);
            dfs(it, x);
        }
    }
}

void initialize(int N, vector<int> A, vector<int> B) {
    for(int i=0; i<N-1; i++) {
        int u = A[i], v = B[i];
        adj[u].push_back(v);
        adj[v].push_back(u);
    }
    dfs(1, 1);
}

void go(int x) {
    for(auto& it: tree[x]) go(it);
    int s1 = 0, s2 = 0;
    for(auto& it: tree[x]) {
        s1 += min(D[0][it], D[1][it] + 1);
        s2 += min(D[1][it], D[0][it] + 1);
    }
    D[0][x] = (A[x] == 2 ? 1e9 : s1);
    D[1][x] = (A[x] == 1 ? 1e9 : s2);
}

int DP() {
    go(1);
    return min(D[0][1], D[1][1]);
}

int cat(int v) {
    A[v] = 1;
    return DP();
}

int dog(int v) {
    A[v] = 2;
    return DP();
}

int neighbor(int v) {
    A[v] = 0;
    return DP();
}

• Subtask 1: 8 / 8
• Subtask 2: 30 / 30
• Subtask 3: 0 / 62