Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

https://github.com/emahtab/connecting-cities-with-minimum-cost

Minimum Spanning Tree
https://github.com/emahtab/connecting-cities-with-minimum-cost

graph kruskals-algorithm leetcode minimum-spanning-tree prims-algorithm

Last synced: about 2 months ago
JSON representation

Minimum Spanning Tree

Awesome Lists containing this project

README 

        
# Connecting cities with minimum cost

## https://leetcode.com/problems/connecting-cities-with-minimum-cost



There are n cities labeled from 1 to n. You are given the integer n and an array connections where connections[i] = [xi, yi, costi] indicates that the cost of connecting city xi and city yi (bidirectional connection) is costi.



Return the minimum cost to connect all the n cities such that there is at least one path between each pair of cities. If it is impossible to connect all the n cities, return -1,



The cost is the sum of the connections' costs used.



![Connecting cities with minimum cost](example.JPG?raw=true)



# Implementation 1 : Minimum Spanning Tree (Prim's Algorithm)

```java

class Solution {

    class Edge {

        int v1, v2, cost;

        Edge(int v1, int v2, int cost){

            this.v1 = v1;

            this.v2 = v2;

            this.cost = cost;

        }

    }

    public int minimumCost(int n, int[][] connections) {

        Map> map = new HashMap<>();

        for(int[] connection : connections) {

            int v1 = connection[0] - 1;

            int v2 = connection[1] - 1;

            map.putIfAbsent(v1, new ArrayList());

            map.get(v1).add(new Edge(v1,v2,connection[2]));

            

            map.putIfAbsent(v2, new ArrayList());

            map.get(v2).add(new Edge(v2,v1,connection[2]));

        }

        

        boolean[] visited = new boolean[n];

        PriorityQueue pq = new PriorityQueue<>((v1, v2) -> v1[1] - v2[1]);

        pq.add(new int[]{0,0});

        

        int minCost = 0;

        int count = 0;

        while(!pq.isEmpty()) {

            int[] vertex = pq.remove();

            if(!visited[vertex[0]]) {

               minCost += vertex[1]; 

               visited[vertex[0]] = true;

               count++; 

                

               // add unvisited neighbors of current vertex to priority queue

               List edges = map.get(vertex[0]);

               for(Edge edge : edges) {

                   if(!visited[edge.v2]) {

                       pq.add(new int[]{edge.v2, edge.cost});

                   }

               } 

            }

        }

        return count == n ? minCost : -1;

    }

}

```



# Implementation 1a : Without creating Edge, a little more succinct code

```java

class Solution {

    public int minimumCost(int n, int[][] connections) {

        Map> graph = new HashMap<>();

        for(int i= 0; i < n; i++)

            graph.put(i, new ArrayList<>());

        for(int[] connection : connections) {

            int u = connection[0]-1, v = connection[1]-1, cost = connection[2];

            graph.get(u).add(new int[]{v, cost});

            graph.get(v).add(new int[]{u, cost});

        }



        PriorityQueue pq = new PriorityQueue<>((n1, n2) -> n1[1] - n2[1]);

        pq.add(new int[]{0, 0});

        int minCost = 0;

        Set visited = new HashSet<>();

        while(!pq.isEmpty()) {

            int[] node = pq.remove();

            if(visited.contains(node[0])) continue;

            minCost += node[1];

            visited.add(node[0]);

            for(int[] neighbor : graph.get(node[0])) {

                if(!visited.contains(neighbor[0]))

                   pq.add(neighbor);

            }

        }

        return visited.size() == n ? minCost : -1;

    }

}

```

# Implementation 2 : Minimum Spanning Tree (Kruskal's Algorithm using Union Find)

```java

class Solution {

    

    public int minimumCost(int n, int[][] connections) {

        int minCost = 0;

        int edgesConnected = 0;

        Arrays.sort(connections, (e1, e2) -> e1[2] - e2[2]);

        int[] parents = new int[n];

        int[] rank = new int[n];

        for(int i = 0; i < n; i++) {

            parents[i] = i;

            rank[i] = 1;

        }

        

        for(int[] connection : connections) {

            boolean flag = union(connection[0] - 1, connection[1] - 1, parents, rank);

            if(flag) {

                minCost += connection[2];

                edgesConnected++;

            }        

        }

        

        return edgesConnected == n - 1 ? minCost : -1;

    }

    

    private boolean union(int v1, int v2, int[] parents, int[] rank) {

        int leader1 = find(v1, parents);

        int leader2 = find(v2, parents);

        if(leader1 != leader2) {

            if(rank[leader1] > rank[leader2]) {

                parents[leader2] = leader1;

            } else if(rank[leader2] > rank[leader1]) {

                parents[leader1] = leader2;

            } else {

                parents[leader2] = leader1;

                rank[leader1]++;

            }

            return true;

        } else {

            return false;

        }

    }

    

    private int find(int v, int[] parents) {

        if(parents[v] == v)

            return parents[v];

        parents[v] = find(parents[v], parents);

        return parents[v];

    }

}

```



# References :

1. Prim's Algorithm : https://www.youtube.com/watch?v=Vw-sktU1zmc (Hindi, very good explanation)

2. Kruskal's Algorithm : https://www.youtube.com/watch?v=YU3Bvek4wjs (Hindi, very good explanation)