Update 2092-find-all-people-with-secret.js

Author: everthis

2092-find-all-people-with-secret.js

@@ -1,3 +1,55 @@
+/**
+ * @param {number} n
+ * @param {number[][]} meetings
+ * @param {number} firstPerson
+ * @return {number[]}
+ */
+const findAllPeople = function(n, meetings, firstPerson) {
+  meetings.sort((a, b) => a[2] - b[2])
+  const uf = new UnionFind(n);
+  uf.connect(0, firstPerson);
+  let ppl = [];
+  for (let i = 0, len = meetings.length; i < len; ) {
+    ppl = [];
+    let time = meetings[i][2];
+    while (i < len && meetings[i][2] === time) {
+      uf.connect(meetings[i][0], meetings[i][1]);
+      ppl.push(meetings[i][0]);
+      ppl.push(meetings[i][1]);
+      i++
+    }
+    for (let n of ppl) {
+      if (!uf.connected(0, n)) uf.reset(n);
+    }
+  }
+  let ans = [];
+  for (let i = 0; i < n; ++i) {
+    if (uf.connected(0, i)) ans.push(i);
+  }
+  return ans;
+};
+
+class UnionFind {
+  constructor(n) {
+    this.arr = Array(n).fill(null)
+    this.arr.forEach((e, i, arr) => arr[i] = i)
+  }
+  connect(a, b) {
+    this.arr[this.find(a)] = this.find(this.arr[b])
+  }
+  find(a) {
+    return this.arr[a] === a ? a : (this.arr[a] = this.find(this.arr[a]))
+  }
+  connected(a, b) {
+    return this.find(a) === this.find(b)
+  }
+  reset(a) {
+    this.arr[a] = a
+  }
+}
+
+// another
+
 /**
  * @param {number} n
  * @param {number[][]} meetings