제출 #1051366

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1051366		Gromp15	슈퍼트리 잇기 (IOI20_supertrees)	C++17	21 / 100	106 ms	24148 KiB

supertrees

#include <bits/stdc++.h>
#include "supertrees.h"
#define sz(x) (int)x.size()
#define all(x) x.begin(), x.end()
using namespace std;

struct dsu {
	vector<int> p, size;
	dsu(int n) : p(n), size(n, 1) {
		iota(all(p), 0);
	}
	int get(int v) {
		return v == p[v] ? v : p[v] = get(p[v]);
	}
	bool merge(int a, int b) {
		a = get(a), b = get(b);
		if (a == b) return 0;
		if (size[a] < size[b]) swap(a, b);
		size[a] += size[b], p[b] = a;
		return 1;
	}
};

int construct(std::vector<std::vector<int>> p) {
	int n = sz(p);
	dsu d(n);
	for (int i = 0; i < n; i++) for (int j = i+1; j < n; j++) if (p[i][j]) d.merge(i, j);
	for (int i = 0; i < n; i++) for (int j = i+1; j < n; j++) if (!p[i][j] && d.get(i) == d.get(j)) return 0;
	vector<vector<int>> ans(n, vector<int>(n));
	for (int i = 0; i < n; i++) if (i != d.get(i)) ans[i][d.get(i)] = ans[d.get(i)][i] = 1;
	build(ans);
	return 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