Submission #301847

#	Time	Username	Problem	Language	Result	Execution time	Memory
301847		Pro100Yan	Connecting Supertrees (IOI20_supertrees)	C++17	21 / 100	290 ms	22264 KiB

supertrees

#include "supertrees.h"
#include <vector>
#include <iostream>
#define ll long long
#define pb push_back

using namespace std;
const int N = 20005;
vector<vector<int>> ans;
int d[N], used[N];

int get_parent(int v){
    if(v == d[v])
      	return v;
    return d[v] = get_parent(d[v]);
}

void add(int u, int v){
    int f = get_parent(u), s = get_parent(v);
    if(f != s)
        d[f] = d[s];
}



int construct(std::vector<std::vector<int>> p) {
	int n = p.size();
	int fault = 0;
	for(int i = 0; i < n; i++)
        d[i] = i;
	for (int i = 0; i < n; i++) {
		std::vector<int> row;
		row.resize(n);
		ans.push_back(row);
	}

    for(int i = 0; i < n; i++){
        for(int j = 0; j < n; j++){
            if(p[i][j]){
                add(i, j);
            }
        }
    }
    int pi, pj;
    for(int i = 0; i < n; i++){
        for(int j = 0; j < n; j++){
            pi = get_parent(i);
            pj = get_parent(j);
            if(p[i][j]){
                if(pi != pj)
                    return 0;
                if(pi != i)
                    ans[i][pi] = ans[pi][i] = 1;
                if(pi != j)
                ans[j][pi] = ans[pi][j] = 1;
            }
            else{
                if(pi == pj)
                    return 0;
            }
        }

    }
    for(int i = 0; i < n; i++){
        for(int j = 0; j < n; j++){
            if(p[i][j] == 2){
                pi = get_parent(i);
                if(used[pi] || pi == i || pi == j)
                    continue;
                ans[i][j] = ans[j][i] = 1;
                used[pi] = 1;
            }


        }

    }
    for(int i = 0; i < n; i++){
        for(int j = 0; j < n; j++){
            if(p[i][j] == 2){
                pi = get_parent(i);
                if(!used[pi])
                    return 0;
            }


        }

    }
	build(ans);
	return 1;
}
/*
5
1 1 0 0 0
1 1 0 0 0
0 0 1 2 2
0 0 2 1 2
0 0 2 2 1
*/

Compilation message (stderr)

supertrees.cpp: In function 'int construct(std::vector<std::vector<int> >)':
supertrees.cpp:28:6: warning: unused variable 'fault' [-Wunused-variable]
   28 |  int fault = 0;
      |      ^~~~~

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