이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include<bits/stdc++.h>
#include "supertrees.h"
#define MAXN 300007
using namespace std;
int n;
int dsu[MAXN],dsu2[MAXN];
vector<int> comp[MAXN],comp2[MAXN];
vector< vector<int> > ans;
int root(int x){
if(dsu[x]==x)return x;
dsu[x]=root(dsu[x]);
return dsu[x];
}
void mergev(int x,int y){
dsu[root(x)]=root(y);
}
int root2(int x){
if(dsu2[x]==x)return x;
dsu2[x]=root2(dsu2[x]);
return dsu2[x];
}
void mergev2(int x,int y){
dsu2[root2(x)]=root2(y);
}
void add_edge(int x,int y){
ans[x][y]=ans[y][x]=1;
}
int construct(vector< vector<int> > p){
n=int(p.size());
ans.resize(n);
for(int i=0;i<n;i++){
ans[i].resize(n);
}
for(int i=0;i<n;i++){
dsu[i]=dsu2[i]=i;
}
for(int i=0;i<n;i++){
for(int f=i+1;f<n;f++){
if(p[i][f]==1)mergev(i,f);
}
}
for(int i=0;i<n;i++){
comp[root(i)].push_back(i);
}
for(int i=0;i<n;i++){
if(comp[i].empty())continue;
for(int f=0;f<comp[i].size();f++){
for(int k=f+1;k<comp[i].size();k++){
if(p[comp[i][f]][comp[i][k]]!=1)return 0;
}
}
for(int f=0;f<comp[i].size()-1;f++){
add_edge(comp[i][f],comp[i][f+1]);
}
}
for(int i=0;i<n;i++){
for(int f=i+1;f<n;f++){
if(p[i][f]==2)mergev2(root(i),root(f));
}
}
for(int i=0;i<n;i++){
if(comp[i].empty())continue;
comp2[root2(i)].push_back(i);
}
for(int i=0;i<n;i++){
if(comp2[i].empty() or comp2[i].size()==1)continue;
for(int f=0;f<comp2[i].size();f++){
for(int k=f+1;k<comp2[i].size();k++){
for(int t:comp[comp2[i][f]]){
for(int s:comp[comp2[i][k]]){
if(p[t][s]!=2)return 0;
}
}
}
}
for(int f=0;f<comp2[i].size();f++){
add_edge(comp2[i][f],comp2[i][(f+1)%int(comp2[i].size())]);
}
}
for(int i=0;i<n;i++){
for(int f=i+1;f<n;f++){
if(root2(root(i))!=root2(root(f)) and p[i][f]!=0)return 0;
if(root2(root(i))==root2(root(f)) and p[i][f]!=2)return 0;
}
}
build(ans);
/*
for(int i=0;i<ans.size();i++){
for(int f=0;f<ans.size();f++){
cout<<ans[i][f]<<" ";
}
cout<<"\n";
}
*/
return 1;
}
/*
int main(){
construct({{1, 1, 1,1}, {1, 1, 1,1}, {1,1,1,1}, {1,1,1,1}});
}
*/
컴파일 시 표준 에러 (stderr) 메시지
supertrees.cpp: In function 'int construct(std::vector<std::vector<int> >)':
supertrees.cpp:60:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
60 | for(int f=0;f<comp[i].size();f++){
| ~^~~~~~~~~~~~~~~
supertrees.cpp:61:28: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
61 | for(int k=f+1;k<comp[i].size();k++){
| ~^~~~~~~~~~~~~~~
supertrees.cpp:65:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
65 | for(int f=0;f<comp[i].size()-1;f++){
| ~^~~~~~~~~~~~~~~~~
supertrees.cpp:84:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
84 | for(int f=0;f<comp2[i].size();f++){
| ~^~~~~~~~~~~~~~~~
supertrees.cpp:85:28: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
85 | for(int k=f+1;k<comp2[i].size();k++){
| ~^~~~~~~~~~~~~~~~
supertrees.cpp:93:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
93 | for(int f=0;f<comp2[i].size();f++){
| ~^~~~~~~~~~~~~~~~
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |