Submission #305814

#TimeUsernameProblemLanguageResultExecution timeMemory
305814tincanConnecting Supertrees (IOI20_supertrees)C++17
21 / 100
305 ms22392 KiB
#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>
#include<vector>
#include<cmath>
#include<map>
#include<stack>
#include<set>
#include<deque>
#include<string>
#include<unordered_map>
#include<bitset>
#include<random>
#include<chrono>
//Eeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeee
using namespace std;
const int MAXN = 1005;
bool vis[MAXN];
int cnt[MAXN];
struct dsu {
	int arr[MAXN];
	dsu() { for (int i = 1; i < MAXN; i++) arr[i] = i; }
	int find(int s) { return arr[s] == s ? s : arr[s] = find(arr[s]); }
	void join(int a, int b) { if (find(a) != find(b)) arr[find(a)] = find(b); }
} st, st2;
vector<int> thng[MAXN];
void build(vector<vector<int>> b);
void make(int i, int i2, vector<vector<int>> &ans) {
	ans[i][i2] = ans[i2][i] = 1;
}
int construct(vector<vector<int>> p) {
	int N = p.size();
	for (int i = 0; i < N; i++) for (int i2 = 0; i2 < N; i2++) if (p[i][i2] == 3) return 0;
	vector<vector<int>> ans(N, vector<int>(N, 0));
	for (int i = 0; i < N; i++) {
		for (int i2 = i + 1; i2 < N; i2++) {
			if (p[i][i2]) {
				if(p[i][i2] == 1) st2.join(i, i2);
				st.join(i, i2);
			}
		}
	}
	for (int i = 0; i < N; i++)
		for (int i2 = i + 1; i2 < N; i2++)
			if (!p[i][i2] && (st.find(i) == st.find(i2))) return 0;
			else if (p[i][i2] == 2 && (st2.find(i) == st2.find(i2))) return 0;
	for (int i = 0; i < N; i++) if(!vis[st2.find(i)]) {
		vis[st2.find(i)] = true;
		++cnt[st.find(st2.find(i))];
	}
	for (int i = 0; i < N; i++) if (cnt[i] == 2) return 0;
	for (int i = 0; i < N; i++) {
		if (thng[st2.find(i)].size()) make(thng[st2.find(i)].back(), i, ans);
		thng[st2.find(i)].push_back(i);
	}
	for (int i = 0; i < N; i++) {
		for (int i2 = i + 1; i2 < N; i2++) {
			if (p[i][i2] == 2) {
				make(st2.find(i), st2.find(i2), ans);
			}
		}
	}
	build(ans); 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...