Submission #1871897


Source Code Expand

#define _USE_MATH_DEFINES
#include <iostream>
#include <fstream>
#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <cstring>
#include <string>
#include <vector>
#include <utility>
#include <complex>
#include <set>
#include <map>
#include <queue>
#include <stack>
#include <deque>
#include <tuple>
#include <bitset>
#include <algorithm>
#include <random>
using namespace std;
typedef long double ld;
typedef long long ll;
typedef vector<int> vint;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
typedef pair<double, double> pdd;
typedef complex<ld> compd;
#define reach(i,a)	for(auto i:a)
#define rep(i,n)	for(int i=0;i<n;i++)
#define srep(i,a,n)	for(int i=a;i<n;i++)
#define REP(i,n)	for(int i=0;i<=n;i++)
#define SREP(i,a,n)	for(int i=a;i<=n;i++)
#define rrep(i,n)	for(int i=n-1;i>=0;i--)
#define RREP(i,n)	for(int i=n;i>=0;i--)
#define all(a)	(a).begin(),(a).end()
#define mp(a,b)	make_pair(a,b)
#define mt	make_tuple
#define pb	push_back
#define fst	first
#define scn second
int bitcnt(ll x) {
	x = ((x & 0xAAAAAAAAAAAAAAAA) >> 1) + (x & 0x5555555555555555);
	x = ((x & 0xCCCCCCCCCCCCCCCC) >> 2) + (x & 0x3333333333333333);
	x = ((x & 0xF0F0F0F0F0F0F0F0) >> 4) + (x & 0x0F0F0F0F0F0F0F0F);
	x = ((x & 0xFF00FF00FF00FF00) >> 8) + (x & 0x00FF00FF00FF00FF);
	x = ((x & 0xFFFF0000FFFF0000) >> 16) + (x & 0x0000FFFF0000FFFF);
	x = ((x & 0xFFFFFFFF00000000) >> 32) + (x & 0x00000000FFFFFFFF);
	return x;
}
int bitcnt(int x) {
	x = ((x & 0xAAAAAAAA) >> 1) + (x & 0x55555555);
	x = ((x & 0xCCCCCCCC) >> 2) + (x & 0x33333333);
	x = ((x & 0xF0F0F0F0) >> 4) + (x & 0x0F0F0F0F);
	x = ((x & 0xFF00FF00) >> 8) + (x & 0x00FF00FF);
	x = ((x & 0xFFFF0000) >> 16) + (x & 0x0000FFFF);
	return x;
}
ll gcd(ll a, ll b) {
	return a%b == 0 ? b : gcd(b, a%b);
}
#define debug(x)	cout<<"case #" << x << ": " << endl
#define DEBUG 0
const ll inf = (ll)1e9;
const ll mod = 1e9 + 7;
const ld eps = 1e-9;
const int dx[] = { 0,1,0,-1 };
const int dy[] = { 1,0,-1,0 };

vector<pii> edge[100000];
// LCA用。parent[i][j]:頂点iの2^j個上の点
int parent[100000][20];
// 木の深さ
int depth[100000];
// 同時に用いる辺の最大の重み
int maxm[100000][20];

// Union Find
int group[100000];
int findgroup(int i) {
	return group[i] == i ? i : group[i] = findgroup(group[i]);
}
void merge(int a, int b) {
	a = findgroup(a);
	b = findgroup(b);
	group[a] = b;
}

// LCAの最初の段階。木に対して、1つ上の親を決める
void dfs(int cur, int p, int e) {
	depth[cur] = depth[p] + 1;
	parent[cur][0] = p;
	maxm[cur][0] = e;
	reach(v, edge[cur]) {
		if (v.first != p)	dfs(v.first, cur, v.second);
	}
}

int main() {
	int n, m;	cin >> n >> m;
	rep(i, n)	group[i] = i;
	// 辺を追加して閉路が出来る(=グループが既に同じ)なら無視してよい
	rep(i, m) {
		int a, b;	cin >> a >> b;	a--;	b--;
		if (findgroup(a) != findgroup(b)) {
			merge(a, b);
			edge[a].push_back(mp(b, i + 1));
			edge[b].push_back(mp(a, i + 1));
		}
	}
	// dfsを行い、木の順序を確定とともに、parent[i][0]=1つ上の親を確定
	rep(i, n)	rep(j, 20) {
		parent[i][j] = i;	maxm[i][j] = -1;
	}
	rep(i, n) {
		if (parent[i][0] == i) {
			dfs(i, i, m + 1);
		}
	}
	// parentの確定を行う
	srep(t, 1, 20) {
		rep(i, n) {
			parent[i][t] = parent[parent[i][t - 1]][t - 1];
			maxm[i][t] = max(maxm[i][t - 1], maxm[parent[i][t - 1]][t - 1]);
		}
	}
	int q;	cin >> q;
	rep(t, q) {
		int x, y;	cin >> x >> y;	x--;	y--;
		// 木が違います
		if (findgroup(x) != findgroup(y)) {
			cout << -1 << endl;
		}
		else {
			// LCAしながらretを下げていく
			int ret = -1;
			// 高さ合わせ
			while (depth[x] > depth[y]) {
				int j = 0;
				while (depth[parent[x][j + 1]] > depth[y])	j++;
				ret = max(ret, maxm[x][j]);
				x = parent[x][j];
			}
			while (depth[y] > depth[x]) {
				int j = 0;
				while (depth[parent[y][j + 1]] > depth[x]) j++;
				ret = max(ret, maxm[y][j]);
				y = parent[y][j];
			}
			// 同時に上っていきます
			while (x != y) {
				int j = 0;
				while (parent[x][j + 1] != parent[y][j + 1])	j++;
				ret = max(ret, max(maxm[x][j], maxm[y][j]));
				x = parent[x][j];
				y = parent[y][j];
			}
			cout << ret << endl;
		}
	}
}

Submission Info

Submission Time
Task H - Union Sets
User fiord
Language C++14 (GCC 5.4.1)
Score 600
Code Size 4355 Byte
Status AC
Exec Time 401 ms
Memory 27392 KB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 600 / 600
Status
AC × 3
AC × 49
Set Name Test Cases
Sample sample_01.txt, sample_02.txt, sample_03.txt
All sample_01.txt, sample_02.txt, sample_03.txt, subtask_1_1.txt, subtask_1_10.txt, subtask_1_11.txt, subtask_1_12.txt, subtask_1_13.txt, subtask_1_14.txt, subtask_1_15.txt, subtask_1_16.txt, subtask_1_17.txt, subtask_1_18.txt, subtask_1_19.txt, subtask_1_2.txt, subtask_1_20.txt, subtask_1_21.txt, subtask_1_22.txt, subtask_1_23.txt, subtask_1_24.txt, subtask_1_25.txt, subtask_1_26.txt, subtask_1_27.txt, subtask_1_28.txt, subtask_1_29.txt, subtask_1_3.txt, subtask_1_30.txt, subtask_1_31.txt, subtask_1_32.txt, subtask_1_33.txt, subtask_1_34.txt, subtask_1_35.txt, subtask_1_36.txt, subtask_1_37.txt, subtask_1_38.txt, subtask_1_39.txt, subtask_1_4.txt, subtask_1_40.txt, subtask_1_41.txt, subtask_1_42.txt, subtask_1_43.txt, subtask_1_44.txt, subtask_1_45.txt, subtask_1_46.txt, subtask_1_5.txt, subtask_1_6.txt, subtask_1_7.txt, subtask_1_8.txt, subtask_1_9.txt
Case Name Status Exec Time Memory
sample_01.txt AC 2 ms 4736 KB
sample_02.txt AC 2 ms 4608 KB
sample_03.txt AC 2 ms 4736 KB
subtask_1_1.txt AC 2 ms 4608 KB
subtask_1_10.txt AC 29 ms 4736 KB
subtask_1_11.txt AC 189 ms 5248 KB
subtask_1_12.txt AC 14 ms 14720 KB
subtask_1_13.txt AC 5 ms 4736 KB
subtask_1_14.txt AC 3 ms 4608 KB
subtask_1_15.txt AC 65 ms 4864 KB
subtask_1_16.txt AC 7 ms 4736 KB
subtask_1_17.txt AC 19 ms 4736 KB
subtask_1_18.txt AC 3 ms 4736 KB
subtask_1_19.txt AC 18 ms 11008 KB
subtask_1_2.txt AC 2 ms 4608 KB
subtask_1_20.txt AC 3 ms 4736 KB
subtask_1_21.txt AC 12 ms 4736 KB
subtask_1_22.txt AC 8 ms 10496 KB
subtask_1_23.txt AC 257 ms 19328 KB
subtask_1_24.txt AC 253 ms 19328 KB
subtask_1_25.txt AC 256 ms 19456 KB
subtask_1_26.txt AC 266 ms 19840 KB
subtask_1_27.txt AC 400 ms 23296 KB
subtask_1_28.txt AC 392 ms 23296 KB
subtask_1_29.txt AC 250 ms 19328 KB
subtask_1_3.txt AC 2 ms 4736 KB
subtask_1_30.txt AC 253 ms 19328 KB
subtask_1_31.txt AC 258 ms 19328 KB
subtask_1_32.txt AC 261 ms 19840 KB
subtask_1_33.txt AC 395 ms 27392 KB
subtask_1_34.txt AC 401 ms 27392 KB
subtask_1_35.txt AC 252 ms 19328 KB
subtask_1_36.txt AC 251 ms 19328 KB
subtask_1_37.txt AC 255 ms 19328 KB
subtask_1_38.txt AC 263 ms 19840 KB
subtask_1_39.txt AC 328 ms 24308 KB
subtask_1_4.txt AC 26 ms 15360 KB
subtask_1_40.txt AC 336 ms 24820 KB
subtask_1_41.txt AC 197 ms 4864 KB
subtask_1_42.txt AC 210 ms 4992 KB
subtask_1_43.txt AC 323 ms 18560 KB
subtask_1_44.txt AC 378 ms 24448 KB
subtask_1_45.txt AC 195 ms 4992 KB
subtask_1_46.txt AC 253 ms 19328 KB
subtask_1_5.txt AC 27 ms 4736 KB
subtask_1_6.txt AC 15 ms 14720 KB
subtask_1_7.txt AC 3 ms 5248 KB
subtask_1_8.txt AC 5 ms 5760 KB
subtask_1_9.txt AC 43 ms 5504 KB