| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 366774 | SlavicG | 슈퍼트리 잇기 (IOI20_supertrees) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#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()
#define fastio ios_base::sync_with_stdio(false);cin.tie(0);cout.tie(0);
const int N = 1001;
vector<int> adj[N];
bool vis[N];
vector<int> component;
int par[N];
int get(int a){
while(par[a]!=a)a=par[a];
return a;
}
void uni(int a,int b){
a = get(a);
b = get(b);
if(a==b)return;
par[b] = a;
}
void dfs(int u)
{
vis[u] = true;
component.pb(u);
for(auto x : adj[u]){
if(!vis[x])dfs(x);
}
}
int construct(vector<vector<int>> p)
{
int n = sz(p);
for(int i = 0;i < n;i++)adj[i].clear();
for(int i =0;i < n;i++)vis[i] = false;
vector<vector<int>> ans;
ans.resize(n);
for(int i=0;i < n;i++)
par[i]=i;
for(int i =0;i < n;i++)ans[i].resize(n);
forn(i,n)forn(j,n)ans[i][j]=0;
bool two = false;
bool one = false;
for(int i = 0;i < n;i++)
{
for(int j = 0;j < n;j++)
{
if(p[i][j] == 1)one = true;
if(p[i][j]){
uni(i,j);
adj[i].pb(j);
}
}
}
for(int i = 0;i < n;i++)
{
for(int j =0;j < n;j++)
{
if(get(i)==get(j) && !p[i][j])return 0;
}
}
for(int i = 0;i < n;i++)
{
if(!vis[i]){
component.clear();
two_cur = false;
dfs(i);
for(int j = 1;j < sz(component);j++)
{
ans[component[j]][component[j-1]] = ans[component[j-1]][component[j]] = 1;
if(p[component[j]][component[j-1]] == 2)two_cur = true;
}
if(sz(component) == 2 && two_cur)return 0;
if(two_cur && sz(component) > 1)ans[component.back()][component[0]] = ans[component[0]][component.back()] = 1;
}
}
build(ans);
return 1;
}