Submission #977668

#	Time	Username	Problem	Language	Result	Execution time	Memory
977668		SmuggingSpun	Connecting Supertrees (IOI20_supertrees)	C++14	11 / 100	156 ms	18068 KiB

supertrees

#include "supertrees.h"
#include<bits/stdc++.h>
using namespace std;
int construct(vector<vector<int>>p){
	int n = p.size();
	bool is_sub1 = true;
	for(int i = 0; i < n; i++){
		for(int j = 0; j < n; j++){
			if(p[i][j] != 1){
				is_sub1 = false;
				i = n;
				break;
			}
		}
	}
	if(is_sub1){
		p = vector<vector<int>>(n, vector<int>(n, 0));
		for(int i = 1; i < n; i++){
			p[i][i - 1] = p[i - 1][i] = 1;
		}
		build(p);
		return 1;
	}
	vector<int>par(n);
	iota(par.begin(), par.end(), 0);
	function<int(int)>find_set;
	find_set = [&] (int n){
		return n == par[n] ? n : par[n] = find_set(par[n]);	
	};
	function<void(int, int)>merge = [&] (int a, int b){
		par[find_set(a)] = find_set(b);
	};
	vector<bool>vis(n, false);
	vector<vector<int>>edge(n);
	for(int i = 0; i < n; i++){
		if(vis[i]){
			for(int j = 0; j < n; j++){
				if((find_set(i) == find_set(j)) != p[i][j]){
					return 0;
				}
			}
			continue;
		}
		for(int j = 0; j < n; j++){
			if(p[i][j] == 0 && find_set(i) == find_set(j)){
				return 0;
			}
			if(p[i][j] == 1){
				merge(i, j);
				vis[j] = true;
			}
		}
	}
	for(int i = 0; i < n; i++){
		edge[find_set(i)].emplace_back(i);
	}
	for(int i = 0; i < n; i++){
		for(int j = 1; j < edge.size(); j++){
			p[edge[i][j]][edge[i][j - 1]] = p[edge[i][j - 1]][edge[i][j]] = 1;
		}
	}
	build(p);
	return 1;
}

Compilation message (stderr)

supertrees.cpp: In function 'int construct(std::vector<std::vector<int> >)':
supertrees.cpp:58:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::vector<int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   58 |   for(int j = 1; j < edge.size(); j++){
      |                  ~~^~~~~~~~~~~~~

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