# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
1106609 | Ahmed57 | 스핑크스 (IOI24_sphinx) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
#include "sphinx.h"
#include "grader.cpp"
using namespace std;
int pr[251];
vector<int> adjj[251];
int find(int x){
if(x==pr[x])return x;
return pr[x] = find(pr[x]);
}
bool mergegroup(int a,int b){
a = find(a);
b = find(b);
if(a==b)return 0;
pr[a] = b;
return 1;
}
vector<int> find_colours(int n,vector<int> X,vector<int> Y){
for(int i = 0;i<X.size();i++){
adjj[X[i]].push_back(Y[i]);
adjj[Y[i]].push_back(X[i]);
}
stack<int> q;
set<int> s;
for(int i = 0;i<n;i++){
s.insert(i);
}
vector<int> col(n,0);
int d = 0;
while(!s.empty()){
q.push((*s.begin()));
s.erase(s.begin());
while(!q.empty()){
int i = q.top();
q.pop();
s.erase(i);
vector<int> v;
for(auto j:adjj[i]){
if(s.find(j)==s.end()){
v.push_back(j);
}
}
int sz = v.size();
int l = 0 , r = sz-1, ans = -1;
while(l<=r){
int mid = (l+r)/2;
vector<int> e(n,n);
e[i] = -1;
for(int j = 0;j<=mid;j++){
e[v[j]] = -1;
}
int cnt = 0;
for(int j = 0;j<n;j++){
pr[j] = j;cnt++;
}
for(int j = 0;j<X.size();j++){
if(e[X[j]]==-1&&e[Y[j]]==-1){
if(X[j]==i||Y[j]==i)continue;
if(col[X[j]]==col[Y[j]])cnt-=mergegroup(X[j],Y[j]);
}else if(e[X[j]]==n&&e[Y[j]]==n){
cnt-=mergegroup(X[j],Y[j]);
}
}
if(cnt==perform_experiment(e)){
l = mid+1;
}else {
ans = mid;
r = mid-1;
}
}
if(ans==-1){
col[i] = d++;
}else{
col[i] = col[v[ans]];
}
for(auto j:adjj[i]){
if(s.find(j)!=s.end()){
q.push(j);
}
}
}
}
return col;
}