Submission #774577

# Submission time Handle Problem Language Result Execution time Memory
774577 2023-07-05T20:11:29 Z NK_ Traffic (CEOI11_tra) C++17
100 / 100
500 ms 48984 KB
// 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 sz(x) int(x.size())
 
#define pb push_back
 
using pi = pair<int, int>;
template<class T> using V = vector<T>;
 
const int INF = 1e9 + 10;
 
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, INF), disc(N, 0), comp(N, -1), st, comps;
 
  	// FIND SCCs
	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);
 
 
 	// SCCs
	V<V<int>> C(N); 
	for(int i = 0; i < N; i++) {
		C[comp[i]].pb(i);
	}
  
  
 	// WEST JUNCTIONS
	V<int> found(N);
	function<void(int)> see = [&](int u) {
		found[u] = 1;
		for(auto v : adj[u]) if (!found[v]) see(v);
	};
	
	V<int> W;
	for(int i = 0; i < N; i++) {
		if (P[i].f == 0) {
			W.pb(i); see(i);
		}
	}
 
	sort(begin(W), end(W), [&](int x, int y) {
		return P[x].s > P[y].s;
	});
 
 
 	// EAST JUNCTIONS
	V<int> E;
	for(int i = 0; i < N; i++) if (found[i]) {
		if (P[i].f == A) E.pb(i);
	}
 
	sort(begin(E), end(E), [&](int x, int y) {
		return P[x].s > P[y].s;
	});
 
 
 	// FIND INTERVALS OF JUNCTIONS
	V<int> L(N, INF), R(N, -INF);
	for(int i = 0; i < sz(E); i++) {
		int e = E[i];
		L[comp[e]] = min(L[comp[e]], i);
		R[comp[e]] = max(R[comp[e]], i);
	}
 
	for(auto c : comps) {
		for(auto u : C[c]) for(auto v : adj[u]) {
			int cv = comp[v];
			L[c] = min(L[c], L[cv]);
			R[c] = max(R[c], R[cv]);
		}
	}
	
	for(auto u : W) cout << max(0, R[comp[u]] - L[comp[u]] + 1) << nl;
 
    return 0;
}
# Verdict Execution time Memory Grader output
1 Correct 0 ms 212 KB Output is correct
2 Correct 0 ms 212 KB Output is correct
3 Correct 0 ms 212 KB Output is correct
4 Correct 0 ms 212 KB Output is correct
5 Correct 0 ms 212 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 0 ms 212 KB Output is correct
2 Correct 0 ms 212 KB Output is correct
3 Correct 1 ms 212 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 0 ms 212 KB Output is correct
2 Correct 1 ms 340 KB Output is correct
3 Correct 1 ms 340 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 2 ms 724 KB Output is correct
2 Correct 3 ms 980 KB Output is correct
3 Correct 2 ms 724 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 11 ms 5136 KB Output is correct
2 Correct 30 ms 7776 KB Output is correct
3 Correct 16 ms 3696 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 24 ms 6444 KB Output is correct
2 Correct 42 ms 9368 KB Output is correct
3 Correct 24 ms 7124 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 51 ms 12976 KB Output is correct
2 Correct 71 ms 16588 KB Output is correct
3 Correct 105 ms 12308 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 75 ms 12580 KB Output is correct
2 Correct 76 ms 15020 KB Output is correct
3 Correct 89 ms 12740 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 125 ms 22732 KB Output is correct
2 Correct 151 ms 31036 KB Output is correct
3 Correct 213 ms 23328 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 260 ms 38128 KB Output is correct
2 Correct 288 ms 44172 KB Output is correct
3 Correct 295 ms 28064 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 413 ms 37368 KB Output is correct
2 Correct 319 ms 44732 KB Output is correct
3 Correct 427 ms 36524 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 139 ms 37084 KB Output is correct
2 Correct 374 ms 48984 KB Output is correct
3 Correct 401 ms 37120 KB Output is correct
4 Correct 500 ms 47976 KB Output is correct