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 |
|
|
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 |