Submission #1367658

#TimeUsernameProblemLanguageResultExecution timeMemory
1367658asli_bgConnecting Supertrees (IOI20_supertrees)C++20
21 / 100
79 ms26124 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];

vector<vi> a;

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];
}

bool vis[MAXN];
int son;

void dfs(int nd){
    vis[nd]=true;
    son=nd;
    FOR(i,n){
        if(!vis[i+1] and a[nd-1][i]==2){
            if(find(nd)!=find(i+1)){
                uni(i+1,nd);
                adj[i+1].pb(nd);
                adj[nd].pb(i+1);

                dfs(i+1);
            }
        }
    }
}

vi tut[MAXN];

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

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

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

    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);
                }
            }
        }
    }

    FORE(i,1,n+1){
        tut[find(i)].pb(i);
    }

    FORE(nd,1,n+1){
        FOR(i,tut[nd].size()){
            FORE(j,i+1,tut[nd].size()){
                int bir=tut[nd][i]-1;
                int iki=tut[nd][j]-1;

                if(a[bir][iki]!=1){
                    return 0;
                }

                a[bir][iki]=-1;
                a[iki][bir]=-1;
            }
        }
    }

    //cout<<"here"<<endl;

    FOR(i,n){
        FOR(j,n){
            if(i!=j and a[i][j]==1) return 0;
        }
    }


    FORE(i,1,n+1){
        if(!vis[i]){
            dfs(i);
            adj[son].pb(i);
            adj[i].pb(son);
        }
    }

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

    FOR(i,n){
        FOR(j,n){
            if(a[i][j]==0){
                if(find(i+1)==find(j+1)){
                    return 0;
                }
            }
            else{
                if(find(i+1)!=find(j+1)){
                    return 0;
                }
                else{
                    if(a[i][j]==2 and tut[find(i+1)].size()==2) return 0;
                }
            }
        }
    }

    
    build(ans);
    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...