Submission #1367630

#TimeUsernameProblemLanguageResultExecution timeMemory
1367630asli_bgConnecting Supertrees (IOI20_supertrees)C++20
11 / 100
65 ms22028 KiB
#include <bits/stdc++.h>
using namespace std;

//#define int long long

#define fi first
#define se second

#define FOR(i,a) for(int i=0;i<(a);i++)
#define FORE(i,a,b) for(int i=(a);i<(b);i++)

#define sp <<' '<<

#define mid (l+r)/2

#define all(x) x.begin(),x.end()
#define carp(a,b) ((a%MOD)*(b%MOD))%MOD
#define topla(a,b) ((a%MOD)+(b%MOD))%MOD

#define pb push_back

#define DEBUG(x) cout<<#x sp x<<endl;
#define cont(x) for(auto el:x) cout<<el<<' ';cout<<endl;
#define contp(x) for(auto el:x) cout<<el.fi<<'-'<<el.se<<' ';cout<<endl;

typedef vector<int> vi;
typedef pair<int,int> pii;
typedef vector<pii> vii;
typedef vector<bool> vb;
typedef long long ll;
typedef vector<ll> vll;
typedef pair<ll,ll> pll;


const int MAXN=2e5+6;
const int INF=1e9+7;

#include "supertrees.h"

int n;
int r[MAXN], p[MAXN];
vi adj[MAXN];

int find(int a){
    while(a!=p[a]) a=p[a];
    return a;
}

void uni(int a,int b){
    a=find(a);
    b=find(b);
    if(r[a]>r[b]) swap(a,b);
    p[a]=p[b];
    if(r[a]==r[b]) r[b]++;
    r[a]=r[b];
}

int construct(vector<vi> a){
    n=a.size();

    vector<vi> ans(n,vi(n));

    FORE(i,1,n+1){
        r[i]=0;
        p[i]=i;
    }

    FOR(i,n){
        FOR(j,n){
            if(a[i][j]==1){
                if(find(i+1)!=find(j+1)){
                    uni(i+1,j+1);
                    adj[i+1].pb(j+1);
                    adj[j+1].pb(i+1);
                }
            }
        }
    }

    set<int> s;
    FORE(i,1,n+1){
        s.insert(find(i));
    }

    int once=*s.begin();
    s.erase(once);

    int cur=once;
    for(auto el:s){
        adj[cur].pb(el);
        adj[el].pb(cur);
        cur=el;
    }

    adj[cur].pb(once);
    adj[once].pb(cur);

    //cont(s);

    FORE(i,1,n+1){
        for(auto kom:adj[i]){
            if(i!=kom) ans[i-1][kom-1]=1;
        }
    }

    build(ans);

    return 1;



	/*int n = p.size();
	vector<vi> answer;
	for (int i = 0; i < n; i++) {
		vi row;
		row.resize(n);
		answer.push_back(row);
	}
	build(answer);
	return 1;*/
}
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...