#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);
else 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});
dfs(srt.back()[1], 1);
for (const auto e:res) cout<<e;
cout<<'\n';
}
| # | Verdict | Execution time | Memory | Grader output |
|---|
| Fetching results... |
| # | Verdict | Execution time | Memory | Grader output |
|---|
| Fetching results... |
| # | Verdict | Execution time | Memory | Grader output |
|---|
| Fetching results... |
| # | Verdict | Execution time | Memory | Grader output |
|---|
| Fetching results... |
| # | Verdict | Execution time | Memory | Grader output |
|---|
| Fetching results... |