Submission #487941

#	Time	Username	Problem	Language	Result	Execution time	Memory
487941		SlavicG	Connecting Supertrees (IOI20_supertrees)	C++17	11 / 100	1196 ms	1974100 KiB

supertrees

#include "supertrees.h"
#include "bits/stdc++.h"
using namespace std;
 
#define ll long long
 
#define       forn(i,n)              for(int i=0;i<n;i++)
#define          all(v)              v.begin(), v.end()
#define         rall(v)              v.rbegin(),v.rend()
 
#define            pb                push_back
#define          sz(a)               (int)a.size()

struct DSU{
    int n;
    vector<int> par, s;

    DSU(int N){
        n = N;
        s.assign(n, 1), par.resize(n);
        iota(all(par), 0);
    }

    int get(int a){
        return par[a] = (a == par[a] ? a : get(par[a]));
    }

    void uni(int a, int b){
        a = get(a), b = get(b);
        if(s[a] > s[b])swap(a, b);
        par[a] = b, s[b] += s[a];
    }

    int get_size(int a){
        return s[get(a)];
    }
    bool same_set(int a, int b){
        return get(a) == get(b);
    }
};

const int N = 1005;
vector<int> adj[N], adj2[N];
vector<int> component;
bool vis[N];
int r[N];

void dfs(int u){
    component.pb(u);
    vis[u] = true;
    for(int v: adj[u])dfs(v);
}

int construct(vector<vector<int>> p){
    int n = sz(p);
    vector<vector<int>> ans(n, vector<int>(n, 0));

    DSU d(n);

    for(int i = 0;i < n; ++i){
        for(int j = 0;j < n; ++j){
            if(p[i][j] == 2 && !d.same_set(i, j)){
                adj[i].pb(j);
                adj[j].pb(i);
                d.uni(i, j);
            }
            if(p[i][j] == 1 && i != j)adj2[i].pb(j), adj2[j].pb(i);
        }
    }


    for(int u = 0;u < n; ++u){
        for(int v: adj2[u]){
            if(!d.same_set(u, v)){
                ans[u][v] = ans[v][u] = 1;
                d.uni(u, v);
            }
        }
    }
    for(int i = 0;i < n; ++i){
        if(!vis[i]){
            component.clear();
            dfs(i);

            for(int i = 0;i < sz(component); ++i){
                if(sz(component) > 1)ans[component[i]][component[(i + 1) % sz(component)]] = ans[component[(i + 1) % sz(component)]][component[i]] = 1;
            }
        }
    }
    build(ans);
    return 1;
}   

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