oj.uz

Submission #759345

# Submission time Handle Problem Language Result Execution time Memory
759345 2023-06-16T08:11:26 Z Sharky Connecting Supertrees (IOI20_supertrees) C++17
11 / 100
 163 ms 23676 KB 
#include "supertrees.h"
#include <bits/stdc++.h>
using namespace std;

const int maxn = 1005;
int N, fa[maxn], sz[maxn];
bool ok = true;

int find(int u) {
    return (fa[u] == u) ? u : fa[u] = find(fa[u]);
}

bool merge(int u, int v) {
    u = find(u), v = find(v);
    if (u == v) return false;
    if (sz[v] > sz[u]) swap(u, v);
    fa[v] = u;
    sz[u] += sz[v];
    return true;
}

bool vis[maxn];
vector<int> edits;

void dfs(int u, vector<vector<int>>& ans, vector<vector<int>>& p) {
    vis[u] = true;
    edits.push_back(u);
    for (int i = 0; i < N; i++) {
        if (!vis[i] && p[u][i] == 1 && find(u) == find(i)) {
            ans[u][i] = ans[i][u] = 1;
            dfs(i, ans, p);
        }
    }
}

void process(vector<int> v, vector<vector<int>>& ans, vector<vector<int>>& p) {
    if ((int) v.size() == 1) return;
    // for (int i = 0; i < v.size(); i++) cout << v[i] << endl;
    int n = v.size(), cnt = 0;
    vector<int> cyc;
    vector<vector<int>> seen(n);
    vector<bool> in_cyc(n, 0);
    bool two = false;
    for (int i : v) {
        vector<bool> b(3, 0);
        for (int j = 0; j < N; j++) {
            if (i == j) continue;
            b[p[i][j]] = true;
            // cout << i << ' ' << j << ' ' << p[i][j] << '\n';
        }
        for (int j = 0; j < N; j++) {
            if (p[i][j] == 1) merge(i, j);
        }
        if (b[2] && !b[1]) {
            cyc.push_back(i);
            // cout << "in cycle!" << ' ' << i << endl;
            in_cyc[i] = true;
        }
        two = (two | b[2]);
    }
    // cout << "alive\n";
    for (int i = 0; i < n; i++) if (!in_cyc[v[i]]) {
        seen[find(v[i])].push_back(v[i]);
        // cout << "corresponding find: " << v[i] << ' ' << find(v[i]) << '\n';
    }
    // cout << "alive\n";
    for (int i = 0; i < n; i++) {
        if (seen[i].empty()) continue;
        dfs(seen[i].front(), ans, p);
        cyc.push_back(seen[i].front());
        in_cyc[seen[i].front()] = true;
        seen[find(seen[i].front())].push_back(seen[i].front());
        for (int j : edits) vis[j] = 0;
    }
    if (two && ((int) cyc.size() == 1 || (int) cyc.size() == 2)) {
        ok = false;
        return;
    }
    for (int i = 0; i < n; i++) if (!seen[i].empty()) cnt++;
    if (cnt > (int) cyc.size()) {
        ok = false;
        return;
    }
    for (int i = 0; i < cyc.size(); i++) {
        int nxt = (i + 1) % cyc.size();
        ans[cyc[i]][cyc[nxt]] = ans[cyc[nxt]][cyc[i]] = 1;
    }
}

int construct(std::vector<std::vector<int>> p) {
    int n = p[0].size();
    N = n;
    vector<vector<int>> ans(n, vector<int> (n, 0));
    for (int i = 0; i < n; i++) {
        fa[i] = i, sz[i] = 1;
        for (int j = 0; j < n; j++) {
            if (p[i][j] >= 3) return 0;
        }
    }
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < n; j++) {
            if (p[i][j]) merge(i, j);
        }
    }
    for (int i = 0; i < n; i++) for (int j = 0; j < n; j++) {
        if (!p[i][j] && find(i) == find(j)) return 0;
    }
    vector<vector<int>> g(n);
    for (int i = 0; i < n; i++) g[find(i)].push_back(i);
    for (int i = 0; i < n; i++) fa[i] = i, sz[i] = 1;
    for (int i = 0; i < n; i++) {
        if (!g[i].empty()) {
            process(g[i], ans, p);
            if (!ok) return 0;
        }
    }
    for (int i = 0; i < n; i++) ans[i][i] = 0;
    build(ans);
    return 1;
}

Compilation message

supertrees.cpp: In function 'void process(std::vector<int>, std::vector<std::vector<int> >&, std::vector<std::vector<int> >&)':
supertrees.cpp:84:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   84 |     for (int i = 0; i < cyc.size(); i++) {
      |                     ~~^~~~~~~~~~~~

Subtask #1
11.0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 212 KB Output is correct
2 Correct 1 ms 212 KB Output is correct
3 Correct 0 ms 212 KB Output is correct
4 Correct 0 ms 308 KB Output is correct
5 Correct 1 ms 300 KB Output is correct
6 Correct 7 ms 1236 KB Output is correct
7 Correct 163 ms 23676 KB Output is correct

Subtask #2
0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 212 KB Output is correct
2 Correct 1 ms 212 KB Output is correct
3 Correct 0 ms 212 KB Output is correct
4 Correct 0 ms 308 KB Output is correct
5 Correct 1 ms 300 KB Output is correct
6 Correct 7 ms 1236 KB Output is correct
7 Correct 163 ms 23676 KB Output is correct
8 Correct 1 ms 212 KB Output is correct
9 Correct 1 ms 212 KB Output is correct
10 Correct 1 ms 212 KB Output is correct
11 Correct 1 ms 212 KB Output is correct
12 Correct 6 ms 1212 KB Output is correct
13 Correct 146 ms 23608 KB Output is correct
14 Correct 1 ms 212 KB Output is correct
15 Correct 1 ms 212 KB Output is correct
16 Correct 4 ms 852 KB Output is correct
17 Correct 87 ms 13676 KB Output is correct
18 Correct 1 ms 212 KB Output is correct
19 Correct 1 ms 212 KB Output is correct
20 Runtime error 26 ms 7008 KB Execution killed with signal 11
21 Halted 0 ms 0 KB -

Subtask #3
0 / 19.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 212 KB Output is correct
2 Correct 0 ms 212 KB Output is correct
3 Correct 0 ms 212 KB Output is correct
4 Correct 0 ms 212 KB Output is correct
5 Correct 0 ms 212 KB Output is correct
6 Correct 0 ms 212 KB Output is correct
7 Correct 0 ms 212 KB Output is correct
8 Correct 6 ms 1108 KB Output is correct
9 Correct 144 ms 22032 KB Output is correct
10 Correct 0 ms 212 KB Output is correct
11 Correct 0 ms 212 KB Output is correct
12 Correct 7 ms 1108 KB Output is correct
13 Correct 154 ms 22020 KB Output is correct
14 Correct 0 ms 212 KB Output is correct
15 Correct 0 ms 212 KB Output is correct
16 Correct 3 ms 724 KB Output is correct
17 Correct 71 ms 12168 KB Output is correct
18 Correct 0 ms 212 KB Output is correct
19 Correct 0 ms 212 KB Output is correct
20 Correct 0 ms 212 KB Output is correct
21 Runtime error 18 ms 6536 KB Execution killed with signal 6
22 Halted 0 ms 0 KB -

Subtask #4
0 / 35.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 212 KB Output is correct
2 Correct 0 ms 212 KB Output is correct
3 Correct 0 ms 212 KB Output is correct
4 Runtime error 19 ms 6900 KB Execution killed with signal 11
5 Halted 0 ms 0 KB -

Subtask #5
0 / 21.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 212 KB Output is correct
2 Correct 1 ms 212 KB Output is correct
3 Correct 0 ms 212 KB Output is correct
4 Correct 0 ms 308 KB Output is correct
5 Correct 1 ms 300 KB Output is correct
6 Correct 7 ms 1236 KB Output is correct
7 Correct 163 ms 23676 KB Output is correct
8 Correct 1 ms 212 KB Output is correct
9 Correct 1 ms 212 KB Output is correct
10 Correct 1 ms 212 KB Output is correct
11 Correct 1 ms 212 KB Output is correct
12 Correct 6 ms 1212 KB Output is correct
13 Correct 146 ms 23608 KB Output is correct
14 Correct 1 ms 212 KB Output is correct
15 Correct 1 ms 212 KB Output is correct
16 Correct 4 ms 852 KB Output is correct
17 Correct 87 ms 13676 KB Output is correct
18 Correct 1 ms 212 KB Output is correct
19 Correct 1 ms 212 KB Output is correct
20 Runtime error 26 ms 7008 KB Execution killed with signal 11
21 Halted 0 ms 0 KB -

Subtask #6
0 / 4.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 212 KB Output is correct
2 Correct 1 ms 212 KB Output is correct
3 Correct 0 ms 212 KB Output is correct
4 Correct 0 ms 308 KB Output is correct
5 Correct 1 ms 300 KB Output is correct
6 Correct 7 ms 1236 KB Output is correct
7 Correct 163 ms 23676 KB Output is correct
8 Correct 1 ms 212 KB Output is correct
9 Correct 1 ms 212 KB Output is correct
10 Correct 1 ms 212 KB Output is correct
11 Correct 1 ms 212 KB Output is correct
12 Correct 6 ms 1212 KB Output is correct
13 Correct 146 ms 23608 KB Output is correct
14 Correct 1 ms 212 KB Output is correct
15 Correct 1 ms 212 KB Output is correct
16 Correct 4 ms 852 KB Output is correct
17 Correct 87 ms 13676 KB Output is correct
18 Correct 1 ms 212 KB Output is correct
19 Correct 1 ms 212 KB Output is correct
20 Runtime error 26 ms 7008 KB Execution killed with signal 11
21 Halted 0 ms 0 KB -