Submission #568430

#	Time	Username	Problem	Language	Result	Execution time	Memory
568430		Jomnoi	Connecting Supertrees (IOI20_supertrees)	C++17	19 / 100	235 ms	24012 KiB

supertrees

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

const int MAX_N = 1e3 + 5;

int n;
int parent[MAX_N];
vector <vector <int>> answer;
vector <int> nodes[MAX_N];

int root(int u) {
	if(u == parent[u]) {
		return u;
	}
	return parent[u] = root(parent[u]);
}

int construct(vector <vector <int>> p) {
	n = p.size();
	answer.resize(n, vector <int> (n, 0));

	iota(parent, parent + n, 0);
	for(int i = 0; i < n; i++) {
		for(int j = i + 1; j < n; j++) {
			if(p[i][j] == 2) {
				parent[root(j)] = root(i);
			}
		}
	}

	for(int i = 0; i < n; i++) {
		for(int j = i + 1; j < n; j++) {
			if(p[i][j] == 0 and root(i) == root(j)) {
				return 0;
			}
		}
	}

	for(int i = 0; i < n; i++) {
		nodes[root(i)].push_back(i);
	}
	for(int i = 0; i < n; i++) {
		if(i == root(i) and nodes[i].size() > 1) {
			if(nodes[i].size() == 2) {
				return 0;
			}

			int p = nodes[i].back();
			for(auto v : nodes[i]) {
				answer[p][v] = answer[v][p] = 1;
				p = v;
			}
		}
	}

	build(answer);
	return 1;
}

• Subtask 1: 0 / 11
• Subtask 2: 0 / 10
• Subtask 3: 19 / 19
• Subtask 4: 0 / 35
• Subtask 5: 0 / 21
• Subtask 6: 0 / 4