minimario

<LCA> Template

Dec 19th, 2015
114
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.92 KB | None | 0 0
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3.  
  4. #define ll long long
  5. #define mp make_pair
  6. #define pb push_back
  7. #define pii pair <int, int>
  8. #define piii pair<pii, int>
  9. #define vi vector<int>
  10. #define vpii vector<pii>
  11.  
  12. #define read1(a) int a; scanf("%d", &a)
  13. #define read2(a, b) int a, b; scanf("%d %d", &a, &b)
  14. #define read3(a, b, c) int a, b, c; scanf("%d %d %d", &a, &b, &c)
  15.  
  16. #define FOR(i, a, b) for (int i=a; i<b; i++)
  17. #define F0R(i, a) for (int i=0; i<a; i++)
  18.  
  19. #define readgi(n) F0R(i, n) { scanf("%d", &arr[i]); }
  20. #define readgs(n) F0R(i, n) { scanf(" %c", &arr[i]); }
  21.  
  22. #define f first
  23. #define s second
  24.  
  25. #define usaco(in, out) freopen(in, "r", stdin); freopen(out, "w", stdout);
  26.  
  27. #define println1(a) printf("%d\n", a);
  28. #define println2(a, b) printf("%d %d\n", a, b);
  29. #define println3(a, b, c) printf("%d %d %d\n", a, b, c);
  30. #define pv(v) for (int i : v) { printf("%d ", i); } printf("\n");
  31.  
  32. const int MOD = 1000000007;
  33. const int MAX = 100005;
  34. const int N = 18;
  35. int arr[MAX];
  36.  
  37. vector<int> g[MAX];
  38. int depth[MAX];
  39. int par[MAX][N];
  40.  
  41. void dfs(int st, int p) {
  42.     par[st][0] = p;
  43.     for (int i : g[st]) {
  44.         if (i == p) { continue; }
  45.         depth[i] = depth[st] + 1;
  46.         dfs(i, st);
  47.     }
  48. }
  49.  
  50. void init() {
  51.     FOR(j, 1, N) { F0R(i, MAX) {
  52.         if (1<<(j) > depth[i]-1) { continue; }
  53.         par[i][j] = par[par[i][j-1]][j-1];
  54.     }}
  55. }
  56.  
  57. int anc(int x, int y) {
  58.     /* yth ancestor of x */
  59.     for (int i=N-1; i>=0; i--) { if ((1<<i)<=y) { x = par[x][i]; y -= (1<<i); } }
  60.     return x;
  61. }
  62.  
  63. int lca(int x, int y) {
  64.     if (depth[x] < depth[y]) { swap(x, y); }
  65.     for (int i = N-1; i >= 0; i--) {
  66.         if (depth[par[x][i]] >= depth[y]) { x = par[x][i]; }
  67.     }
  68.     if (x == y) { return x; }
  69.     for (int i=N-1; i>=0; i--) {
  70.         if (par[x][i] != par[y][i]) { x = par[x][i]; y = par[y][i]; }
  71.     }
  72.     return par[x][0];
  73. }
  74.  
  75. int main() {
  76.     depth[1] = 1;
  77.     dfs(1, -1); init();
  78. }
Advertisement
Add Comment
Please, Sign In to add comment