Submission #1260862

#TimeUsernameProblemLanguageResultExecution timeMemory
1260862Bui_Quoc_CuongCats or Dogs (JOI18_catdog)C++20
Compilation error
0 ms0 KiB
#include <bits/stdc++.h>
using namespace std;
template <class A, class B>
    bool maximize(A &a, const B b) {
        if (a < b) {
            a = b;
            return true;
        } return false;
    }
template <class A, class B>
    bool minimize(A &a, const B b) {
        if (a > b) {
            a = b;
            return true;
        } return false;
    }
#define pb push_back
#define FOR(i, a, b) for(int i = a; i <= (int)b; i++)
#define FORD(i, a, b) for(int i = a; i >= (int)b; i--)
#define fi first
#define se second
#define ALL(A) A.begin(), A.end()

typedef vector<int> vi;
typedef pair<int, int> ii;
typedef long long ll;

const int N = 3e5 + 5;
const int oo = 2e9 + 29032008;
const ll INF = 1e18 + 29032008;

int n, q;
vi g[N];
int a[N];

namespace sub1 {
	ll dp[N][3];

	void dfs(int u, int p) {
		dp[u][1] = dp[u][2] = 2e9;
		if (a[u] == 0 || a[u] == 1) dp[u][1] = 0;
		if (a[u] == 0 || a[u] == 2) dp[u][2] = 0;
		for (int &v : g[u]) if (v != p) {
			dfs(v, u);
			dp[u][1]+= min(dp[v][1], dp[v][2] + 1);
			dp[u][2]+= min(dp[v][2], dp[v][1] + 1);
		}
	}

	int cat(int u) {
		a[u] = 1;
		dfs(1, - 1);
		return min(dp[1][1], dp[1][2]);
	}

	int dog(int u) {
		a[u] = 2;
		dfs(1, - 1);
		return min(dp[1][1], dp[1][2]);
	}

	int neighbor(int u) {
		a[u] = 0;
		dfs(1, - 1);
		return min(dp[1][1], dp[1][2]);
	}

	void solve() {	
		while (q--) {
			int type, u; cin >> type >> u;
			if (type == 1) cout << cat(u);
			if (type == 2) cout << dog(u);
			if (type == 3) cout << neighbor(u);
			cout << "\n";			
		}
	}
}

void initialize(int _N, vector<int> A, vector<int> B){
	n = _N;
    for(int i = 0; i < n - 1; ++i){
        int u = A[i], v = B[i];
        g[u].emplace_back(v);
        g[v].emplace_back(u);
    }
}

void solve() {
	cin >> n;
	vector <int> A(n - 1), B(n - 1);
	for (int i = 0; i < n - 1; i++) {
		cin >> A[i] >> B[i];
	}	
	initialize(n, A, B);

	cin >> q;
	return sub1::solve();
}	

int main() {
    ios_base::sync_with_stdio(false); cin.tie(nullptr);

    int NTEST = 1; // cin >> NTEST;
    while (NTEST--) solve();
    
    cerr << "\n[Time Elapsed] " << 0.001 * clock() << "s\n";
    return 0;
}

Compilation message (stderr)

/usr/bin/ld: /tmp/ccS5fpPH.o: in function `main':
grader.cpp:(.text.startup+0x0): multiple definition of `main'; /tmp/cc9mwxuV.o:catdog.cpp:(.text.startup+0x0): first defined here
/usr/bin/ld: /tmp/ccS5fpPH.o: in function `main':
grader.cpp:(.text.startup+0x219): undefined reference to `neighbor(int)'
/usr/bin/ld: grader.cpp:(.text.startup+0x258): undefined reference to `dog(int)'
/usr/bin/ld: grader.cpp:(.text.startup+0x261): undefined reference to `cat(int)'
collect2: error: ld returned 1 exit status