oj.uz

Submission #106595

# Submission time Handle Problem Language Result Execution time Memory
106595 2019-04-19T07:11:14 Z teomrn Pipes (CEOI15_pipes) C++14
100 / 100
 1531 ms 8588 KB 
#include <stdio.h>
#include <algorithm>
#include <vector>
#include <numeric>
#include <assert.h>
using namespace std;

const int NMAX = 100010;

struct UF {
    int g[NMAX];
    int tata[NMAX];

    UF() {
        iota(tata, tata + NMAX, 0);
        fill(g, g + NMAX, 1);
    }

    int find(int x) {
        if (tata[tata[x]] != tata[x])
            tata[x] = find(tata[x]);
        return tata[x];
    }

    bool unite(int a, int b) {
        a = find(a);
        b = find(b);
        if (a == b)
            return 0;

        if (g[a] < g[b])
            swap(a, b);
        g[a] += g[b];
        tata[b] = a;

        return 1;
    }
} simple, strong;

namespace Tree {
    int h[NMAX];
    int father[NMAX];
    int up[NMAX]; /// ultimul cu care stiu sigur ca m-am unit
    vector <int> adia[NMAX]; /// suma ar trebui sa fie max NMAX :)
    int n;

    int find(int nod) {
        if (up[up[nod]] != up[nod])
            up[nod] = find(up[nod]);
        return up[nod];
    }

    void make_strong(int a, int b) {
        a = find(a);
        b = find(b);
        while (1) {
            if (a == b)
                return;

            if (h[a] < h[b])
                swap(a, b);
            /// acum il unesc pe a cu tatal sau
            assert(strong.find(a) != strong.find(father[a]));
            assert(father[a] != 0);
            strong.unite(a, father[a]);
            up[a] = father[a];
            a = find(a);
        }
    }

    void dfs(int nod, int tata) {
        father[nod] = tata;
        h[nod] = 1 + h[tata];
        up[nod] = (strong.find(nod) == strong.find(tata) ? find(tata) : nod);
        for (auto i : adia[nod])
            if (i != tata)
                dfs(i, nod);
    }

    void add_edge(int a, int b) {
        if (simple.g[simple.find(a)] < simple.g[simple.find(b)])
            swap(a, b);

        if (simple.unite(a, b)) {
            adia[a].push_back(b);
            adia[b].push_back(a);
            dfs(b, a);
        }
        else
            make_strong(a, b);
    }

    void dfs_ans(int nod, int tata) {
        for (auto i : adia[nod]) {
            if (i == tata)
                continue;
            if (strong.find(i) != strong.find(nod))
                printf("%d %d\n", i, nod);
            dfs_ans(i, nod);
        }
    }

    void get_ans() {
        for (int i = 1; i <= n; i++)
            if (father[i] == 0)
                dfs_ans(i, 0);
    }

}

int main()
{
    int n, m, a, b;
    scanf("%d%d", &n, &m);
    Tree::n = n;
    iota(Tree::up, Tree::up + NMAX, 0);

    while (m--) {
        scanf("%d%d", &a, &b);
        Tree::add_edge(a, b);
    }

    Tree::get_ans();

    return 0;
}

Compilation message

pipes.cpp: In function 'int main()':
pipes.cpp:114:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     scanf("%d%d", &n, &m);
     ~~~~~^~~~~~~~~~~~~~~~
pipes.cpp:119:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
         scanf("%d%d", &a, &b);
         ~~~~~^~~~~~~~~~~~~~~~

Subtask #1
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 5 ms 4580 KB Output is correct
2 Correct 5 ms 4608 KB Output is correct

Subtask #2
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 8 ms 4736 KB Output is correct
2 Correct 8 ms 4736 KB Output is correct

Subtask #3
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 115 ms 4856 KB Output is correct
2 Correct 113 ms 4856 KB Output is correct

Subtask #4
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 196 ms 4992 KB Output is correct
2 Correct 231 ms 4992 KB Output is correct

Subtask #5
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 328 ms 5588 KB Output is correct
2 Correct 300 ms 5752 KB Output is correct

Subtask #6
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 480 ms 7520 KB Output is correct
2 Correct 457 ms 7412 KB Output is correct

Subtask #7
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 756 ms 7792 KB Output is correct
2 Correct 755 ms 7720 KB Output is correct

Subtask #8
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 925 ms 8560 KB Output is correct
2 Correct 959 ms 8584 KB Output is correct

Subtask #9
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 1318 ms 8560 KB Output is correct
2 Correct 1339 ms 8584 KB Output is correct

Subtask #10
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 1531 ms 8368 KB Output is correct
2 Correct 1436 ms 8588 KB Output is correct