Submission #244476

# Submission time Handle Problem Language Result Execution time Memory
244476 2020-07-04T06:44:19 Z oolimry Pipes (CEOI15_pipes) C++14
70 / 100
1309 ms 11516 KB
//HAPPY BIRTHDAY!!!
#include <bits/stdc++.h>
#define push_back emplace_back
using namespace std;
typedef pair<int,int> ii;

struct UFDS{
	int n;
	int p[100005];
	
	void init(int _n){
		n = _n;
		for(int i = 1;i <= n;i++) p[i] = i;
	}
	
	int find(int u){
		if(p[u] == u) return u;
		else{
			p[u] = find(p[u]);
			return p[u];
		}
	}
	
	void unionSet(int u, int P){
		u = find(u); P = find(P);
		if(u == P) return;
		p[u] = P;
	}
} tree, bridge;

vector<int> nodes[100005];
int p[100005];
int depth[100005];
vector<ii> possible;

void reroot(int u){
	vector<int> order;
	while(u != 0){
		order.push_back(u);
		u = p[u];
	}
	
	int sz = order.size(); order.push_back(0);
	for(int i = 0;i < sz;i++){
		p[order[i+1]] = order[i];
		depth[order[i]] = sz-i-1;
	}
	order.clear();
}

signed main(){
	ios_base::sync_with_stdio(false); cin.tie(0);
	
	int n, E; cin >> n >> E;
	
	tree.init(n+2); bridge.init(n+2);
	
	for(int i = 1;i <= n;i++) nodes[i].emplace_back(i);
	
	while(E--){
		int u, v; cin >> u >> v;
		if(bridge.find(u) == bridge.find(v)) continue;
		
		if(tree.find(u) == tree.find(v)){ ///bridge formed
			
			vector<int> path;
			
			while(u != v){
				if(depth[u] > depth[v]) swap(u,v); ///v is deeper now
				
				path.push_back(v);
				v = p[v];
			}
			
			int P = u;
			//cout << P << "\n";
			for(int x : path){
				p[x] = P;
				p[x] = P;
				depth[x] = depth[P]+1;
				bridge.unionSet(x, P);
			}
		}
		else{ ///connect two trees to 1 tree
			int U = tree.find(u), V = tree.find(v);
			
			if(nodes[U].size() > nodes[V].size()){
				swap(U,V);
				swap(u,v);
			}
			
			//tree.unionSet(U, V);
			tree.p[U] = V;
			reroot(u);
			p[u] = v;
			while(!nodes[U].empty()){
				int B = nodes[U].back();
				depth[B] += depth[v] + 1;
				nodes[V].emplace_back(B);
				nodes[U].pop_back();
			}
			
			//cout << u << " " << v << ":\n";
			//for(int i = 1;i <= n;i++) cout << p[i] << " ";
			//cout << "\n";
			//for(int i = 1;i <= n;i++) cout << depth[i] << " ";
			//cout << "\n";
			possible.push_back(ii(u,v));
		}
	}
	
	for(ii e : possible){
		
		if(bridge.find(e.first) != bridge.find(e.second)){
			cout << e.first << " " << e.second << "\n";;
		}
		
		
	}
}
# Verdict Execution time Memory Grader output
1 Correct 6 ms 2688 KB Output is correct
2 Correct 6 ms 2688 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 10 ms 3072 KB Output is correct
2 Incorrect 10 ms 3200 KB Wrong number of edges
# Verdict Execution time Memory Grader output
1 Correct 103 ms 2944 KB Output is correct
2 Incorrect 99 ms 3112 KB Wrong number of edges
# Verdict Execution time Memory Grader output
1 Correct 177 ms 3456 KB Output is correct
2 Correct 209 ms 5368 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 301 ms 6132 KB Output is correct
2 Correct 245 ms 6036 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 438 ms 8680 KB Output is correct
2 Incorrect 381 ms 8812 KB Wrong number of edges
# Verdict Execution time Memory Grader output
1 Correct 682 ms 9104 KB Output is correct
2 Correct 673 ms 9268 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 867 ms 11516 KB Output is correct
2 Correct 812 ms 11240 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 1080 ms 10092 KB Output is correct
2 Correct 1014 ms 11416 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 1309 ms 10116 KB Output is correct
2 Correct 1302 ms 11064 KB Output is correct