제출 #599476

#제출 시각아이디문제언어결과실행 시간메모리
599476SlavicGStranded Far From Home (BOI22_island)C++17
0 / 100
554 ms66024 KiB
#include "bits/stdc++.h"
using namespace std;
using ll = long long;

const int N = 2e5 + 10;
ll cnt[N], p[N], c[N], mx[N];
set<int> s[N];
vector<int> adj[N];
int get(int a) {return p[a] = (p[a] == a ? a : get(p[a]));}

int uni(int a, int b, int vb) {
    a = get(a), b = get(b);
    bool swapped = false;
    if(s[a].size() > s[b].size()) {
        swap(a, b);
        swapped = true;
    }
    if(!swapped) {
        if(cnt[a] >= vb) {
            for(auto x: s[a]) s[b].insert(x);
        }
    } else if(cnt[b] >= vb){
        for(auto x: s[a]) s[b].insert(x);
    }

    p[a] = b;
    cnt[b] += cnt[a];
    return b;
}
int main() {
    int n, m; cin >> n >> m;
    vector<pair<int, int>> paiu;
    for(int i = 0; i < n; ++i) {
        cin >> c[i];
        paiu.push_back({c[i], i});
    }
    for(int i = 0; i < n; ++i) {
        p[i] = i;
        cnt[i] = c[i];
        s[i].insert(i);
    }
    sort(paiu.begin(), paiu.end());
    vector<bool> vis(n, false);
    for(int i = 0; i < m; ++i) {
        int u, v; cin >> u >> v; --u, --v;
        adj[u].push_back(v);
        adj[v].push_back(u);
    }

    int root = -1;
    for(int i = 0; i < n; ++i) {
        int u = paiu[i].second;
        for(int v: adj[u]) {
            if(!vis[v]) continue;
            if(get(u) != get(v)) root = uni(v, u, c[u]);
        }
        vis[u] = true;
    }

    if(n == 1) {
        cout << "1\n";
    } else {
        vector<int> ans(n, 0);
        for(auto x: s[root]) ans[x] = 1;
        for(auto x: ans) cout << x;
    }
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...