| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 700845 | primenumber_zz | 철로 (IOI14_rail) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include<bits/stdc++.h>
using namespace std;
int N;
int cnt = 0,res = N-1;
vector<int>road[1515];
struct UnionFind {
vector<int>par,size,cnt;
vector<vector<int>>cld;
UnionFind() {};
UnionFind(int n) {
par.resize(n);
size.resize(n,1);
cnt.resize(n);
cld.resize(n);
for(int i = 0; i < n; i++) {
par[i] = i;
cld[i].push_back(i);
}
}
int find(int x) {
if(par[x] == x) {
return x;
}
return par[x] = find(par[x]);
}
bool same(int u,int v) {
u = find(u);
v = find(v);
return (u == v);
}
void unite(int u,int v) {
u = find(u);
v = find(v);
if(u == v) return;
if(size[u] > size[v]) swap(u,v);
for(int i:cld[u]) {
cld[v].push_back(i);
for(int j:road[i]) {
if(!same(i,j) && same(j,v)) {
cnt[v] -= 2;
}
}
}
par[u] = v;
size[v] += size[u];
cnt[v] += cnt[u];
}
int consize(int x) {
x = find(x);
return size[x];
}
int concnt(int x) {
x = find(x);
return cnt[x];
}
};
UnionFind uf;
void initialize(int n) {
N = n;
UnionFind init(n);
uf = init;
}
int hasEdge(int u, int v){
if(uf.same(u,v)) {
return 0;
}
int a = uf.consize(u);
int b = uf.concnt(u);
if(a*(N-a) == b+1) {
uf.unite(u,v);
res--;
return 1;
}
a = uf.consize(v);
b = uf.concnt(v);
if(a*(N-a) == b+1) {
uf.unite(u,v);
res--;
return 1;
}
if(N*(N-1)/2-cnt == res) {
res--;
uf.unite(u,v);
return 1;
}
uf.cnt[uf.find(u)]++;
uf.cnt[uf.find(v)]++;
cnt++;
return 0;
}
/*
int main() {
int N;
cin >> N;
initialize(N);
for(int i = 0; i < N*(N-1)/2; i++) {
int u,v;
cin >> u >> v;
cout << hasEdge(u,v) << endl;
}
}*/