Submission #363562

#	Time	Username	Problem	Language	Result	Execution time	Memory
363562		silverfish	Friend (IOI14_friend)	C++14	35 / 100	4 ms	512 KiB

friend

#include "friend.h"
#include <bits/stdc++.h>
using namespace std;

const int N = 1e3;

vector<int> adj[N];
long long dp[N][2];
int vis[N];

void dfs(int u, int p, int *conf){

	for(int v : adj[u]){
		if(v != p) dfs(v, u, conf);
	}

	for(int v : adj[u]){
		if(v != p){
			dp[u][0] += max(dp[v][1], dp[v][0]);
			dp[u][1] += dp[v][0];
		}
	}
	dp[u][1] += 1LL*conf[u];
}

int findSample(int n, int conf[],int host[],int protocol[]){
	bool subtask2 = 1, subtask3 = 1, subtask4 = 1, subtask5 = 1;
	for(int i = 1; i < n; ++i){
		if(protocol[i] != 1) subtask2 = 0;
		if(protocol[i] != 2) subtask3 = 0;
		if(protocol[i] != 0) subtask4 = 0;
		if(conf[i] != 1) subtask5 = 0;
	}
	if(subtask2){
		long long ans = 0;
		for(int i = 0; i < n; ++i) ans += 1LL*conf[i];
		return ans;
	}
	if(subtask3){
		return *max_element(conf, conf+n);
	}

	if(subtask4){
		//build tree
		for(int i = 1; i < n; ++i){
			adj[host[i]].push_back(i);
			adj[i].push_back(host[i]);
		}

		dfs(0, 0, conf);
		return max(dp[0][1], dp[0][0]);
	}

	if(subtask5){
		for(int i = 1; i < n; ++i){
			if(protocol[i] == 0){
				adj[host[i]].push_back(i);
				adj[i].push_back(host[i]);
			}else{
				for(int j : adj[host[i]]){
					adj[i].push_back(j);
					adj[j].push_back(i);
				}
			}
		}
		array<long long,2> ans = {0, 0};
		queue<array<int,2>> q;
		q.push({0, 0});
		// node, depth
		vis[0] = 1;
		++ans[0%2];
		while(q.size()){
			auto x = q.front();
			int d = x[1], u = x[0];
			q.pop();
			++ans[d%2];
			for(int v : adj[u]){
				if(!vis[v]){
					vis[v] = 1;
					q.push({v, d+1});
				}
			}
		}
		return max(ans[0], ans[1]);
	}

	return -1;
}

• Subtask 1: 0 / 11
• Subtask 2: 8 / 8
• Subtask 3: 8 / 8
• Subtask 4: 19 / 19
• Subtask 5: 0 / 23
• Subtask 6: 0 / 31