1. 程式人生 > 其它 >LeetCode—947. 移除最多的同行或同列石頭(Most Stones Removed with Same Row or Column)——分析及程式碼(Java)

LeetCode—947. 移除最多的同行或同列石頭(Most Stones Removed with Same Row or Column)——分析及程式碼(Java)

技術標籤:資料結構與演算法LeetCodeJava題解

LeetCode—947. 移除最多的同行或同列石頭[Most Stones Removed with Same Row or Column]——分析及程式碼[Java]

一、題目

n 塊石頭放置在二維平面中的一些整數座標點上。每個座標點上最多隻能有一塊石頭。
如果一塊石頭的 同行或者同列 上有其他石頭存在,那麼就可以移除這塊石頭。
給你一個長度為 n 的陣列 stones ,其中 stones[i] = [xi, yi] 表示第 i 塊石頭的位置,返回 可以移除的石子 的最大數量。

示例 1:

輸入:stones = [[0,0],[0,1],[1,0],[1,2],[2,1],[2,2]]
輸出:5
解釋:一種移除 5 塊石頭的方法如下所示:
1. 移除石頭 [2,2] ,因為它和 [2,1] 同行。
2. 移除石頭 [2,1] ,因為它和 [0,1] 同列。
3. 移除石頭 [1,2] ,因為它和 [1,0] 同行。
4. 移除石頭 [1,0] ,因為它和 [0,0] 同列。
5. 移除石頭 [0,1] ,因為它和 [0,0] 同行。
石頭 [0,0] 不能移除,因為它沒有與另一塊石頭同行/列。

示例 2:

輸入:stones = [[0,0],[0,2],[1,1],[2,0],[2,2]]
輸出:3
解釋:一種移除 3 塊石頭的方法如下所示:
1. 移除石頭 [2,2] ,因為它和 [2,0] 同行。
2. 移除石頭 [2,0] ,因為它和 [0,0] 同列。
3. 移除石頭 [0,2] ,因為它和 [0,0] 同行。
石頭 [0,0] 和 [1,1] 不能移除,因為它們沒有與另一塊石頭同行/列。

示例 3:

輸入:stones = [[0,0]]
輸出:0
解釋:[0,0] 是平面上唯一一塊石頭,所以不可以移除它。

提示:

  • 1 <= stones.length <= 1000
  • 0 <= xi, yi <= 104
  • 不會有兩塊石頭放在同一個座標點上

來源:力扣(LeetCode)
連結:https://leetcode-cn.com/problems/most-stones-removed-with-same-row-or-column
著作權歸領釦網路所有。商業轉載請聯絡官方授權,非商業轉載請註明出處。

二、分析及程式碼

1. 並查集

(1)思路

將出現在同一行/列的石頭計入同一連通分量,根據題意,每一連通分量最少只保留 1 塊石頭,進而可求得答案。

為簡化過程,可將橫、縱座標用同一陣列表示,其中 0 - 10000 表示橫座標,10001 - 20001 表示縱座標,即可避免衝突。

(2)程式碼

class Solution {
    public int removeStones(int[][] stones) {
        int [] parent = new int[20002];//10001行,10001列
        Arrays.fill(parent, -1);
        for (int i = 0; i < stones.length; i++)
            union(parent, stones[i][0], stones[i][1] + 10001);
        int ans = stones.length;
        for (int i = 0; i < parent.length; i++)
            if (parent[i] == i)
                ans--;
        return ans;
    }

    public int find(int[] parent, int index) {
        if (parent[index] < 0)
            parent[index] = index;
        if (parent[index] != index)
            parent[index] = find(parent, parent[index]);
        return parent[index];
    }

    public void union(int[] parent, int index1, int index2) {
        parent[find(parent, index2)] = find(parent, index1);
    }
}

(3)結果

執行用時 :5 ms,在所有 Java 提交中擊敗了 98.01% 的使用者;
記憶體消耗 :38.9 MB,在所有 Java 提交中擊敗了 29.44% 的使用者。

三、其他

暫無。