제출 #603452

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
603452		Tigryonochekk	Connecting Supertrees (IOI20_supertrees)	C++17	21 / 100	353 ms	69040 KiB

supertrees

#include <iostream>
#include "supertrees.h"
#include <vector>
#include <set>
using namespace std;
const int N = 1002;

vector<vector<int>> answer;
set<int> dsu[N];
int com[N];

void dfs(int v) {
	for (auto to : dsu[v]) {
		if (com[to]) continue;
		com[to] = com[v];
		dfs(to);
	}
}

int construct(vector<vector<int>> p) {
	int n = p.size();
	answer.resize(n);
	for (int i = 0; i < n; i++) {
		answer[i].resize(n);
	}
	for (int i = 0; i < n; i++) {
		for (int j = 0; j < n; j++) {
			if (j == i) continue;
			if (p[i][j] == 1) {
				dsu[i].insert(j);
			}
		}
	}
	int k = 1;
	for (int i = 0; i < n; i++) {
		if (!com[i]) {
			com[i] = k++;
			dfs(i);
		}
	}
	for (int i = 0; i < n; i++) {
		for (int j = 0; j < n; j++) {
			if (j == i) continue;
			if (com[i] == com[j] && dsu[i].find(j) == dsu[i].end()) return 0;
			if (com[i] != com[j] && dsu[i].find(j) != dsu[i].end()) return 0;
		}
	}
	for (int i = 0; i < n; i++) {
		if (com[i]) {
			com[i] = 0;
			for (int j : dsu[i]) {
				answer[i][j] = 1;
				answer[j][i] = 1;
				com[j] = 0;
			}
		}
	}
	build(answer);
	return 1;
}

/*
3
1 0 0
0 1 1
0 1 1
*/

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