oj.uz

Submission #1072042

# Submission time Handle Problem Language Result Execution time Memory
1072042 2024-08-23T13:38:30 Z shmax Fountain Parks (IOI21_parks) C++17
55 / 100
 1855 ms 83904 KB 
#include "parks.h"
#include <bits/stdc++.h>


#pragma GCC optimize("Ofast")
#pragma GCC target("avx,avx2,fma")
#pragma GCC optimization("unroll-loops")

using namespace std;

using i32 = int;
//#define int long long
#define len(x) (int)(x.size())
template<typename T>
using vec = vector<T>;

struct DSU {
public:
    DSU() : _n(0) {}

    explicit DSU(int n) : _n(n), parent_or_size(n, -1) {}

    int unite(int a, int b) {
        assert(0 <= a && a < _n);
        assert(0 <= b && b < _n);
        int x = leader(a), y = leader(b);
        if (x == y) return x;
        if (-parent_or_size[x] < -parent_or_size[y]) std::swap(x, y);
        parent_or_size[x] += parent_or_size[y];
        parent_or_size[y] = x;
        return x;
    }

    bool one(int a, int b) {
        assert(0 <= a && a < _n);
        assert(0 <= b && b < _n);
        return leader(a) == leader(b);
    }

    int leader(int a) {
        assert(0 <= a && a < _n);
        if (parent_or_size[a] < 0) return a;
        return parent_or_size[a] = leader(parent_or_size[a]);
    }

    int size(int a) {
        assert(0 <= a && a < _n);
        return -parent_or_size[leader(a)];
    }

    std::vector<std::vector<int>> groups() {
        std::vector<int> leader_buf(_n), group_size(_n);
        for (int i = 0; i < _n; i++) {
            leader_buf[i] = leader(i);
            group_size[leader_buf[i]]++;
        }
        std::vector<std::vector<int>> result(_n);
        for (int i = 0; i < _n; i++) {
            result[i].reserve(group_size[i]);
        }
        for (int i = 0; i < _n; i++) {
            result[leader_buf[i]].push_back(i);
        }
        result.erase(
                std::remove_if(result.begin(), result.end(),
                               [&](const std::vector<int> &v) { return v.empty(); }),
                result.end());
        return result;
    }

private:
    int _n;
    // root node: -1 * component size
    // otherwise: parent
    std::vector<int> parent_or_size;
};


i32 construct_roads(std::vector<i32> _x, std::vector<i32> _y) {
    int n = len(_x);
    map<pair<int, int>, int> mp;
    for (int i = 0; i < n; i++) {
        mp[{_x[i], _y[i]}] = i;
    }
    mt19937 rnd(chrono::steady_clock::now().time_since_epoch().count());
    vec<pair<int, int>> edges;
    for (int i = 0; i < n; i++) {
        vec<pair<int, int>> moves = {{-2, 0},
                                     {2,  0}};
        for (auto [dx, dy]: moves) {
            int nx = _x[i] + dx;
            int ny = _y[i] + dy;
            if (!mp.count({nx, ny})) continue;
            edges.push_back({i, mp[{nx, ny}]});
        }
    }
    for (int i = 0; i < n; i++) {
        vec<pair<int, int>> moves = {{0, -2},
                                     {0, 2}};
        for (auto [dx, dy]: moves) {
            int nx = _x[i] + dx;
            int ny = _y[i] + dy;
            if (!mp.count({nx, ny})) continue;
            edges.push_back({i, mp[{nx, ny}]});
        }
    }
    for (int times = 0; times < 5; times++) {
        DSU dsu(n);


        vec<pair<int, int>> used_edges;
        for (auto [a, b]: edges) {
            if (dsu.one(a, b)) continue;
            dsu.unite(a, b);
            used_edges.emplace_back(a, b);
        }
        if (dsu.size(0) != n) return 0;
        shuffle(used_edges.begin(), used_edges.end(), rnd);
        map<pair<int, int>, int> benches;
        vec<pair<int, int>> benches_pos;
        int idx = 0;
        for (auto [a, b]: used_edges) {
            auto [x1, y1] = pair<int, int>{_x[a], _y[a]};
            auto [x2, y2] = pair<int, int>{_x[b], _y[b]};
            if (x1 == x2) {
                if (!benches.count({x1 - 1, (y1 + y2) / 2}))
                    benches[{x1 - 1, (y1 + y2) / 2}] = idx++, benches_pos.emplace_back(x1 - 1, (y1 + y2) / 2);
                if (!benches.count({x1 + 1, (y1 + y2) / 2}))
                    benches[{x1 + 1, (y1 + y2) / 2}] = idx++, benches_pos.emplace_back(x1 + 1, (y1 + y2) / 2);
            } else {
                if (!benches.count({(x1 + x2) / 2, y1 - 1}))
                    benches[{(x1 + x2) / 2, y1 - 1}] = idx++, benches_pos.emplace_back((x1 + x2) / 2, y1 - 1);
                if (!benches.count({(x1 + x2) / 2, y1 + 1}))
                    benches[{(x1 + x2) / 2, y1 + 1}] = idx++, benches_pos.emplace_back((x1 + x2) / 2, y1 + 1);
            }
        }
        vec<vec<int>> g(n - 1 + len(benches));
        int idt = 0;
        for (auto [a, b]: used_edges) {
            auto [x1, y1] = pair<int, int>{_x[a], _y[a]};
            auto [x2, y2] = pair<int, int>{_x[b], _y[b]};
            if (x1 == x2) {
                g[idt].push_back(n - 1 + benches[{x1 - 1, (y1 + y2) / 2}]);
                g[idt].push_back(n - 1 + benches[{x1 + 1, (y1 + y2) / 2}]);
            } else {
                g[idt].push_back(n - 1 + benches[{(x1 + x2) / 2, y1 - 1}]);
                g[idt].push_back(n - 1 + benches[{(x1 + x2) / 2, y1 + 1}]);
            }
            idt++;
        }

//        matching
//        shuffle
        for (int i = 0; i < n - 1; i++) {
            shuffle(g[i].begin(), g[i].end(), rnd);
        }

        vec<int> match(n - 1 + len(benches), -1);
        vec<bool> used(n - 1 + len(benches), false);
        function<bool(int)> dfs = [&](int v) -> bool {
            if (used[v]) return false;
            used[v] = true;
            for (auto u: g[v]) {
                if (match[u] == -1 || dfs(match[u])) {
                    match[u] = v;
                    match[v] = u;
                    return true;
                }
            }
            return false;
        };

        bool flag = true;
        do {
            flag = false;
            fill(used.begin(), used.end(), false);
            for (int i = 0; i < n - 1; i++) {
                if (match[i] == -1) {
                    if (dfs(i)) {
                        flag = true;
                    }
                }
            }
        } while (flag);

        int cnt = 0;
        for (int i = 0; i < n - 1; i++) {
            if (match[i] != -1) cnt++;
        }

        if (cnt == n - 1) {
            vec<i32> x_benches, y_benches;
            vec<i32> u, v;
            for (int i = 0; i < n - 1; i++) {
                x_benches.push_back(benches_pos[match[i] - n + 1].first);
                y_benches.push_back(benches_pos[match[i] - n + 1].second);
                u.push_back(used_edges[i].first);
                v.push_back(used_edges[i].second);
            }
            build(u, v, x_benches, y_benches);
            return 1;
        }


    }

    return 0;
}

Compilation message

parks.cpp:7: warning: ignoring '#pragma GCC optimization' [-Wunknown-pragmas]
    7 | #pragma GCC optimization("unroll-loops")
      |

Subtask #1
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 1 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 0 ms 348 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
6 Correct 0 ms 348 KB Output is correct
7 Correct 1 ms 348 KB Output is correct
8 Correct 0 ms 348 KB Output is correct
9 Correct 309 ms 41468 KB Output is correct
10 Correct 18 ms 4444 KB Output is correct
11 Correct 126 ms 22476 KB Output is correct
12 Correct 27 ms 6476 KB Output is correct
13 Correct 34 ms 5368 KB Output is correct
14 Correct 1 ms 344 KB Output is correct
15 Correct 3 ms 604 KB Output is correct
16 Correct 288 ms 41408 KB Output is correct

Subtask #2
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 1 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 0 ms 348 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
6 Correct 0 ms 348 KB Output is correct
7 Correct 1 ms 348 KB Output is correct
8 Correct 0 ms 348 KB Output is correct
9 Correct 309 ms 41468 KB Output is correct
10 Correct 18 ms 4444 KB Output is correct
11 Correct 126 ms 22476 KB Output is correct
12 Correct 27 ms 6476 KB Output is correct
13 Correct 34 ms 5368 KB Output is correct
14 Correct 1 ms 344 KB Output is correct
15 Correct 3 ms 604 KB Output is correct
16 Correct 288 ms 41408 KB Output is correct
17 Correct 1 ms 348 KB Output is correct
18 Correct 1 ms 348 KB Output is correct
19 Correct 0 ms 348 KB Output is correct
20 Correct 0 ms 348 KB Output is correct
21 Correct 0 ms 348 KB Output is correct
22 Correct 0 ms 348 KB Output is correct
23 Correct 703 ms 69756 KB Output is correct
24 Correct 1 ms 344 KB Output is correct
25 Correct 2 ms 856 KB Output is correct
26 Correct 3 ms 860 KB Output is correct
27 Correct 5 ms 860 KB Output is correct
28 Correct 220 ms 26808 KB Output is correct
29 Correct 398 ms 40900 KB Output is correct
30 Correct 551 ms 55424 KB Output is correct
31 Correct 703 ms 68260 KB Output is correct
32 Correct 0 ms 348 KB Output is correct
33 Correct 0 ms 348 KB Output is correct
34 Correct 1 ms 344 KB Output is correct
35 Correct 0 ms 348 KB Output is correct
36 Correct 0 ms 348 KB Output is correct
37 Correct 1 ms 344 KB Output is correct
38 Correct 0 ms 348 KB Output is correct
39 Correct 0 ms 348 KB Output is correct
40 Correct 0 ms 344 KB Output is correct
41 Correct 0 ms 348 KB Output is correct
42 Correct 0 ms 348 KB Output is correct
43 Correct 2 ms 600 KB Output is correct
44 Correct 3 ms 604 KB Output is correct
45 Correct 302 ms 35264 KB Output is correct
46 Correct 482 ms 52876 KB Output is correct
47 Correct 451 ms 52820 KB Output is correct

Subtask #3
0 / 15.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 1 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 0 ms 348 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
6 Correct 0 ms 348 KB Output is correct
7 Correct 1 ms 348 KB Output is correct
8 Correct 0 ms 348 KB Output is correct
9 Correct 309 ms 41468 KB Output is correct
10 Correct 18 ms 4444 KB Output is correct
11 Correct 126 ms 22476 KB Output is correct
12 Correct 27 ms 6476 KB Output is correct
13 Correct 34 ms 5368 KB Output is correct
14 Correct 1 ms 344 KB Output is correct
15 Correct 3 ms 604 KB Output is correct
16 Correct 288 ms 41408 KB Output is correct
17 Correct 1 ms 348 KB Output is correct
18 Correct 1 ms 348 KB Output is correct
19 Correct 0 ms 348 KB Output is correct
20 Correct 0 ms 348 KB Output is correct
21 Correct 0 ms 348 KB Output is correct
22 Correct 0 ms 348 KB Output is correct
23 Correct 703 ms 69756 KB Output is correct
24 Correct 1 ms 344 KB Output is correct
25 Correct 2 ms 856 KB Output is correct
26 Correct 3 ms 860 KB Output is correct
27 Correct 5 ms 860 KB Output is correct
28 Correct 220 ms 26808 KB Output is correct
29 Correct 398 ms 40900 KB Output is correct
30 Correct 551 ms 55424 KB Output is correct
31 Correct 703 ms 68260 KB Output is correct
32 Correct 0 ms 348 KB Output is correct
33 Correct 0 ms 348 KB Output is correct
34 Correct 1 ms 344 KB Output is correct
35 Correct 0 ms 348 KB Output is correct
36 Correct 0 ms 348 KB Output is correct
37 Correct 1 ms 344 KB Output is correct
38 Correct 0 ms 348 KB Output is correct
39 Correct 0 ms 348 KB Output is correct
40 Correct 0 ms 344 KB Output is correct
41 Correct 0 ms 348 KB Output is correct
42 Correct 0 ms 348 KB Output is correct
43 Correct 2 ms 600 KB Output is correct
44 Correct 3 ms 604 KB Output is correct
45 Correct 302 ms 35264 KB Output is correct
46 Correct 482 ms 52876 KB Output is correct
47 Correct 451 ms 52820 KB Output is correct
48 Correct 0 ms 348 KB Output is correct
49 Correct 0 ms 348 KB Output is correct
50 Correct 0 ms 348 KB Output is correct
51 Correct 0 ms 348 KB Output is correct
52 Correct 0 ms 348 KB Output is correct
53 Correct 0 ms 348 KB Output is correct
54 Correct 0 ms 348 KB Output is correct
55 Incorrect 1855 ms 55364 KB Solution announced impossible, but it is possible.
56 Halted 0 ms 0 KB -

Subtask #4
20.0 / 20.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 1 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 0 ms 348 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
6 Correct 0 ms 348 KB Output is correct
7 Correct 1 ms 348 KB Output is correct
8 Correct 0 ms 348 KB Output is correct
9 Correct 309 ms 41468 KB Output is correct
10 Correct 18 ms 4444 KB Output is correct
11 Correct 126 ms 22476 KB Output is correct
12 Correct 27 ms 6476 KB Output is correct
13 Correct 34 ms 5368 KB Output is correct
14 Correct 1 ms 344 KB Output is correct
15 Correct 3 ms 604 KB Output is correct
16 Correct 288 ms 41408 KB Output is correct
17 Correct 1 ms 348 KB Output is correct
18 Correct 0 ms 348 KB Output is correct
19 Correct 0 ms 348 KB Output is correct
20 Correct 695 ms 70400 KB Output is correct
21 Correct 731 ms 66236 KB Output is correct
22 Correct 751 ms 68940 KB Output is correct
23 Correct 596 ms 73144 KB Output is correct
24 Correct 215 ms 16724 KB Output is correct
25 Correct 356 ms 23744 KB Output is correct
26 Correct 269 ms 23364 KB Output is correct
27 Correct 684 ms 82752 KB Output is correct
28 Correct 746 ms 82564 KB Output is correct
29 Correct 753 ms 82368 KB Output is correct
30 Correct 798 ms 82452 KB Output is correct
31 Correct 1 ms 344 KB Output is correct
32 Correct 33 ms 5196 KB Output is correct
33 Correct 114 ms 8432 KB Output is correct
34 Correct 739 ms 71364 KB Output is correct
35 Correct 11 ms 1504 KB Output is correct
36 Correct 68 ms 6072 KB Output is correct
37 Correct 156 ms 11968 KB Output is correct
38 Correct 229 ms 27056 KB Output is correct
39 Correct 341 ms 36692 KB Output is correct
40 Correct 471 ms 48316 KB Output is correct
41 Correct 588 ms 57532 KB Output is correct
42 Correct 744 ms 66756 KB Output is correct
43 Correct 0 ms 344 KB Output is correct
44 Correct 0 ms 348 KB Output is correct
45 Correct 0 ms 348 KB Output is correct
46 Correct 0 ms 348 KB Output is correct
47 Correct 0 ms 348 KB Output is correct
48 Correct 0 ms 348 KB Output is correct
49 Correct 0 ms 348 KB Output is correct
50 Correct 0 ms 344 KB Output is correct
51 Correct 1 ms 348 KB Output is correct
52 Correct 0 ms 348 KB Output is correct
53 Correct 0 ms 348 KB Output is correct
54 Correct 2 ms 648 KB Output is correct
55 Correct 3 ms 600 KB Output is correct
56 Correct 305 ms 35452 KB Output is correct
57 Correct 494 ms 53056 KB Output is correct
58 Correct 486 ms 53792 KB Output is correct
59 Correct 0 ms 344 KB Output is correct
60 Correct 0 ms 348 KB Output is correct
61 Correct 0 ms 344 KB Output is correct
62 Correct 750 ms 82620 KB Output is correct
63 Correct 730 ms 83276 KB Output is correct
64 Correct 737 ms 82108 KB Output is correct
65 Correct 6 ms 856 KB Output is correct
66 Correct 8 ms 1372 KB Output is correct
67 Correct 293 ms 34992 KB Output is correct
68 Correct 499 ms 54460 KB Output is correct
69 Correct 733 ms 71872 KB Output is correct

Subtask #5
20.0 / 20.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 1 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 0 ms 348 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
6 Correct 0 ms 348 KB Output is correct
7 Correct 1 ms 348 KB Output is correct
8 Correct 0 ms 348 KB Output is correct
9 Correct 309 ms 41468 KB Output is correct
10 Correct 18 ms 4444 KB Output is correct
11 Correct 126 ms 22476 KB Output is correct
12 Correct 27 ms 6476 KB Output is correct
13 Correct 34 ms 5368 KB Output is correct
14 Correct 1 ms 344 KB Output is correct
15 Correct 3 ms 604 KB Output is correct
16 Correct 288 ms 41408 KB Output is correct
17 Correct 690 ms 83652 KB Output is correct
18 Correct 719 ms 83904 KB Output is correct
19 Correct 708 ms 68796 KB Output is correct
20 Correct 794 ms 69164 KB Output is correct
21 Correct 660 ms 67236 KB Output is correct
22 Correct 1 ms 348 KB Output is correct
23 Correct 80 ms 10900 KB Output is correct
24 Correct 26 ms 2764 KB Output is correct
25 Correct 109 ms 8820 KB Output is correct
26 Correct 224 ms 14844 KB Output is correct
27 Correct 332 ms 34700 KB Output is correct
28 Correct 414 ms 43448 KB Output is correct
29 Correct 530 ms 52740 KB Output is correct
30 Correct 650 ms 60896 KB Output is correct
31 Correct 774 ms 69060 KB Output is correct
32 Correct 714 ms 72156 KB Output is correct
33 Correct 608 ms 83392 KB Output is correct
34 Correct 5 ms 860 KB Output is correct
35 Correct 9 ms 1512 KB Output is correct
36 Correct 293 ms 34832 KB Output is correct
37 Correct 476 ms 53964 KB Output is correct
38 Correct 732 ms 70848 KB Output is correct

Subtask #6
0 / 30.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 1 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 0 ms 348 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
6 Correct 0 ms 348 KB Output is correct
7 Correct 1 ms 348 KB Output is correct
8 Correct 0 ms 348 KB Output is correct
9 Correct 309 ms 41468 KB Output is correct
10 Correct 18 ms 4444 KB Output is correct
11 Correct 126 ms 22476 KB Output is correct
12 Correct 27 ms 6476 KB Output is correct
13 Correct 34 ms 5368 KB Output is correct
14 Correct 1 ms 344 KB Output is correct
15 Correct 3 ms 604 KB Output is correct
16 Correct 288 ms 41408 KB Output is correct
17 Correct 1 ms 348 KB Output is correct
18 Correct 1 ms 348 KB Output is correct
19 Correct 0 ms 348 KB Output is correct
20 Correct 0 ms 348 KB Output is correct
21 Correct 0 ms 348 KB Output is correct
22 Correct 0 ms 348 KB Output is correct
23 Correct 703 ms 69756 KB Output is correct
24 Correct 1 ms 344 KB Output is correct
25 Correct 2 ms 856 KB Output is correct
26 Correct 3 ms 860 KB Output is correct
27 Correct 5 ms 860 KB Output is correct
28 Correct 220 ms 26808 KB Output is correct
29 Correct 398 ms 40900 KB Output is correct
30 Correct 551 ms 55424 KB Output is correct
31 Correct 703 ms 68260 KB Output is correct
32 Correct 0 ms 348 KB Output is correct
33 Correct 0 ms 348 KB Output is correct
34 Correct 1 ms 344 KB Output is correct
35 Correct 0 ms 348 KB Output is correct
36 Correct 0 ms 348 KB Output is correct
37 Correct 1 ms 344 KB Output is correct
38 Correct 0 ms 348 KB Output is correct
39 Correct 0 ms 348 KB Output is correct
40 Correct 0 ms 344 KB Output is correct
41 Correct 0 ms 348 KB Output is correct
42 Correct 0 ms 348 KB Output is correct
43 Correct 2 ms 600 KB Output is correct
44 Correct 3 ms 604 KB Output is correct
45 Correct 302 ms 35264 KB Output is correct
46 Correct 482 ms 52876 KB Output is correct
47 Correct 451 ms 52820 KB Output is correct
48 Correct 0 ms 348 KB Output is correct
49 Correct 0 ms 348 KB Output is correct
50 Correct 0 ms 348 KB Output is correct
51 Correct 0 ms 348 KB Output is correct
52 Correct 0 ms 348 KB Output is correct
53 Correct 0 ms 348 KB Output is correct
54 Correct 0 ms 348 KB Output is correct
55 Incorrect 1855 ms 55364 KB Solution announced impossible, but it is possible.
56 Halted 0 ms 0 KB -