oj.uz

Submission #150433

# Submission time Handle Problem Language Result Execution time Memory
150433 2019-09-01T08:23:58 Z ummm(#3574, cerberus, aayush9, knandy) Trip to the Galapagos Islands (FXCUP4_island) C++17
0 / 100
 357 ms 372068 KB 
#include "island.h"

#include <bits/stdc++.h>
using namespace std;

const int GAP = 666;

struct UnionFind {
	int n;
	vector<int> uf,si,c;
	UnionFind(int n):n(n),uf(n),si(n,1){
		for(int i=0;i<n;++i) uf[i]=i;
	}
	int find(int x){return x==uf[x]?x:find(uf[x]);}
	bool join(int x, int y, bool keep_track = 1){
		if((x=find(x))==(y=find(y))) return false;
		if(si[x]<si[y]) swap(x,y);
		si[x]+=si[y];uf[y]=x;
		// cout << "Joining " << x << ' ' << y << '\n';
		if(keep_track) c.push_back(y);
		return true;
	}
	void rollback(){
		while(c.size()){
			int x=c.back(); c.pop_back();
			si[uf[x]]-=si[x];uf[x]=x;;
		}
	}
};

vector<UnionFind> UF;
// struct UnionFindPathCompression {
// 	int n;
// 	vector<int> uf;
// 	UnionFindPathCompression(int n):n(n),uf(n){
// 		for(int i=0;i<n;++i) uf[i]=i;
// 	}
// 	int find(int x){return uf[x] = x==uf[x]?x:find(uf[x]);}
// 	bool join(int x, int y){
// 		if((x=find(x))==(y=find(y))) return false;
// 		if(rand()%2) uf[y]=x;
// 		return true;
// 	}
// };
// vector<UnionFindPathCompression> UFPC;

vector<pair<int,int>> edges(300005);
vector<int> pos[100005];

int M, N;

map<int,int> memo[100005];

void Init(int K, vector<int> F, vector<int> S, vector<int> E){
	N = F.size();
	M = S.size();

	edges.resize(M);

	for(int i=0;i<M;++i) edges[M-i-1] = {S[i], E[i]};

	for(int i=0;i<(int)F.size();++i)
		pos[F[i]].push_back(i);

	UnionFind cur(N);

	for(int i=0;i<M;++i){
		if(i%GAP==0) 
			UF.push_back(cur);
		cur.join(edges[i].first, edges[i].second, 0);
	}
	// cout << '\n';
}

int Separate(int A, int B){
	if(A>B) swap(A,B);
	if(memo[A].count(B)) return memo[A][B];

	int bucket = 0, start = 0, ans = -1;

	int lo = 0, hi = UF.size()-1;
	while(lo < hi){
		int mid = (lo + hi + 1)/2;
		int b = mid;
		for(auto i:pos[A]) UF[b].join(pos[A][0], i, 1);
		for(auto i:pos[B]) UF[b].join(pos[B][0], i, 1);

		if(UF[b].find(pos[A][0]) == UF[b].find(pos[B][0])){
			lo = mid;
			bucket = lo;
		} else {
			hi = mid - 1;
		}
		UF[b].rollback();
	}

	start = bucket * GAP;

	for(auto i:pos[A]) UF[bucket].join(pos[A][0], i, 1);
	for(auto i:pos[B]) UF[bucket].join(pos[B][0], i, 1);

	for(int i=start;i<start+GAP && i<M; ++i){
		UF[bucket].join(edges[i].first, edges[i].second, 1);
		if(UF[bucket].find(pos[A][0]) == UF[bucket].find(pos[B][0])){
			ans = M - i;
			break;
		}
	}
	assert(ans != -1);
	UF[bucket].rollback();
	return memo[A][B] = ans;
}

Subtask #1
0 / 31.0

# Verdict Execution time Memory Grader output
1 Runtime error 331 ms 372068 KB Execution killed with signal 11 (could be triggered by violating memory limits)
2 Halted 0 ms 0 KB -

Subtask #2
0 / 39.0

# Verdict Execution time Memory Grader output
1 Correct 12 ms 9728 KB Output is correct
2 Correct 11 ms 9728 KB Output is correct
3 Incorrect 357 ms 369508 KB Output isn't correct
4 Halted 0 ms 0 KB -

Subtask #3
0 / 30.0

# Verdict Execution time Memory Grader output
1 Correct 12 ms 9728 KB Output is correct
2 Correct 11 ms 9728 KB Output is correct
3 Runtime error 331 ms 372068 KB Execution killed with signal 11 (could be triggered by violating memory limits)
4 Halted 0 ms 0 KB -