例題:力扣547 一道經典的並查集
以前看到這種題總想着暴力去做,又費時間又費精力,而且還冒着tle的風險。
以後這種題用並查集,ez的很。
class Solution {
public:
int pre[2000] = { 0 };
int find(int x)
{
while (pre[x] != x)x = pre[x];
return x;
}
void merge(int x, int y)
{
int a = find(x);
int b = find(y);
if (a != b)pre[b] = a;
}
int findCircleNum(vector<vector<int>>& M)
{
int index = 0;
for (int i = 0; i < M.size(); i++)pre[i] = i;
for (int i = 0; i < M.size(); i++)for (int j = i; j < M.size(); j++)if (M[i][j])merge(i, j);
for (int i = 0; i < M.size(); i++)if (pre[i] == i)index++;
return index;
}
};