Submission #72960

#	Time	Username	Problem	Language	Result	Execution time	Memory
72960		ekrem	Cats or Dogs (JOI18_catdog)	C++98	38 / 100	3071 ms	52188 KiB

catdog

#include <bits/stdc++.h>
#include "catdog.h"
#define st first
#define nd second
#define mp make_pair
#define pb push_back
#define inf 1000000007
#define N 1000005
using namespace std;

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

int a[N], par[22][N];
ii dp[N];
vi gg[N], g[N];

void f(int node){
	int bir = 0, iki = 0;
	for(int i = 0; i < g[node].size(); i++){
		int coc = g[node][i];
		f(coc);
		bir += dp[coc].st;
		iki += dp[coc].nd;
	}
	if(a[node] == 1)
		iki = inf;
	if(a[node] == 2)
		bir = inf;
	// cout << node << " -> " << bir << " " << iki << endl;
	//bir = f(node, 0)
	//iki = f(node, 1)
	dp[node].st = min(bir, iki + 1);
	dp[node].nd = min(bir + 1, iki);
}

void hazirla(int node, int pr){
	par[0][node] = pr;
	for(int i = 0; i < gg[node].size(); i++)
		if(gg[node][i] != pr){
			g[node].pb(gg[node][i]);
			hazirla(gg[node][i], node);
		}
}

void initialize(int n, vi a, vi b){
	for(int i = 0; i < n - 1; i++){
		gg[a[i]].pb(b[i]);
		gg[b[i]].pb(a[i]);
	}
	hazirla(1, 0);
}

int cat(int v){//cout << "AMK:\n";
	a[v] = 1;
	f(1);
	return min(dp[1].st, dp[1].nd);
}

int dog(int v){//cout << "AMK:\n";
	a[v] = 2;
	f(1);
	return min(dp[1].st, dp[1].nd);
}

int neighbor(int v){//cout << "AMK:\n";
	a[v] = 0;
	f(1);
	return min(dp[1].st, dp[1].nd);
}


// int main(){
// 	freopen("in.txt", "r", stdin);
// 	freopen("out.txt", "w", stdout);
// 	int n, q;
// 	scanf("%d",&n);
// 	vi a(n - 1), b(n - 1);
// 	for(int i = 0; i < n - 1; i++)
// 		scanf("%d %d",&a[i] ,&b[i]);
// 	scanf("%d",&q);
// 	vi t(q), v(q);
// 	for(int i = 0; i < q; i++)
// 		scanf("%d %d",&t[i] ,&v[i]);

// 	initialize(n, a, b);
// 	vi ans(q);
// 	for(int i = 0; i < q; i++){
// 		if(t[i] == 1)
// 			ans[i] = cat(v[i]);
// 		else if(t[i] == 2)
// 			ans[i] = dog(v[i]);
// 		else
// 			ans[i] = neighbor(v[i]);
// 	}
// 	for(int i = 0; i < q; i++)
// 		printf("%d\n",ans[i]);
// 	return 0;
// }

Compilation message (stderr)

catdog.cpp: In function 'void f(int)':
catdog.cpp:20:19: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for(int i = 0; i < g[node].size(); i++){
                 ~~^~~~~~~~~~~~~~~~
catdog.cpp: In function 'void hazirla(int, int)':
catdog.cpp:39:19: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for(int i = 0; i < gg[node].size(); i++)
                 ~~^~~~~~~~~~~~~~~~~

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