Submission #586421

#	Time	Username	Problem	Language	Result	Execution time	Memory
586421		LastRonin	Friend (IOI14_friend)	C++14	Compilation error	0 ms	0 KiB

friend

#include "friend.h"
#include <bits/stdc++.h>
#define ll long long
#define pb push_back
using namespace std;

const ll N = 2e3 + 10;

vector<ll> g[N];
bool was[N]; 
ll we[N];


ll col[N];
ll cnt[3] = {0};

void dfs2(ll v, ll a) {
	comp.pb(v);
	assert(a == 1 || a == 2);
	cnt[a]++;
	assert(col[v] == 0);
	assert(was[v] == 0);
	col[v] = a;
	was[v] = 1;		
	for(auto u : g[v]) {
		if(!was[u]) {
			dfs2(u, a^3);
		}
		assert(col[v] != 0);
		assert(col[u] != 0);
		assert(col[v] != col[u]);
	}
}


int findSample(int n,int confidence[],int host[],int protocol[]){
    bool eq = 1;
    for(int j = 0; j < n; j++)
		we[j] = confidence[j], assert(we[j] == 1);
	for(int j = 1; j < n; j++) {
		assert(host[j] != j);
		if(protocol[j] == 0) {
			g[j].pb(host[j]);
			g[host[j]].pb(j);
		} else if(protocol[j] == 1) {
			for(auto u : g[host[j]]) {
				g[j].pb(u);
				g[u].pb(j);
			}
		}    
	}
	ll answ = 0;
	for(int i = 0; i < n; i++)
		comp.clear(), was[i] = 0, assert(we[i] == 1);
	for(int i = 0; i < n; i++) {
		if(!was[i]) {
		    cnt[1] = cnt[2] = 0;
			dfs2(i, 1);
			answ += max(cnt[1], cnt[2]);
		}
	}
	return answ;
}
/*
6
13 3 6 20 10 15
0 0 0 1 1 2 2 1 0 0
6
13 3 6 20 10 15
0 0 0 1 1 2 2 1 0 0

*/
/*
#define __MAXSIZE__ 100002
#include "friend.h"
using namespace std;

// Confidence
int confidence[__MAXSIZE__];

// Host
int host[__MAXSIZE__];

// Protocol
int protocol[__MAXSIZE__];

// Main
int main(void)
{
	int n,i;

	// Number of people
	assert(scanf("%d",&n)==1);
	
	// Confidence
	for(i=0;i<n;i++)
		assert(scanf("%d",&confidence[i])==1);
	
	// Host and Protocol
	for(i=1;i<n;i++)
		assert(scanf("%d %d",&host[i],&protocol[i])==2);
	
	// Answer
	printf("%d\n",findSample(n,confidence,host,protocol));
	return 0;
}
/**/

Compilation message (stderr)

friend.cpp:107:1: warning: "/*" within comment [-Wcomment]
  107 | /**/
      |  
friend.cpp: In function 'void dfs2(long long int, long long int)':
friend.cpp:18:2: error: 'comp' was not declared in this scope
   18 |  comp.pb(v);
      |  ^~~~
friend.cpp: In function 'int findSample(int, int*, int*, int*)':
friend.cpp:54:3: error: 'comp' was not declared in this scope
   54 |   comp.clear(), was[i] = 0, assert(we[i] == 1);
      |   ^~~~
friend.cpp:37:10: warning: unused variable 'eq' [-Wunused-variable]
   37 |     bool eq = 1;
      |          ^~