Submission #13407

#	Time	Username	Problem	Language	Result	Execution time	Memory
13407		gs14004	우호 조약 체결 (JOI14_friends)	C++14	0 / 100	56 ms	7892 KiB

friends

#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);
}

• Subtask 1: 0 / 5
• Subtask 2: 0 / 30
• Subtask 3: 0 / 65