| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 308994 | tjdgus4384 | 슈퍼트리 잇기 (IOI20_supertrees) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include<bits/stdc++.h>
using namespace std;
vector<int> graph[1001], trees[1001];
vector<vector<int> > b, P;
vector<int> v[1001];
bool visited[1001];
void dfs(int x, int cnt){
graph[cnt].push_back(x);
for(int i = 0;i < v[x].size();i++){
if(visited[v[x][i]]) continue;
visited[v[x][i]] = true;
dfs(v[x][i], cnt);
}
}
void make_tree(int x){
for(int i = 0;i < v[x].size();i++){
if(visited[v[x][i]]) continue;
if(P[x][v[x][i]] != 1) continue;
visited[v[x][i]] = true;
make_tree(v[x][i]);
b[x][v[x][i]] = 1;
b[v[x][i]][x] = 1;
}
}
void build(vector<vector<int> > b){
for(int i = 0;i < b.size();i++){
for(int j = 0;j < b.size();j++){
printf("%d ", b[i][j]);
}
printf("\n");
}
}
int construct(vector<vector<int> > p){
P = p;
b.resize(p.size());
for(int i = 0;i < p.size();i++){
b[i].resize(p.size());
for(int j = 0;j < p.size();j++){
if(i == j) continue;
v[i].push_back(j);
}
}
int cnt = 0;
for(int i = 0;i < p.size();i++){
if(visited[i]) continue;
visited[i] = true;
dfs(i, cnt); cnt++;
}
for(int i = 0;i < p.size();i++) visited[i] = false;
for(int i = 0;i < cnt;i++){
vector<int> nv;
for(int j = 0;j < graph[i].size();j++){
if(visited[graph[i][j]]) continue;
visited[graph[i][j]] = true;
make_tree(graph[i][j]);
nv.push_back(graph[i][j]);
}
if(nv.size() <= 1) continue;
for(int j = 0;j < nv.size() - 1;j++){
b[nv[j]][nv[j+1]] = 1;
b[nv[j+1]][nv[j]] =1;
}
b[nv[0]][nv.back()] = 1;
b[nv.back()][nv[0]] = 1;
if(p[nv[0]][nv[1]] == 3 && nv.size() >= 4){
b[nv[0]][nv[2]] = b[nv[2]][nv[0]] = 1;
}
}
build(b);
return 1;
}
int main(){
int n;
scanf("%d", &n);
vector<vector<int> > p(n);
for(int i = 0;i < n;i++){
p[i].resize(n);
for(int j = 0;j < n;j++){
scanf("%d", &p[i][j]);
}
}
printf("%d", construct(p));
return 0;
}