Submission #1119072

#	Time	Username	Problem	Language	Result	Execution time	Memory
1119072		Requiem	Cats or Dogs (JOI18_catdog)	C++17	38 / 100	3044 ms	12864 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 subtask2{
    bool check(){
        return n <= 1000;
    }

    int dp[MAXN][2];
    //cat: 0
    //dog: 1.
    void dfs(int u, int p){
        dp[u][0] = dp[u][1] = 0;
        for(auto id: g[u]){
            int v = edge[id].se + edge[id].fi - u;
            if (v == p) continue;

            dfs(v, u);

            dp[u][0] += min(dp[v][0], dp[v][1] + 1);

            dp[u][1] += min(dp[v][1], dp[v][0] + 1);
        }

        if (catNode[u]) dp[u][1] = inf;
        else if (dogNode[u]) dp[u][0] = inf;

//        cout << u << ' ' << dp[u][0] << ' ' << dp[u][1] << endl;
    }

    int solve(){
        dfs(1, -1);
        return min(dp[1][0], dp[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);
    }
}

int cat(int u){
    catNode[u] = true;
    return subtask2::solve();

}

int dog(int u){
    dogNode[u] = true;
    return subtask2::solve();
}

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

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

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