Union Find

uf.ts

class UF {
    public vertices: Array<number>;

    constructor(n: number) {
        this.vertices = Array.from({length: n}, (_, i) => i);
    }

    union(x: number, y: number){
        this.vertices[this.find(x)] = this.find(y);
    }

    find(x: number) {
        return x === this.vertices[x]? x: this.find(this.vertices[x]);
    }
    
    connected(x: number, y: number) {
        return this.find(x) === this.find(y);
    }

    count() {
        return this.vertices.filter((p, i) => p === i).length;
    }
}