이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <cstdio>
#include <vector>
using namespace std;
struct disj{
int pa[200005], r, sz[200005];
void init(int n){
for(int i=1; i<=n; i++) pa[i] = i, sz[i] = 1;
}
int find(int x){
if(pa[x] == x) return x;
return pa[x] = find(pa[x]);
}
long long uni(int p, int q){
p = find(p);
q = find(q);
if(p == q) return 0;
long long ret = 1ll * sz[p] * sz[q];
if(r&1) pa[p] = q, sz[q] += sz[p];
else pa[q] = p, sz[p] += sz[q];
return ret;
}
}disj;
int n,m;
vector<int> g[100005];
int main(){
scanf("%d %d",&n,&m);
disj.init(n);
for (int i=0; i<m; i++) {
int x,y;
scanf("%d %d",&x,&y);
g[x].push_back(y);
}
long long ret = 0;
for (int i=1; i<=n; i++) {
for (int j=0; j+1<g[i].size(); j++) {
ret += disj.uni(g[i][j],g[i][j+1]) * 2;
}
}
for (int i=1; i<=n; i++) {
for (int &j : g[i]){
if(disj.find(i) != disj.find(j)) ret++;
}
}
printf("%lld",ret);
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |