제출 #1341598

#제출 시각아이디문제언어결과실행 시간메모리
1341598limits열쇠 (IOI21_keys)C++20
0 / 100
0 ms344 KiB
#pragma GCC optimize("Ofast")
#pragma GCC optimize("O3,unroll-loops")

#include <bits/stdc++.h>

using namespace std;

#define f0r(i, n) for (auto i = 0; i < (n); ++i)
#define fnr(i, n, k) for (auto i = (n); i < (k); ++i)
#define all(v) (v).begin(), (v).end()
#define pb push_back
#define F first
#define S second
#define ctn(x) cout << x << '\n'
#define forl(a, l) for (auto a : l)
#define ctl(l) for (auto &a : (l)) cout << a << ' '; cout << endl;
#define lb(v, x) (lower_bound(all(v), x) - begin(v))
#define ub(v, x) (upper_bound(all(v), x) - begin(v))
#define pq priority_queue

template <class T>
using V = vector<T>;
using ll = long long;
using vi = V<int>;
using vl = V<ll>;
using pi = pair<int, int>;
using ti = tuple<int, int, int>;
using Adj = V<V<pi>>;
using vvi = V<vi>;

Adj G;

void dfs(int v, V<bool> &vis, vi &out) {
	vis[v] = 1;
	out.pb(v);
	for (auto &[c, w]: G[v]) if (!vis[c]) dfs(c, vis, out);
}

vi find_reachable(vi r, vi u, vi v, vi c) {
	int n = r.size(), m = c.size();
	vi ans(n);

	G = Adj(n);

	f0r(i, m) {
		G[u[i]].pb({v[i], c[i]});
		G[v[i]].pb({u[i], c[i]});
	}

	V<bool> vis(n);
	vi sp, sz(n);
	f0r(i, n) if (!vis[i]) {
		vi out;
		dfs(i, vis, out);
		if (out.size() == 1 || r[i]) sp.pb(i);
		sz[i] = out.size();
	}

	if (sp.size()) {
		forl(x, sp) ans[x] = 1;
	} else {
		int m = *min_element(all(sz));
		f0r(i, n) if (sz[i] == m) ans[i] = 1;
	}
	return ans;
}
#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...