Calculate the shortest distance between two lakes in a matrix of ones and zeroes where the land was ones and the water was zeros

Question

I have asked this question in one if the interview, I have worked on Graph(BFS/DFS) , but I am not sure how to modify program of finding number of connected components solution