Submission #1119007

#	Time	Username	Problem	Language	Result	Execution time	Memory
1119007		Requiem	Cats or Dogs (JOI18_catdog)	C++17	0 / 100	8 ms	8016 KiB

catdog

#include "catdog.h"
#include<bits/stdc++.h>
//#define int long long
#define pb push_back
#define fast ios_base::sync_with_stdio(false), cin.tie(nullptr), cout.tie(nullptr);
#define MOD 1000000007
#define inf 1e9
#define fi first
#define se second
#define FOR(i,a,b) for(int i=a;i<=b;i++)
#define FORD(i,a,b) for(int i=a;i>=b;i--)
#define sz(a) ((int)(a).size())
#define endl '\n'
#define pi 3.14159265359
#define TASKNAME "catdogs"
using namespace std;
template<typename T> bool maximize(T &res, const T &val) { if (res < val){ res = val; return true; }; return false; }
template<typename T> bool minimize(T &res, const T &val) { if (res > val){ res = val; return true; }; return false; }
typedef pair<int,int> ii;
typedef pair<int,ii> iii;
typedef vector<int> vi;
/**
Cat / Dogs:

Cho 1 cây n đỉnh với n - 1 cạnh.

Một số đỉnh có màu đỏ / xanh.

Tìm tập cạnh tối thiểu sao cho:
==> trên đường đi giữa 1 cặp đỉnh xanh / đỏ bất kì đều tồn tại 1 cạnh trong tập đã chọn

subtask 2: N <= 1000, Q <= 1000.
O(N * Q).
Ta dựng cây lên.
Với 1 subtree của thằng u, ta nhận coi là nó là thằng blue / red.

Nếu số thằng blue > red thì mình cho blue lên, và chặn hết đống red.
**/
const int MAXN = 3e5 + 9;
int n;
ii edge[MAXN];
vector<int> g[MAXN];
bool catNode[MAXN], dogNode[MAXN];

namespace subtask3{
    bool check(){
        return true;
    }

    int par[MAXN], dp[MAXN], val[MAXN];

    void dfs(int u, int p){
        par[u] = p;

        for(auto id: g[u]){
            int v = edge[id].fi + edge[id].se - u;
            if (v == p) continue;
            dfs(v, u);
        }
    }

    void recalc(int u){
        int numDog = 0;
        int numCat = 0;

        for(auto id: g[u]){
            int v = edge[id].fi + edge[id].se - u;
            if (v == par[u]) continue;

            val[u] += val[v];

            if (dp[v] > 0) numDog++;
            else if (dp[v] < 0) numCat++;
        }

        if (dogNode[u]) {
            val[u] += numCat;
            dp[u] = 1;
        }
        else if (catNode[u]){
            val[u] += numDog;
            dp[u] = -1;
        }
        else {
            val[u] += min(numCat, numDog);
            if (numCat < numDog) dp[u] = 1;
            else if (numCat > numDog) dp[u] = -1;
        }
    }

    int update(int u){
        while(true){
            recalc(u);
            if (u == 1) break;
            u = par[u];
        }
        return val[1];
    }

    void init(){
        dfs(1, -1);
    }
}
void initialize(int N, vector<int> A, vector<int> B){
    n = N;
    memset(catNode, false, sizeof(catNode));
    memset(dogNode, false, sizeof(dogNode));
    FOR(i, 0, n - 2){
        edge[i] = {A[i], B[i]};
        g[A[i]].pb(i);
        g[B[i]].pb(i);
    }
    subtask3::dfs(1, -1);
}

int cat(int u){
    catNode[u] = true;
    return subtask3::update(u);

}

int dog(int u){
    dogNode[u] = true;
    return subtask3::update(u);
}

int neighbor(int u){
    catNode[u] = dogNode[u] = false;
    return subtask3::update(u);
}

/**
Warning:
Đọc sai đề???
Cận lmao
Code imple thiếu case nào không.
Limit.
**/

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