제출 #654210

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
654210		baojiaopisu	Stranded Far From Home (BOI22_island)	C++14	100 / 100	369 ms	36104 KiB

island

#include<bits/stdc++.h>

using namespace std;

using ll = long long;
using ld = long double;
using ull = unsigned long long;

using pii = pair<int, int>;
using pll = pair<ll, ll>;
using pld = pair<ld, ld>;

#define fi first
#define se second
#define pb push_back
#define pf push_front
#define mp make_pair
#define ins insert
#define btpc __builtin_popcount
#define btclz __builtin_clz

#define sz(x) (int)(x.size());
#define all(x) x.begin(), x.end()
#define debug(...) " [" << #__VA_ARGS__ ": " << (__VA_ARGS__) << "] "

mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());

int d4x[4] = {1, 0, -1, 0}; int d4y[4] = {0, 1, 0, -1};
int d8x[8] = {0, 1, 1, 1, 0, -1, -1, -1};
int d8y[8] = {1, 1, 0, -1, -1, -1, 0, 1};

template<class X, class Y>
    bool minimize(X &x, const Y &y) {
        if (x > y)
        {
            x = y;
            return true;
        }
        return false;
    }
template<class X, class Y>
    bool maximize(X &x, const Y &y) {
        if (x < y)
        {
            x = y;
            return true;
        }
        return false;
    }

const int MOD = 1e9 + 7; //998244353

template<class X, class Y>
	void add(X &x, const Y &y) {
		x = (x + y);
		if(x >= MOD) x -= MOD;
	}

template<class X, class Y> 
	void sub(X &x, const Y &y) {
		x = (x - y);
		if(x < 0) x += MOD;
	}

/* Author : Le Ngoc Bao Anh, 12A5, LQD High School for Gifted Student*/

const ll INF = 1e9;
const int N = 2e5 + 10;

int par[N], id[N], ans[N];
ll sum[N], a[N];
vector<int> g[N], son[N];

int find_par(int u) {
	if(par[u] < 0) return u;
	par[u] = find_par(par[u]);
	return par[u];
}

void merge(int u, int v) {
	u = find_par(u), v = find_par(v);
	if(u == v) return;
	if(par[u] > par[v]) swap(u, v);
	par[u] += par[v];
	sum[u] += sum[v];
	par[v] = u;
	for(auto x : son[v]) son[u].pb(x);
}

bool cmp(int x, int y) {
	return a[x] < a[y];
}

void solve() {
	int n, m; cin >> n >> m;
	for(int i = 1; i <= n; i++) cin >> a[i];

	for(int i = 1; i <= m; i++) {
		int u, v; cin >> u >> v;
		g[u].pb(v);
		g[v].pb(u);
	}

	for(int i = 1; i <= n; i++) id[i] = i, sum[i] = a[i], son[i].pb(i), par[i] = -1;
	sort(id + 1, id + 1 + n, cmp);

	for(int i = 1; i <= n; i++) {
		int u = id[i];
		for(auto v : g[u]) {
			if(a[v] < a[u]) {
				v = find_par(v);
				if(sum[v] < a[u]) {
					for(auto x : son[v]) ans[x] = 1;
					son[v].clear();
				}
			}

			if(a[v] <= a[u]) {
				merge(u, v);
			}
		}
	}

	for(int i = 1; i <= n; i++) cout << (ans[i] ^ 1);
}

int main()
{
    ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);

    int tc = 1, ddd = 0;
    // cin >> tc;
    while(tc--) {
        //ddd++;
        //cout << "Case #" << ddd << ": ";
        solve();
    }
}

컴파일 시 표준 에러 (stderr) 메시지

island.cpp: In function 'int main()':
island.cpp:131:17: warning: unused variable 'ddd' [-Wunused-variable]
  131 |     int tc = 1, ddd = 0;
      |                 ^~~

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