Submission #1343629

#TimeUsernameProblemLanguageResultExecution timeMemory
1343629jellybeanStranded Far From Home (BOI22_island)C++20
0 / 100
1101 ms192152 KiB
#include <bits/stdc++.h>
using namespace std;
#define int long long
#define pb push_back
#define dd(x) cout<<#x<<" is "<<x<<endl;
typedef pair<int,int> pii;

const int N = 2e5+5;
int a[N], ans[N];
int n;

void dfs(int x, vector<vector<int>> &adj, vector<vector<int>> &newadj, vector<int> &vis){
	for(auto i: adj[x]){
		if(vis[i]) continue;
		vis[i] = 1;
		newadj[x].pb(i);
		newadj[i].pb(x);
		dfs(i, adj, newadj, vis);
	}
}

void findsum(int x, int &sum, vector<vector<int>> &adj, vector<int> &vis){
	sum += a[x];
	for(auto i: adj[x]){
		if(vis[i]) continue;
		vis[i] = 1;
		findsum(i, sum, adj, vis);
	}
}

void recurse(vector<vector<int>> &adj, int x){
	int mxnode = -1, mx = 0;
	for(int i=1; i<=n; i++){
		if(adj[i].empty()) continue;
		if(a[i] > mx) mx = a[i], mxnode = i;
	}
	if(mxnode == -1) return;
	
	int sum = 0;
	vector<int> vis(n+1,0);
	vis[mxnode] = 1;
	findsum(mxnode, sum, adj, vis);
	if(sum < x) return;
	
	vector<vector<int>> adj1(n+1);
	for(int i=1; i<=n; i++){
		for(auto j: adj[i]){
			if(a[i] == mx or a[j] == mx){
				if(a[i] == mx) ans[i] = 1;
				if(a[j] == mx) ans[j] = 1;
			} else {
				adj1[i].pb(j);
			}
		}
	}
	
	/*dd(0);
	for(int i=1; i<=n; i++){
		for(auto j: adj1[i]) cout<<i<<' '<<j<<endl;
	}*/
	
	vis = vector<int>(n+1,0);
	for(int i=1; i<=n; i++){
		if(vis[i]) continue;
		vis[i] = 1;
		vector<vector<int>> newadj(n+1);
		dfs(i, adj1, newadj, vis);
		/*dd(mxnode)
		for(int i=1; i<=n; i++) cout<<vis[i]<<" ";cout<<endl;
		
		for(int i=1; i<=n; i++){
			dd(1)
			for(auto j: newadj[i]) cout<<i<<' '<<j<<endl;
		}*/
		recurse(newadj, mx);
		
	}
}

signed main(){
	ios::sync_with_stdio(0); cin.tie(0);
	
	int m; cin>>n>>m;
	for(int i=1; i<=n; i++) cin>>a[i];
	
	if(n==1){
		cout<<1;
		return 0;
	}
	
	vector<vector<int>> adj(n+1);
	for(int i=0; i<m; i++){
		int u,v; cin>>u>>v;
		adj[u].pb(v);
		adj[v].pb(u);
	}
	
	recurse(adj, 0);
	
	for(int i=1; i<=n; i++) cout<<ans[i];
	
	return 0;
}
#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...