oj.uz

Submission #1098908

# Submission time Handle Problem Language Result Execution time Memory
1098908 2024-10-10T10:04:35 Z crafticat Stranded Far From Home (BOI22_island) C++17
0 / 100
 1000 ms 21388 KB 
#include <bits/stdc++.h>

#include <utility>

using namespace std;
#define F0R(i, n) for (int i= 0; i < n;i++)
template<typename T>
using V = vector<T>;
using vi = V<int>;
using vvi = V<vi>;
#define pb push_back
using pi = pair<int, int>;
#define all(x) begin(x), end(x)
#define f first
#define s second

V<int> initSum;

struct dsu {
    vi par;
    vi siz;
    vi sum;

    dsu(int n, vi initSum) {
        par.resize(n, -1);
        siz.resize(n, 1);
        sum = std::move(initSum);
    }

    int get(int x) {
        if (par[x] == -1) return x;
        return get(par[x]);
    }

    void combine(int a, int b) {
        a = get(a), b = get(b);
        if (a == b) return;

        if (initSum[b] > initSum[a]) swap(a,b);
        siz[a] += siz[b];
        sum[a] += sum[b];
        par[b] = a;
    }
};

//TODO CHECK CONNECTIVITY
vvi genGraph(vi par) {
    vvi g(par.size());

    F0R(i, par.size()) {
        if (par[i] == -1) continue;
        g[par[i]].pb(i);
        g[i].pb(par[i]);
    }
    return g;
}
V<bool> possible;
dsu* dsu;
vvi g;

void dfsPos(int x, int p) {
    if (p == -1 || dsu->sum[x] >= initSum[p]) possible[x] = true;
    else return;

    for (auto child : g[x]) {
        if (child == p) continue;
        dfsPos(child, x);
    }
}

int main() {
    ios_base::sync_with_stdio(false); cin.tie(nullptr);
    int n, m; cin >> n >> m;
    V<pair<int, pi>> edges;
    initSum.resize(n + 1);

    F0R(i, n)
        cin >> initSum[i + 1];

    F0R(i, m) {
        int a, b; cin >> a>> b;
        edges.pb({max(initSum[a], initSum[b]), {a, b}});
    }

    sort(all(edges));
    dsu = new struct dsu(n + 1, initSum);

    for (auto [x,pa] : edges) {
        auto [a,b] = pa;
        dsu->combine(a,b);
    }

    g = genGraph(dsu->par);
    int root = dsu->get(1);
    bool pos = true;
    F0R(i, n)
        if (dsu->get(i + 1) != root) pos = false;

    possible.resize(n + 1);

    if (pos) dfsPos(root, -1);
    F0R(i, n) {
        cout << possible[i + 1] << " ";
    }

    return 0;
}

Compilation message

island.cpp: In function 'vvi genGraph(vi)':
island.cpp:6:36: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
    6 | #define F0R(i, n) for (int i= 0; i < n;i++)
......
   50 |     F0R(i, par.size()) {
      |         ~~~~~~~~~~~~~               
island.cpp:50:5: note: in expansion of macro 'F0R'
   50 |     F0R(i, par.size()) {
      |     ^~~

Subtask #1
0 / 10.0

# Verdict Execution time Memory Grader output
1 Incorrect 0 ms 348 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #2
0 / 10.0

# Verdict Execution time Memory Grader output
1 Incorrect 0 ms 344 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #3
0 / 15.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Incorrect 94 ms 21388 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #4
0 / 30.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 344 KB Output is correct
2 Execution timed out 1070 ms 21180 KB Time limit exceeded
3 Halted 0 ms 0 KB -

Subtask #5
0 / 35.0

# Verdict Execution time Memory Grader output
1 Incorrect 0 ms 348 KB Output isn't correct
2 Halted 0 ms 0 KB -