Submission #912273

#	Time	Username	Problem	Language	Result	Execution time	Memory
912273		pragmatist	Stranded Far From Home (BOI22_island)	C++17	0 / 100	1091 ms	12764 KiB

island

/*
JUDGE_ID : 331438IX
*/
 
#include<iostream>
#include<cassert>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<vector>
#include<stack>
#include<set>
#include<queue>
#include<map>
 
using namespace std;
 
const int N = (int)2e5 + 7;
const int inf = (int)1e9 + 7;
const int MOD = (int)998244353;
const long long INF = (long long)1e18 + 7; 
 
int mult(int x, int y) {
	return 1ll*x*y%MOD;
}
 
int sum(int x, int y) {
	x += y;
	if(x >= MOD) {
		x -= MOD;
	}
	return x;
}
 
int sub(int x, int y) {
	x -= y;
	if(x < 0) {
		x += MOD;
	}
	return x;
}
 
bool is_prime(int x) {
	if(x == 1) {
		return 0;
	}
	for(int i = 2; i * i <= x; ++i) {
		if(x % i == 0) {
			return 0;
		}
	}
	return 1;
}
 
int n, m, a[N], p[N];
bool ans[N];
vector<int> pos[N];

int get(int v) {
	return (p[v] == v ? v : p[v] = get(p[v]));
}
long long s[N];

void comb(int u, int v) {
	u = get(u);
	v = get(v);
	if(u == v) {
		return;
	}
	p[u] = v;
	s[v] += s[u];
}

long long val[N];

int b[N]; 

bool cmp2(int i, int j) {
	return a[i] < a[j];
}

int main() {	
	ios_base::sync_with_stdio(NULL);
	cin.tie(0);         
	cin >> n >> m;
	int mx = 0;
	for(int i = 1; i <= n; ++i) {
		cin >> a[i];
		mx = max(mx, a[i]);
		p[i] = i;
		b[i] = i;
		s[i] = a[i];
	}
	vector<pair<int, int> > e;
	for(int i = 1; i <= m; ++i) {
		int u, v;
		cin >> u >> v;
		e.push_back({u, v});
	}    
	sort(b+1, b+1+n, cmp2);
	for(int i = 1; i <= n; ++i) {
		if(a[b[i]] == mx) {
			break;
		}
		int r = i;
		while(r<n && a[b[r]] == a[b[r+1]]) {
			r++;
		}             
		for(auto [u, v] : e) {
			if(a[u] == a[v] && a[u] == a[b[i]]) {
				comb(u, v);
			}
		}
		for(auto [u, v] : e) {
			if(a[u] == a[b[i]]) {
				long long cur = s[get(u)];
				if(cur >= a[v]) {
					comb(u, v);
				}
			}
			if(a[v] == a[b[i]]) {
				long long cur = s[get(v)];
				if(cur >= a[u]) {
					comb(u, v);
				}
			}
		}
		i = r;
	}
	for(auto [u, v] : e) {
		if(a[u] == mx && a[v] == mx) {
			comb(u, v);
		}
	}
	for(int i = 1; i <= n; ++i) {
		cout << (get(i) == get(b[n]));
	}
	return 0;
}

• Subtask 1: 0 / 10
• Subtask 2: 0 / 10
• Subtask 3: 0 / 15
• Subtask 4: 0 / 30
• Subtask 5: 0 / 35