Submission #393103

#TimeUsernameProblemLanguageResultExecution timeMemory
393103abdzagConnecting Supertrees (IOI20_supertrees)C++17
11 / 100
805 ms26020 KiB
#include<bits/stdc++.h>
#include"supertrees.h"
#define rep(i,a,b) for(int i=a;i<int(b);i++)
#define rrep(i,a,b)for(int i=a;i>int(b);i--)
#define all(v) v.begin(),v.end()
#define trav(a,v) for(auto& a: v)
typedef long long ll;
using namespace std;
const long long inf = 1e15;

vector<ll> p(2001);
vector<ll> Rank(2001);
ll Ufindset(ll i) {
	if (p[i] == i)return i;
	return p[i] = Ufindset(p[i]);
}
void UjoinSet(ll i, ll j) {
	ll si = Ufindset(i);
	ll sj = Ufindset(j);
	if (sj == si)return;
	if ( Rank[sj]> Rank[si])swap(sj, si);
	p[sj] = si;
	if (Rank[sj] == Rank[si]) Rank[si]++;
}

int construct(vector<vector<int>> v) {
	rep(i, 0, 2001)p[i] = i;
	rep(i, 0, v.size()) {
		rep(j, 0, v.size()) {
			if (v[i][j]) UjoinSet(i, j);
		}
	}
	vector<vector<int>> b(v.size(), vector<int>(v.size()));
	rep(i, 0, v.size()) {
		if (p[i] != i) {
			b[i][p[i]]=1;
			b[p[i]][i] = 1;
		}
	}
	vector<vector<int>> check(v.size(), vector<int>(v.size()));
	rep(i, 0, v.size()) {
		if (p[i] == i) {
			vector<int> curvec(v.size());
			queue<ll> q;
			q.push(i);
			while (!q.empty()) {
				ll cur = q.front();
				q.pop();
				curvec[cur] = 1;
				rep(j,0,b[i].size())if(b[i][j])if(!curvec[j])q.push(j);
			}
			if (curvec != v[i]) {
				return 0;
			}
		}	
	}
	build(b);
	return 1;
}

#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...