🧑‍💻 improves tree diamater implementation

iagorrr · Jan 25, 2024 · cabd383 · cabd383
1 parent 7d285e9
commit cabd383
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 30 deletions.
diff --git a/algorithms/graphs/tree-diameter-(dp).cpp b/algorithms/graphs/tree-diameter-(dp).cpp
@@ -0,0 +1,19 @@
+const int MAXN(1'000'000);
+int N;
+vi G[MAXN];
+
+int diameter, toLeaf[MAXN];
+void calcDiameter(int u = 0, int p = -1) {
+  int d1, d2;
+  d1 = d2 = -1;
+
+  for (auto v : G[u]) {
+    if (v != p) {
+      calcDiameter(v, u);
+      d1 = max(d1, toLeaf[v]);
+      tie(d1, d2) = minmax({d1, d2});
+    }
+  }
+  toLeaf[u] = d2 + 1;
+  diameter = max(diameter, d1 + d2 + 2);
+}
diff --git a/algorithms/graphs/tree-diameter.tex → algorithms/graphs/tree-diameter-(dp).tex b/algorithms/graphs/tree-diameter.tex → algorithms/graphs/tree-diameter-(dp).tex
diff --git a/algorithms/graphs/tree-diameter.cpp b/algorithms/graphs/tree-diameter.cpp