Submission #1348521

#TimeUsernameProblemLanguageResultExecution timeMemory
1348521Jawad_Akbar_JJConnecting Supertrees (IOI20_supertrees)C++17
65 / 100
1095 ms22172 KiB
#include <iostream>
#include <map>
#include <vector>
#include "supertrees.h"

using namespace std;
const int N = 1<<10;
vector<int> nei[N];
int rt[N], num[N], vis[N];

void Add(int i, int j, vector<vector<int>> &adj){
	if (!adj[i][j])
		nei[i].push_back(j);
	if (!adj[j][i])
		nei[j].push_back(i);
	adj[i][j] = adj[j][i] = 1;
}

void dfs(int u){
	num[u]++;
	vis[u] = 1;

	for (int i : nei[u])
		if (vis[i] != 1)
			dfs(i);
	vis[u] = 0;
}

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

	for (int i=0;i<n;i++)
 		rt[i] = -1;

	for (int i=0;i<n;i++){
		if (rt[i] == -1){
			rt[i] = i;
			for (int j=0;j<n;j++)
				if (i != j and p[i][j] == 1)
					rt[j] = i, Add(i, j, adj);
		}
	}

	for (int i=0;i<n;i++){
		int t = 1;
		for (int j=i+1;t and j<n;j++){
			if (i == rt[i] and j == rt[j] and p[i][j] == 2)
				Add(i, j, adj), t = 0;
		}
		for (int j=0;t and j < i;j++){
			if (i == rt[i] and j == rt[j] and p[i][j] == 2)
				Add(i, j, adj), t = 0;
		}
	}
	for (int i=0;i<n;i++){
		dfs(i);
		for (int j=0;j<n;j++){
			if (num[j] != p[i][j])
				return 0;
			num[j] = 0;
		}
	}
	build(adj);
	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...