Submission #774327

#	Time	Username	Problem	Language	Result	Execution time	Memory
774327		NK_	Traffic (CEOI11_tra)	C++17	8 / 100	398 ms	50984 KiB

tra

// Success consists of going from failure to failure without loss of enthusiasm
#include <bits/stdc++.h>

using namespace std;

#define nl '\n'
#define f first
#define s second
#define mp make_pair

#define pb push_back

using pi = pair<int, int>;
template<class T> using V = vector<T>;

int main() {
	cin.tie(0)->sync_with_stdio(0);
	
	int N, M, A, B; cin >> N >> M >> A >> B;

	V<pi> P(N); for(auto& x : P) cin >> x.f >> x.s;

	V<V<int>> adj(N);

	for(int i = 0; i < M; i++) {
		int u, v, d; cin >> u >> v >> d; --u, --v, --d;

		adj[u].pb(v);
		if (d) adj[v].pb(u);
	}

	V<int> low(N), disc(N), comp(N, -1), st, comps;

	int t = 0;
	function<int(int)> dfs = [&](int u) {
		st.pb(u);
		low[u] = disc[u] = ++t;

		for(auto& v : adj[u]) {
			if (comp[v] == -1) low[u] = min(low[u], disc[v] ?: dfs(v));
		}

		if (low[u] == disc[u]) {
			comps.pb(u);
			for(int v = -1; v != u; ) comp[v = st.back()] = u, st.pop_back();
		}	

		return low[u];
	};

	for(int i = 0; i < N; i++) if (!disc[i]) dfs(i);
	// for(int i = 0; i < N; i++) {
		// cout << i << " " << comp[i] << endl;
	// }
	// for(auto x : comps) cout << x << " ";
	// cout << endl;
	// reverse(begin(comps), end(comps));


	V<V<int>> C(N); V<int> E(N, 0), W; 
	for(int i = 0; i < N; i++) {
		C[comp[i]].pb(i);
		if (P[i].f == A) E[comp[i]]++;
		if (P[i].f == 0) W.pb(i);
	}

	sort(begin(W), end(W), [&](int x, int y) {
		return P[x].s > P[y].s;
	});

	V<int> PAR(N); iota(begin(PAR), end(PAR), 0);
	function<int(int)> get = [&](int u) { return PAR[u] == u ? u : PAR[u] = get(PAR[u]); };

	for(auto c : comps) {
		// cout << "C: " << c << " " << E[c] << endl;
		set<int> seen = {c};
		for(auto u : C[c]) for(auto v : adj[u]) {
			int cv = comp[v];
			if (seen.insert(cv).second) {
				// cout << u << " -> " << c << " " << comp[v] << endl;
				if (get(c) != get(cv)) {
					PAR[cv] = c;
					E[c] += E[cv]; 
				}
			}
		}
	}
	
	for(auto u : W) cout << E[comp[u]] << nl;

    return 0;
}

• Subtask 1: 8 / 8
• Subtask 2: 0 / 8
• Subtask 3: 0 / 8
• Subtask 4: 0 / 8
• Subtask 5: 0 / 8
• Subtask 6: 0 / 8
• Subtask 7: 0 / 8
• Subtask 8: 0 / 8
• Subtask 9: 0 / 9
• Subtask 10: 0 / 9
• Subtask 11: 0 / 9
• Subtask 12: 0 / 9