Submission #1320478

#TimeUsernameProblemLanguageResultExecution timeMemory
1320478888313666Stranded Far From Home (BOI22_island)C++20
10 / 100
115 ms31076 KiB
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define _ <<' '<<
#define print(x) cout<<#x<<": "<<(x)<<'\n'

int n,m;
vector<vector<int>> adj;
vector<array<ll, 2>> s;
vector<int> idx, res;
vector<array<bool, 2>> vis;

struct dsu{
    vector<int> par;
    vector<ll> sz;

    dsu() {
        par.resize(n);
        iota(par.begin(), par.end(), 0);
        sz.resize(n);
        for (int i=0; i<n; i++) sz[i]=s[i][0];
    }

    int find(const int i) {
        if (par[i]==i) return i;
        return par[i]=find(par[i]);
    }

    bool unite(int i, int j) {
        i=find(i), j=find(j);
        if (i==j) return false;
        par[j]=i;
        sz[i]+=sz[j];
        return true;
    }

    ll size(const int i) {
        return sz[find(i)];
    }
};

void dfs(const int u, bool f) {
    vis[u][f]=1;
    f&=res[s[u][1]];
    res[s[u][1]]=f;
    vis[u][f]=1;
    for (const auto v:adj[u]) if (!vis[v][f]) dfs(v, f);
}

int main(){
    cin.tie(0)->sync_with_stdio(0);
    cout.tie(0);
    cin>>n>>m;
    adj.resize(n);
    s.resize(n);
    res.assign(n, 1);
    for (int i=0; i<n; i++) {
        cin>>s[i][0];
        s[i][1]=i;
    }
    sort(s.begin(), s.end());
    idx.resize(n);
    for (int i=0; i<n; i++) idx[s[i][1]]=i;
    for (int i=0; i<m; i++) {
        int a,b;
        cin>>a>>b;
        a=idx[--a];
        b=idx[--b];
        if (s[a][0]>=s[b][0]) adj[a].push_back(b);
        else adj[b].push_back(a);
    }
    dsu f;
    int l=0, r=0, w=0;
    priority_queue<pair<ll, int>, vector<pair<ll, int>>, greater<>> q;
    while (l<n) {
        w=s[l][0];
        while (r<n-1 && s[r+1][0]==w) ++r;
        ++r;
        while (!q.empty() && q.top().first<w) {
            const auto [d, i]=q.top();
            q.pop();
            if (f.size(i)<w) res[s[i][1]]=0;
        }
        for (int i=l; i<r; i++) for (const auto j:adj[i]) f.unite(i, j);
        for (int i=l; i<r; i++) q.push({f.size(i), i});
        l=r;
    }
    vis.assign(n, {0,0});
    for (int i=n-1; i>=0; --i) if (!vis[i][res[s[i][1]]]) dfs(i, res[s[i][1]]);
    for (const auto e:res) cout<<e;
    cout<<'\n';
}
#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...