Submission #395063

#	Time	Username	Problem	Language	Result	Execution time	Memory
395063		wind_reaper	Traffic (CEOI11_tra)	C++17	24 / 100	657 ms	39880 KiB

tra

#include <bits/stdc++.h>

using namespace std;

const int INF = 2'000'000'000;
const int MXN = 300'000;

vector<int> g[MXN];
int X[MXN], Y[MXN], ord[MXN], D[2][MXN];
vector<int> pref, west, east;
vector<bool> vis(MXN);

pair<int, int> dfs(int node){
	if(vis[node])
		return {D[0][node], D[1][node]};
	
	vis[node] = 1;

	if(ord[node] != -1){
		D[0][node] = min(D[0][node], ord[node]);
		D[1][node] = max(D[1][node], ord[node]);
	}

	for(int v : g[node]){
		pair<int, int> R = dfs(v);
		D[0][node] = min(D[0][node], R.first);
		D[1][node] = max(D[1][node], R.second);
	}

	
	return {D[0][node], D[1][node]};
}

int32_t main(){
	ios_base::sync_with_stdio(false); 
	cin.tie(NULL); 
	
	int N, M, A, B;
	cin >> N >> M >> A >> B;

	memset(ord, -1, sizeof ord);
	memset(D[1], -1, sizeof D[1]);

	for(int i = 0; i < N; i++)
		D[0][i] = INF;

	for(int i = 0; i < N; i++){
		cin >> X[i] >> Y[i];
		if(X[i] == 0) west.push_back(i);
		else if (X[i] == A) east.push_back(i);
	}

	for(int i = 0; i < M; i++){
		int u, v, k;
		cin >> u >> v >> k;
		--u, --v;
		g[u].push_back(v);
		if(k == 2) g[v].push_back(u);
	}

	sort(west.begin(), west.end(), [&](int i, int j){
		return Y[i] > Y[j];
	});

	sort(east.begin(), east.end(), [&](int i, int j){
		return Y[i] > Y[j];
	});

	for(int i = 0; i < east.size(); i++)
		ord[east[i]] = i;

	for(int v : west){
		dfs(v);	
	}

	// for(int i = 0; i < N; i++){
	// 	for(int v : g[i]){
	// 		D[0][i] = min(D[0][i], D[0][v]);
	// 		D[1][i] = max(D[1][i], D[1][v]);
	// 	}
	// }

	fill(vis.begin(), vis.end(), 0);

	for(int v : west)
		dfs(v);

	int x = east.size();
	
	pref.resize(x+1);
	for(int i = 1; i <= x; i++)
		pref[i] = pref[i-1] + vis[east[i-1]];

	for(int v : west){
		if(D[1][v] == -1){
			cout << 0 << '\n';
			continue;
		}

		cout << pref[D[1][v]+1] - pref[D[0][v]] << '\n';
	}

	return 0; 
}

Compilation message (stderr)

tra.cpp: In function 'int32_t main()':
tra.cpp:69:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   69 |  for(int i = 0; i < east.size(); i++)
      |                 ~~^~~~~~~~~~~~~

• Subtask 1: 8 / 8
• Subtask 2: 8 / 8
• Subtask 3: 0 / 8
• Subtask 4: 8 / 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