Suppose there are n towns connected by m bidirectional roads. There are s towns among them with a police station. We want to find out the distance of each town from the nearest police station. If the town itself has one the distance is 0.
Example:
Input : Number of Vertices = 6 Number of Edges = 9 Towns with Police Station : 1, 5 Edges: 1 2 1 6 2 6 2 3 3 6 5 4 6 5 3 4 5 3
Output : 1 0 2 1 3 1 4 1 5 0 6 1
Naive Approach: We can loop through the vertices and from each vertex run a BFS to find the closest town with police station from that vertex. This will take O(V.E).
Naive approach implementation using BFS from each vertex:
C++
// C++ program to demonstrate distance to // nearest source problem using BFS // from each vertex #include <bits/stdc++.h> using namespace std; #define N 100000 + 1 #define inf 1000000 // This array stores the distances of the // vertices from the nearest source int dist[N]; // a hash array where source[i] = 1 // means vertex i is a source int source[N]; // The BFS Queue // The pairs are of the form (vertex, distance // from current source) deque<pair< int , int > > BFSQueue; // visited array for remembering visited vertices int visited[N]; // The BFS function void BFS(vector< int > graph[], int start) { // clearing the queue while (!BFSQueue.empty()) BFSQueue.pop_back(); // push_back starting vertices BFSQueue.push_back({ start, 0 }); while (!BFSQueue.empty()) { int s = BFSQueue.front().first; int d = BFSQueue.front().second; visited[s] = 1; BFSQueue.pop_front(); // stop at the first source we reach during BFS if (source[s] == 1) { dist[start] = d; return ; } // Pushing the adjacent unvisited vertices // with distance from current source = this // vertex's distance + 1 for ( int i = 0; i < graph[s].size(); i++) if (visited[graph[s][i]] == 0) BFSQueue.push_back({ graph[s][i], d + 1 }); } } // This function calculates the distance of each // vertex from nearest source void nearestTown(vector< int > graph[], int n, int sources[], int S) { // reseting the source hash array for ( int i = 1; i <= n; i++) source[i] = 0; for ( int i = 0; i <= S - 1; i++) source[sources[i]] = 1; // loop through all the vertices and run // a BFS from each vertex to find the distance // to nearest town from it for ( int i = 1; i <= n; i++) { for ( int i = 1; i <= n; i++) visited[i] = 0; BFS(graph, i); } // Printing the distances for ( int i = 1; i <= n; i++) cout << i << " " << dist[i] << endl; } void addEdge(vector< int > graph[], int u, int v) { graph[u].push_back(v); graph[v].push_back(u); } // Driver Code int main() { // Number of vertices int n = 6; vector< int > graph[n + 1]; // Edges addEdge(graph, 1, 2); addEdge(graph, 1, 6); addEdge(graph, 2, 6); addEdge(graph, 2, 3); addEdge(graph, 3, 6); addEdge(graph, 5, 4); addEdge(graph, 6, 5); addEdge(graph, 3, 4); addEdge(graph, 5, 3); // Sources int sources[] = { 1, 5 }; int S = sizeof (sources) / sizeof (sources[0]); nearestTown(graph, n, sources, S); return 0; } |
Java
// Java program to demonstrate distance to // nearest source problem using BFS // from each vertex import java.util.ArrayList; import java.util.Arrays; import java.util.Deque; import java.util.LinkedList; class Pair { int first, second; public Pair( int first, int second) { this .first = first; this .second = second; } } class GFG{ static final int N = 100000 + 1 ; static final int inf = 1000000 ; // This array stores the distances of the // vertices from the nearest source static int [] dist = new int [N]; // a hash array where source[i] = 1 // means vertex i is a source static int [] source = new int [N]; // The BFS Queue // The pairs are of the form (vertex, distance // from current source) static Deque<Pair> BFSQueue = new LinkedList<>(); // deque<pair<int, int> > BFSQueue; // visited array for remembering visited vertices static int [] visited = new int [N]; // The BFS function static void BFS(ArrayList<Integer>[] graph, int start) { // Clearing the queue while (!BFSQueue.isEmpty()) BFSQueue.removeLast(); // push_back starting vertices BFSQueue.add( new Pair(start, 0 )); while (!BFSQueue.isEmpty()) { int s = BFSQueue.peekFirst().first; int d = BFSQueue.peekFirst().second; visited[s] = 1 ; BFSQueue.removeFirst(); // Stop at the first source // we reach during BFS if (source[s] == 1 ) { dist[start] = d; return ; } // Pushing the adjacent unvisited vertices // with distance from current source = this // vertex's distance + 1 for ( int i = 0 ; i < graph[s].size(); i++) if (visited[graph[s].get(i)] == 0 ) BFSQueue.add( new Pair( graph[s].get(i), d + 1 )); } } // This function calculates the distance of each // vertex from nearest source static void nearestTown(ArrayList<Integer>[] graph, int n, int sources[], int S) { // Reseting the source hash array for ( int i = 1 ; i <= n; i++) source[i] = 0 ; for ( int i = 0 ; i <= S - 1 ; i++) source[sources[i]] = 1 ; // Loop through all the vertices and run // a BFS from each vertex to find the distance // to nearest town from it for ( int i = 1 ; i <= n; i++) { for ( int j = 1 ; j <= n; j++) visited[j] = 0 ; BFS(graph, i); } // Printing the distances for ( int i = 1 ; i <= n; i++) System.out.println(i + " " + dist[i]); } static void addEdge(ArrayList<Integer>[] graph, int u, int v) { graph[u].add(v); graph[v].add(u); } // Driver Code public static void main(String[] args) { // Number of vertices int n = 6 ; @SuppressWarnings ( "unchecked" ) ArrayList<Integer>[] graph = new ArrayList[n + 1 ]; Arrays.fill(graph, new ArrayList<>()); // Edges addEdge(graph, 1 , 2 ); addEdge(graph, 1 , 6 ); addEdge(graph, 2 , 6 ); addEdge(graph, 2 , 3 ); addEdge(graph, 3 , 6 ); addEdge(graph, 5 , 4 ); addEdge(graph, 6 , 5 ); addEdge(graph, 3 , 4 ); addEdge(graph, 5 , 3 ); // Sources int sources[] = { 1 , 5 }; int S = sources.length; nearestTown(graph, n, sources, S); } } // This code is contributed by sanjeev2552 |
Python3
# Python3 program to demonstrate distance to # nearest source problem using BFS # from each vertex N = 100001 inf = 1000000 # This array stores the distances of the # vertices from the nearest source dist = [ 0 for i in range (N)]; # a hash array where source[i] = 1 # means vertex i is a source source = [ 0 for i in range (N)]; # The BFS Queue # The pairs are of the form (vertex, distance # from current source) BFSQueue = [] # visited array for remembering visited vertices visited = [ 0 for i in range (N)]; # The BFS function def BFS(graph, start): # clearing the queue while ( len (BFSQueue) ! = 0 ): BFSQueue.pop(); # append starting vertices BFSQueue.append([ start, 0 ]); while ( len (BFSQueue) ! = 0 ): s = BFSQueue[ 0 ][ 0 ]; d = BFSQueue[ 0 ][ 1 ]; visited[s] = 1 ; BFSQueue.pop( 0 ); # stop at the first source we reach during BFS if (source[s] = = 1 ): dist[start] = d; return ; # Pushing the adjacent unvisited vertices # with distance from current source = this # vertex's distance + 1 for i in range ( len (graph[s])): if (visited[graph[s][i]] = = 0 ): BFSQueue.append([ graph[s][i], d + 1 ]); # This function calculates the distance of each # vertex from nearest source def nearestTown(graph, n, sources, S): global source, dist # reseting the source hash array for i in range ( 1 , n + 1 ): source[i] = 0 ; for i in range (S): source[sources[i]] = 1 ; # loop through all the vertices and run # a BFS from each vertex to find the distance # to nearest town from it for i in range ( 1 , n + 1 ): for j in range ( 1 , n + 1 ): visited[j] = 0 ; BFS(graph, i); # Printing the distances for i in range ( 1 , n + 1 ): print ( '{} {}' . format (i,dist[i])) def addEdge(graph, u, v): graph[u].append(v); graph[v].append(u); # Driver Code if __name__ = = '__main__' : # Number of vertices n = 6 graph = [[] for i in range (n + 1 )]; # Edges addEdge(graph, 1 , 2 ); addEdge(graph, 1 , 6 ); addEdge(graph, 2 , 6 ); addEdge(graph, 2 , 3 ); addEdge(graph, 3 , 6 ); addEdge(graph, 5 , 4 ); addEdge(graph, 6 , 5 ); addEdge(graph, 3 , 4 ); addEdge(graph, 5 , 3 ); # Sources sources = [ 1 , 5 ] S = len (sources) nearestTown(graph, n, sources, S); # This code is contributed by rutvik_56 |
Output:
1 0 2 1 3 1 4 1 5 0 6 1
Time Complexity: O(V.E)
Efficient Method A better method is to use the Djikstra’s algorithm in a modified way. Let’s consider one of the sources as the original source and the other sources to be vertices with 0 cost paths from the original source. Thus we push all the sources into the Djikstra Queue with distance = 0, and the rest of the vertices with distance = infinity. The minimum distance of each vertex from the original source now calculated using the Dijkstra’s Algorithm are now essentially the distances from the nearest source.
Explanation: The C++ implementation uses a set of pairs (distance from the source, vertex) sorted according to the distance from the source. Initially, the set contains the sources with distance = 0 and all the other vertices with distance = infinity.
On each step, we will go to the vertex with minimum distance(d) from source, i.e, the first element of the set (the source itself in the first step with distance = 0). We go through all it’s adjacent vertices and if the distance of any vertex is > d + 1 we replace its entry in the set with the new distance. Then we remove the current vertex from the set. We continue this until the set is empty.
The idea is there cannot be a shorter path to the vertex at the front of the set than the current one since any other path will be a sum of a longer path (>= it’s length) and a non-negative path length (unless we are considering negative edges).
Since all the sources have a distance = 0, in the beginning, the adjacent non-source vertices will get a distance = 1. All vertices will get distance = distance from their nearest source.
Implementation of Efficient Approach:
C++
// C++ program to demonstrate // multi-source BFS #include <bits/stdc++.h> using namespace std; #define N 100000 + 1 #define inf 1000000 // This array stores the distances of the vertices // from the nearest source int dist[N]; // This Set contains the vertices not yet visited in // increasing order of distance from the nearest source // calculated till now set<pair< int , int > > Q; // Util function for Multi-Source BFS void multiSourceBFSUtil(vector< int > graph[], int s) { set<pair< int , int > >::iterator it; int i; for (i = 0; i < graph[s].size(); i++) { int v = graph[s][i]; if (dist[s] + 1 < dist[v]) { // If a shorter path to a vertex is // found than the currently stored // distance replace it in the Q it = Q.find({ dist[v], v }); Q.erase(it); dist[v] = dist[s] + 1; Q.insert({ dist[v], v }); } } // Stop when the Q is empty -> All // vertices have been visited. And we only // visit a vertex when we are sure that a // shorter path to that vertex is not // possible if (Q.size() == 0) return ; // Go to the first vertex in Q // and remove it from the Q it = Q.begin(); int next = it->second; Q.erase(it); multiSourceBFSUtil(graph, next); } // This function calculates the distance of // each vertex from nearest source void multiSourceBFS(vector< int > graph[], int n, int sources[], int S) { // a hash array where source[i] = 1 // means vertex i is a source int source[n + 1]; for ( int i = 1; i <= n; i++) source[i] = 0; for ( int i = 0; i <= S - 1; i++) source[sources[i]] = 1; for ( int i = 1; i <= n; i++) { if (source[i]) { dist[i] = 0; Q.insert({ 0, i }); } else { dist[i] = inf; Q.insert({ inf, i }); } } set<pair< int , int > >::iterator itr; // Get the vertex with lowest distance, itr = Q.begin(); // currently one of the souces with distance = 0 int start = itr->second; multiSourceBFSUtil(graph, start); // Printing the distances for ( int i = 1; i <= n; i++) cout << i << " " << dist[i] << endl; } void addEdge(vector< int > graph[], int u, int v) { graph[u].push_back(v); graph[v].push_back(u); } // Driver Code int main() { // Number of vertices int n = 6; vector< int > graph[n + 1]; // Edges addEdge(graph, 1, 2); addEdge(graph, 1, 6); addEdge(graph, 2, 6); addEdge(graph, 2, 3); addEdge(graph, 3, 6); addEdge(graph, 5, 4); addEdge(graph, 6, 5); addEdge(graph, 3, 4); addEdge(graph, 5, 3); // Sources int sources[] = { 1, 5 }; int S = sizeof (sources) / sizeof (sources[0]); multiSourceBFS(graph, n, sources, S); return 0; } |
Output:
1 0 2 1 3 1 4 1 5 0 6 1
Time Complexity: O(E.logV)
More Efficient Approach: An even better method is to use the Multisource BFS which is a modification of BFS.We will put the all source vertices to the queue at first rather than a single vertex which was in case of standard BFS.This way Multisource BFS will first visit all the source vertices. After that it will visit the vertices which are at a distance of 1 from all source vertices, then at a distance of 2 from all source vertices and so on and so forth.
Below is the implementation of the above approach:
C++
// C++ program to demonstrate Multi-source BFS #include<bits/stdc++.h> using namespace std; #define N 1000000 // This array stores the distances of the vertices // from the nearest source int dist[N]; //This boolean array is true if the current vertex // is visited otherwise it is false bool visited[N]; void addEdge(vector< int > graph[], int u, int v) { //Function to add 2 edges in an undirected graph graph[u].push_back(v); graph[v].push_back(u); } // Multisource BFS Function void Multisource_BFS(vector< int > graph[],queue< int >q) { while (!q.empty()) { int k = q.front(); q.pop(); for ( auto i:graph[k]) { if (!visited[i]) { // Pushing the adjacent unvisited vertices // with distance from current source = this // vertex's distance + 1 q.push(i); dist[i] = dist[k] + 1; visited[i] = true ; } } } } // This function calculates the distance of each // vertex from nearest source void nearestTown(vector< int > graph[], int n, int sources[], int s) { //Create a queue for BFS queue< int > q; //Mark all the source vertices as visited and enqueue it for ( int i = 0;i < s; i++) { q.push(sources[i]); visited[sources[i]]= true ; } Multisource_BFS(graph,q); //Printing the distances for ( int i = 1; i <= n; i++) { cout<< i << " " << dist[i] << endl; } } // Driver code int main() { // Number of vertices int n = 6; vector< int > graph[n + 1]; // Edges addEdge(graph, 1, 2); addEdge(graph, 1, 6); addEdge(graph, 2, 6); addEdge(graph, 2, 3); addEdge(graph, 3, 6); addEdge(graph, 5, 4); addEdge(graph, 6, 5); addEdge(graph, 3, 4); addEdge(graph, 5, 3); // Sources int sources[] = { 1, 5 }; int S = sizeof (sources) / sizeof (sources[0]); nearestTown(graph, n, sources, S); return 0; } |
Output:
1 0 2 1 3 1 4 1 5 0 6 1
Time Complexity: O(V+E)