oj.uz

Submission #910060

# Submission time Handle Problem Language Result Execution time Memory
910060 2024-01-17T20:18:39 Z asdasdqwer Pipes (CEOI15_pipes) C++14
40 / 100
 802 ms 23652 KB 
#include <bits/stdc++.h>
using namespace std;

struct DSU {
    vector<int> p,r;
    DSU(int n) :p(n),r(n) {
        for (int i=0;i<n;i++)p[i]=i;
    }

    int get(int i) {
        if (p[i]!=i && p[i]!=p[p[i]]) p[i]=get(p[i]);
        return p[i];
    }

    bool unite(int a,int b) {
        a=get(a),b=get(b);
        if (a==b)return false;
        if (r[a]<r[b])swap(a,b);
        p[b]=a;
        if (r[a]==r[b])r[a]++;
        return true;
    }
};

signed main(){
    ios::sync_with_stdio(false);
    cin.tie(0);

    int n,m;cin>>n>>m;
    DSU d1(n), d2(n);
    vector<array<int,2>> edg;

    for (int i=0;i<m;i++) {
        int a,b;cin>>a>>b;a--;b--;
        if (d1.unite(a,b)) {
            edg.push_back({a,b});
        }

        else {
            d2.unite(a,b);
        }
    }

    vector<vector<int>> g(n);
    vector<int> cnt(n,0);
    for (auto x:edg) {
        g[x[0]].push_back(x[1]);
        g[x[1]].push_back(x[0]);
    }

    vector<vector<int>> lft(n, vector<int>(20, -1));
    bitset<100001> vis;
    vector<int> d(n,0);

    function<void(int,int)> dfs=[&](int node, int pv) {
        vis[node]=true;
        for (int x:g[node]) {
            if (x==pv)continue;
            lft[x][0]=node;
            d[x]=d[node]+1;
            dfs(x, node);
        }
    };

    for (int i=0;i<n;i++) {
        if (!vis[i]) dfs(i,-1);
    }

    for (int j=1;j<20;j++) {
        for (int i=0;i<n;i++) {
            lft[i][j]=lft[i][j-1];
            if (lft[i][j] != -1) {
                lft[i][j]=lft[lft[i][j]][j-1];
            }
        }
    }

    vector<vector<int>> bkt(n);

    for (int i=0;i<n;i++) {
        bkt[d2.get(i)].push_back(i);
    }

    function<int(int,int)> jmp=[&](int a, int steps) -> int {
        int cnt=0;
        while (steps) {
            if ((steps&1)==1) {
                a=lft[a][cnt];
                if (a==-1)return -1;
            }
            steps>>=1;cnt++;
        }
        return a;
    };

    function<int(int,int)> lca=[&](int a,int b) -> int {
        if (d[a]<d[b])swap(a,b);
        a=jmp(a,d[a]-d[b]);
        if (a==b)return a;

        for (int i=19;i>=0;i--) {
            if (lft[a][i] != lft[b][i]) {
                a=lft[a][i];
                b=lft[b][i];
            }
        }
        return lft[a][0];
    };

    vector<int> dp(n, 0);

    for (int i=0;i<n;i++) {
        if (bkt[i].size() < 2) continue;
        
        int st = bkt[i][0];

        for (int x:bkt[i]) {
            if (x==st)continue;
            int l=lca(st, x);
            dp[st]++;
            dp[x]++;
            dp[l] -= 2;
        }
    }

    for (int i=0;i<n;i++) vis[i]=0;
    
    function<void(int)> dfs2=[&](int node) {
        vis[node]=true;
        for (int x:g[node]) {
            if (vis[x])continue;
            dfs2(x);
            dp[node]+=dp[x];
        }
    };

    for (int i=0;i<n;i++) {
        if (!vis[i])dfs2(i);
    }

    for (int i=0;i<n;i++) {
        if (dp[i]==0 && lft[i][0] != -1) {
            cout<<i+1<<" "<<lft[i][0]+1<<"\n";
        }
    }
}

Subtask #1
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 344 KB Output is correct
2 Correct 0 ms 348 KB Output is correct

Subtask #2
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 1372 KB Output is correct
2 Correct 3 ms 1372 KB Output is correct

Subtask #3
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 65 ms 1108 KB Output is correct
2 Correct 63 ms 1104 KB Output is correct

Subtask #4
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 109 ms 2584 KB Output is correct
2 Correct 139 ms 2584 KB Output is correct

Subtask #5
0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 186 ms 5836 KB Output is correct
2 Runtime error 179 ms 21436 KB Memory limit exceeded
3 Halted 0 ms 0 KB -

Subtask #6
0 / 10.0

# Verdict Execution time Memory Grader output
1 Runtime error 269 ms 17068 KB Memory limit exceeded
2 Halted 0 ms 0 KB -

Subtask #7
0 / 10.0

# Verdict Execution time Memory Grader output
1 Runtime error 401 ms 19376 KB Memory limit exceeded
2 Halted 0 ms 0 KB -

Subtask #8
0 / 10.0

# Verdict Execution time Memory Grader output
1 Runtime error 547 ms 23652 KB Memory limit exceeded
2 Halted 0 ms 0 KB -

Subtask #9
0 / 10.0

# Verdict Execution time Memory Grader output
1 Runtime error 752 ms 23648 KB Memory limit exceeded
2 Halted 0 ms 0 KB -

Subtask #10
0 / 10.0

# Verdict Execution time Memory Grader output
1 Runtime error 802 ms 22504 KB Memory limit exceeded
2 Halted 0 ms 0 KB -