# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
318977 | neki | 슈퍼트리 잇기 (IOI20_supertrees) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
#include "supertreees.h"
#define loop(i, a, b) for(long long i=a;i<b;i++)
#define pool(i, a, b) for(long long i=a-1;i>=b;i--)
#define fore(i, a) for(auto&& i:a)
#define fi first
#define se second
#define ps(a) push_back(a)
#define pb(a) pop_back(a)
#define sc scanf
#define vc vector
#define pa pair<ll, ll>
#define ll int
#define lb lower_bound
#define ub upper_bound
#define all(a) a.begin(), a.end()
#define llmax LLONG_MAX/2
#define llmin -LLONG_MAX/2
using namespace std;
#define mn 1010
#define pa pair<ll, ll>
#define ld long double
ll arr[mn], sti[mn], com[mn];
vc<ll> ss[mn], cc[mn];
int construct(vc<vc<int>> p){
ll n=p.size();
loop(i, 0, n) arr[i]=-1, sti[i]=-1, com[i]=-1;
ll cnt=0;
loop(i, 0, n){
loop(j, 0, i){
if(p[i][j]==3) return 0;
if(p[i][j]){
if(arr[i]==-1) arr[i]=arr[j];
else if(arr[i]!=arr[j]) return 0;
}
else{
if(arr[i]==arr[j])return 0;
}
}
if(arr[i]==-1) arr[i]=cnt++;
}
loop(i, 0, n){
loop(j, 0, i){
if(p[i][j]){if(arr[i]!=arr[j])return 0;}
else {if(arr[i]==arr[j])return 0;}
}
}
ll cur=0;
loop(i, 0, n){
loop(j, 0, i){
if(p[i][j]==1){
if(sti[i]==-1) sti[i]=sti[j];
else if(sti[i]!=sti[j]) return 0;
}
else{
if(sti[i]==sti[j])return 0;
}
}
if(sti[i]==-1) sti[i]=cur++;
}
loop(i, 0, n){
loop(j, 0, i){
if(p[i][j]==1){ if(sti[i]!=sti[j])return 0;}
else{ if(sti[i]==sti[j])return 0;}
}
}
loop(i, 0, n) ss[sti[i]].ps(i);
loop(i, 0, n) com[sti[i]]=arr[i];
loop(i, 0, cur) cc[com[i]].ps(i);
vc<int> samp(n, 0);
vc<vc<int>> ans(n, samp);
loop(i, 0, cnt){
if(cc[i].size()==0) return 1;
if(cc[i].size()==1){
vc<ll> u=ss[cc[i][0]];
if(u.size()>1) loop(v, 0, (ll)u.size()-1) ans[u[v]][u[v+1]]=1, ans[u[v+1]][u[v]]=1;
}
else if(cc[i].size()==2) return 0;
else{
fore(t, cc[i]){
vc<ll> u=ss[cc[i][t]];
if(u.size()>1) loop(v, 0, (ll)u.size()-1) ans[u[v]][u[v+1]]=1, ans[u[v+1]][u[v]]=1;
}
loop(t, 0, cc[i].size()){
ans[ss[cc[i][t]][0]][ss[cc[i][((ll)t+1)%((ll)cc[i].size())]][0]]=1;
ans[ss[cc[i][(t+1)%((ll)cc[i].size())]][0]][ss[cc[i][t]][0]]=1;
}
}
}
build(ans);
return 1;
}