summaryrefslogtreecommitdiff
path: root/lib/include/clusters.hpp
blob: 0728b3e065a1d9c1b35d695cc1c2eb9e59f17c61 (plain)
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

#include "graph.hpp"
#include <vector>

class ClusterTree {
private:
  std::vector<signed> p;
  std::vector<unsigned> o;

public:
  ClusterTree(unsigned n) : p(n, -1), o(n, 1) {}

  unsigned findroot(unsigned i) {
    if (p[i] < 0)
      return i;
    else
      return p[i] = this->findroot(p[i]);
  }

  void join(unsigned i, unsigned j) {
    if (o[i] < o[j]) {
      p[i] = j;
      o[j] += o[i];
    } else {
      p[j] = i;
      o[i] += o[j];
    }
  }

  void add_bond(const graph::edge& b) {
    unsigned i = this->findroot(b.v[0]);
    unsigned j = this->findroot(b.v[1]);

    if (i != j) {
      this->join(i, j);
    }
  }

  bool same_component(unsigned v0, unsigned v1) {
    unsigned i = this->findroot(v0);
    unsigned j = this->findroot(v1);

    return i == j;
  }
};