제출 #1059479

#제출 시각아이디문제언어결과실행 시간메모리
1059479Gromp15열쇠 (IOI21_keys)C++17
37 / 100
3086 ms25836 KiB
#include <bits/stdc++.h>
#define ar array
#define sz(x) (int)x.size()
#define all(x) x.begin(), x.end()
using namespace std;
template<typename T> bool ckmin(T &a, const T &b) { return a > b ? a = b, 1 : 0; }
template<typename T> bool ckmax(T &a, const T &b) { return a < b ? a = b, 1 : 0; }

std::vector<int> find_reachable(std::vector<int> r, std::vector<int> u, std::vector<int> v, std::vector<int> c) {
	int n = sz(r), m = sz(u);
	vector<vector<ar<int, 2>>> adj(n);
	for (int i = 0; i < m; i++) {
		adj[u[i]].push_back({v[i], c[i]});
		adj[v[i]].push_back({u[i], c[i]});
	}
	vector<int> p(n);
	int mn = 1e9;
	for (int i = 0; i < n; i++) {
		vector<bool> vis(n), got(n);
		auto dfs = [&](auto&& s, int v) -> void {
			got[r[v]] = 1, vis[v] = 1;
			for (auto [u, col] : adj[v]) if (!vis[u] && got[col]) {
				s(s, u);
			}
		};
		dfs(dfs, i);
		for (int j = 0; j < n; j++) p[i] += vis[j];
		ckmin(mn, p[i]);
	}
	vector<int> ans(n);
	for (int i = 0; i < n; i++) ans[i] = p[i] == mn;
	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...