Submission #446834

#TimeUsernameProblemLanguageResultExecution timeMemory
446834Nima_NaderiTraffic (CEOI11_tra)C++14
48 / 100
5099 ms86764 KiB
//In the name of God
#pragma GCC optimize("Ofast")
#include<bits/stdc++.h>
using namespace std;

typedef int ll;
const ll MXN = 3e5 + 10;
ll n, m, h, w, ts;
ll te[MXN], tw[MXN], dp[MXN], cmp[MXN], comp[MXN];
vector<ll> adj[MXN], adt[MXN], Top, G[MXN], Gp[MXN], C[MXN];
bool east[MXN], west[MXN], vis[MXN], is[MXN];
vector<pair<ll, ll>> ANS;
bool kr[MXN];
inline void add_edge(ll u, ll v){
	adj[u].push_back(v);
	adt[v].push_back(u);
}
void dfs(ll u){
	vis[u] = 1;
	for(auto v : adj[u]){
			if(!vis[v]) dfs(v);
	}
	Top.push_back(u);
}
void sfd(ll u, ll c){
	comp[u] = c, C[c].push_back(u);
	for(auto v : adt[u]){
		if(!comp[v]) sfd(v, c);
	}
}
vector<ll> V;
ll DFS(ll u){
	ll res = te[u];
	vis[u] = 1; V.push_back(u);
	for(auto v : G[u]){
		if(!vis[v]) res += DFS(v);
	}
	return res;
}
void SFD(ll u, ll x){
	vis[u] = 1; dp[u] += x;
	V.push_back(u);
	for(auto v : Gp[u]){
		if(!vis[v] && kr[v]) SFD(v, x);
	}
}
int main(){
	ios::sync_with_stdio(0);cin.tie(0); cout.tie(0);
	cin >> n >> m >> h >> w;
	for(int i = 1; i <= n; i ++){
		ll x, y; cin >> x >> y;
		east[i] = (x == h), west[i] = (x == 0);
		cmp[i] = -y;
	}
	for(int i = 1; i <= m; i ++){
		ll u, v, f; cin >> u >> v >> f;
		add_edge(u, v);
		if(f > 1) add_edge(v, u);
	}
	for(int i = 1; i <= n; i ++){
		if(!vis[i]) dfs(i);
	}
	reverse(Top.begin(), Top.end());
	for(int j = 1; j <= n; j ++){
		ll i = Top[j - 1];
		if(!comp[i]) sfd(i, ++ ts);
	}
	for(int i = 1; i <= n; i ++){
		te[comp[i]] += east[i];
		tw[comp[i]] += west[i];
	}
	for(int i = 1; i <= ts; i ++){
		for(auto u : C[i]){
			for(auto v : adj[u]){
				if(comp[v] != i && !is[comp[v]]){
					is[comp[v]] = 1;
					G[i].push_back(comp[v]);
					Gp[comp[v]].push_back(i);
				}
			}
		}
		for(auto X : G[i]) is[X] = 0;
	}
	for(int i = 1; i <= ts; i ++){
		kr[i] = (tw[i] > 0);
		if(kr[i]) continue;
		for(auto X : Gp[i]){
			if(kr[X]){
				kr[i] = 1;
				break;
			}
		}
	}
	/*
		cout << ts << '\n';
		cout << "comps : \n";
		for(int i = 1; i <= n; i ++){
			cout << comp[i] << ' ';
		} cout << '\n';
		cout << "edges of scc : \n";
		for(int i = 1; i <= ts; i ++){
			cout << "scc " << i << " : \n";
			cout << "te = " << te[i] << '\n';
			for(auto j : G[i]) cout << j << ' ';
			cout << '\n';
		}
	 */
	fill(vis, vis + ts + 1, 0);
	/*
	for(int i = 1; i <= ts; i ++){
		//dp[i] = te[i];
		//for(auto j : G[i]) dp[i] += dp[j];
			if(G[i].size() <= 1){
			dp[i] = te[i];
			if(G[i].size()) dp[i] += dp[G[i][0]];
			continue;
		}
		dp[i] = DFS(i);
		for(auto X : V) vis[X] = 0;
		V.clear();
	}
	*/
	for(int i = 1; i <= ts; i ++){
		if(te[i]){
			SFD(i, te[i]);
			for(auto X : V) vis[X] = 0;
			V.clear();
		}
	}
	for(int i = 1; i <= n; i ++){
		if(west[i]){
			ANS.push_back({cmp[i], dp[comp[i]]});
		}
	}
	sort(ANS.begin(), ANS.end());
	for(auto X : ANS) cout << X.second << '\n';
	return 0;
}
#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...
#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...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...