-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathBOJ11812.cpp
83 lines (74 loc) · 2.25 KB
/
BOJ11812.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
#define _CRT_SECURE_NO_WARNINGS
#include <bits/stdc++.h>
using namespace std;
#define ll long long
int N, K, Q;
vector<ll> parent2(ll here, vector<ll>& plist) {
plist.push_back(here);
if (here == 1) {
return plist;
}
else {
return parent2(here / 2, plist);
}
}
vector<ll> parentK(ll here, vector<ll>& plist) {
plist.push_back(here);
if (here == 1) {
return plist;
}
if ((here - 1) % K == 0) {
return parentK((here - 1) / K, plist);
}
else if ((here) % K == 0) {
return parentK((here) / K, plist);
}
else{
return parentK((here / K) + 1, plist);
}
}
int main()
{
scanf("%d %d %d", &N, &K, &Q);
for (int i = 0; i < Q; ++i) {
ll a, b;
scanf("%lld %lld", &a, &b);
vector<ll> ap;
vector<ll> bp;
if (K == 1) {
if (a > b) {
printf("%lld\n", a - b);
}
else {
printf("%lld\n", b - a);
}
}
else if (K == 2) {
vector<ll>aparent = parent2(a, ap);
vector<ll>bparent = parent2(b, bp);
for (int j = 0; j < bparent.size(); ++j) {
vector<ll>::iterator iter1 = find(aparent.begin(), aparent.end(), bparent[j]);
if (iter1 != aparent.end()) {
ll lca = aparent[iter1 - aparent.begin()];
vector<ll>::iterator iter2 = find(bparent.begin(), bparent.end(), lca);
printf("%d\n", (iter1 - aparent.begin()) + (iter2 - bparent.begin()));
break;
}
}
}
else if (K>2){
vector<ll>aparent = parentK(a, ap);
vector<ll>bparent = parentK(b, bp);
for (int j = 0; j < bparent.size(); ++j) {
vector<ll>::iterator iter1 = find(aparent.begin(), aparent.end(), bparent[j]);
if (iter1 != aparent.end()) {
ll lca = *iter1;
vector<ll>::iterator iter2 = find(bparent.begin(), bparent.end(), lca);
printf("%d\n", (iter1 - aparent.begin()) + (iter2 - bparent.begin()));
break;
}
}
}
}
return 0;
}