Submission #1320471

#TimeUsernameProblemLanguageResultExecution timeMemory
1320471888313666Stranded Far From Home (BOI22_island)C++20
10 / 100
115 ms33500 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<ll> s; vector<int> res; vector<array<int, 2>> vis; vector<array<ll, 2>> srt; struct dsu{ int n; vector<int> par; vector<ll> sz; dsu(const int n) { this->n=n; par.resize(n); sz=s; iota(par.begin(), par.end(), 0); } int find(const int i) { if (par[i]==i) return i; return par[i]=find(par[i]); } ll size(const int i) { return sz[find(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; } }; bool bfs(const int st) { priority_queue<pair<ll, int>, vector<pair<ll, int>>, greater<>> q; vector<int> vis(n, 0); ll cur=s[st]; for (const auto v:adj[st]) q.push({s[v], v}); vis[st]=1; while (!q.empty()) { const auto [d, u]=q.top(); q.pop(); if (vis[u]) continue; vis[u]=1; if (d>cur) return false; cur+=d; for (const auto v:adj[u]) if (!vis[v]) q.push({s[v], v}); } return true; } void dfs(const int u, int f) { vis[u][f]=1; f&=res[u]; res[u]=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); srt.resize(n); for (auto &e:s) cin>>e; for (int i=0; i<m; i++) { int a,b; cin>>a>>b; if (s[--a]>=s[--b]) adj[a].push_back(b); if (s[a]<=s[b]) adj[b].push_back(a); } for (int i=0; i<n; i++) { srt[i][0]=s[i]; srt[i][1]=i; } sort(srt.begin(), srt.end()); int l=0, r=0, w=0; dsu f(n); priority_queue<pair<ll, int>, vector<pair<ll, int>>, greater<>> q; while (l!=n) { w=srt[l][0]; while (r<n-1 && srt[r+1][0]==w) ++r; ++r; while (!q.empty() && q.top().first<w) { const auto [d, i]=q.top(); //cout<<d _ i _ f.size(srt[i][1])<<'\n'; q.pop(); if (f.size(srt[i][1])<w) res[srt[i][1]]=0; } //cout<<l _ r<<'\n'; for (int i=l; i<r; ++i) { for (const auto v:adj[srt[i][1]]) f.unite(srt[i][1], v); } for (int i=l; i<r; i++) { //cout<<'a' _ f.size(srt[i][1]) _ i<<'\n'; q.push({f.size(srt[i][1]), i}); } l=r; } vis.assign(n, {0,0}); for (int i=n-1; i>=0; --i) { if (!vis[srt[i][1]][res[srt[i][1]]]) dfs(srt[i][1], res[srt[i][1]]); } //dfs(srt.back()[1], 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...