merged

1 files changed, 4 insertions, 6 deletions

diff --git a/datastructures/unionFind.cpp b/datastructures/unionFind.cpp
index 03ff63e..35843cd 100644
--- a/datastructures/unionFind.cpp
+++ b/datastructures/unionFind.cpp
@@ -12,12 +12,10 @@ int findSet(int n) { // Pfadkompression
 }
 
 void linkSets(int a, int b) { // Union by rank.
-	if (unions[a] > unions[b]) unions[a] = b;
-	else if (unions[b] > unions[a]) unions[b] = a;
-	else {
-		unions[a] = b;
-		unions[b]--;
-}}
+	if (unions[b] > unions[a]) swap(a, b);
+	if (unions[b] == unions[a]) unions[b]--;
+	unions[a] = b;
+}
 
 void unionSets(int a, int b) { // Diese Funktion aufrufen.
 	if (findSet(a) != findSet(b)) linkSets(findSet(a), findSet(b));