LeetCode 947. 移除最多的同行或同列石头--并查集

947. 移除最多的同行或同列石头

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
不会有两块石头放在同一个坐标点上

题解

思路比较简单，简单来说就是，如果点x和点y同行或同列，那么就可以连接，那么很多点通过这种方式就可以连接成一个集合，那么这个集合只需要剩下一个点，其他点全部删除，同样，我们可以得到很多个这样的集合，每个集合点的数组为ans[x]，那么答案就是res+=ans[x]-1。

AC代码

class Solution {
public:
    int f[1000];
    int fin(int x)
    {
        if(x==f[x])return x;
        else return f[x]=fin(f[x]);
    }
    int ans[1000];
    map<int,bool>vis;
    vector<int>q;
    int removeStones(vector<vector<int>>& stones) {
        int n=stones.size();
        for(int i=0;i<n;i++)
        f[i]=i;
        for(int i=0;i<n;i++)
        {
            for(int j=i+1;j<n;j++)
            {
                if(stones[i][0]==stones[j][0]||stones[i][1]==stones[j][1])
                {
                    int fx=fin(i);
                    int fy=fin(j);
                    f[fx]=fy;
                }
            }
        }
        vis.clear();
        memset(ans,0,sizeof(ans));
        for(int i=0;i<n;i++)
        {
            int fx=fin(i);
            ans[fx]++;
            if(vis[fx]==false)
            {
                vis[fx]=true;
                q.push_back(fx);
            }
        }
        int res=0;
        for(int i=0;i<q.size();i++)
        {
            res+=(ans[q[i]]-1);
        }
        return res;
    }
};