力扣(LeetCode)934

題目地址:
https://leetcode-cn.com/probl...
題目描述:
在給定的二維二進制數組 A 中,存在兩座島。(島是由四面相連的 1 造成的一個最大組。)java

如今,咱們能夠將 0 變爲 1,以使兩座島鏈接起來,變成一座島。python

返回必須翻轉的 0 的最小數目。(能夠保證答案至少是 1。)算法

解答:
題的意思就是說,有兩片島嶼,島嶼是指聯通的1組成的區域,求這兩片島嶼之間的最短距離。
思路是這樣的,用圖的深度優先搜索dfs算法,來找出第一個島嶼裏的所有點,加入list1,而後
一樣的方法找到第二個島嶼的所有點,加入list2,而後求list1中任意一點到list2中任意一點
距離的最小值便可,這裏的距離是指abs(x1-x2)+abs(y1-y2)-1
abs爲絕對值,x1,y1爲list1的某點的座標,x2,y2爲list2的某點座標。
這一題考察的是圖的深度優先搜索
java ac代碼:數組

class Solution {

class Node
{
    int x,y;
    public Node(int x,int y)
    {
        this.x = x;
        this.y = y;
    }
}

public int shortestBridge(int[][] A) {
    
    boolean[][] visited = new boolean[A.length][A[0].length];
    List<Node> l1 = new ArrayList(1000);
    List<Node> l2 = new ArrayList(1000);
    
    
    for(int i = 0;i < A.length;i++)
    {
        boolean flag = false;
        for(int j = 0;j < A[0].length;j++)
            if(A[i][j] == 1&&!visited[i][j]){
                dfs(A,visited,i,j,l1);
                flag = true;
                break;
            }
        if(flag)break;
    }
    
   for(int i = 0;i < A.length;i++)
    {
        boolean flag = false;
        for(int j = 0;j < A[0].length;j++)
            if(A[i][j] == 1&&!visited[i][j]){
                dfs(A,visited,i,j,l2);
                flag = true;
                break;
            }
        if(flag)break;
    }
    
    
    int ans = Integer.MAX_VALUE;
    
     for(int i = 0;i < l1.size();i++)
        for(int j = 0;j < l2.size();j++)
           ans = Math.min(ans,Math.abs(l1.get(i).x-l2.get(j).x)+Math.abs(l1.get(i).y-l2.get(j).y)-1);
            
    
    return ans;
    
    
}


void dfs(int[][] A,boolean[][] visited,int x,int y,List<Node> l)
{
    if(!(x >= 0&&x < A.length&& y>=0 && y< A[0].length) || visited[x][y]||A[x][y] !=1 )
        return;
    
    visited[x][y] = true;
    l.add(new Node(x,y));
    dfs(A,visited,x-1,y,l);
    dfs(A,visited,x+1,y,l);
    dfs(A,visited,x,y-1,l);
    dfs(A,visited,x,y+1,l);
    
}

}

用時400多ms,若是用python可能會超時。this

相關文章
相關標籤/搜索