Union found
Use array.
class UnionFound { public: vector<int> v; int cnt; UnionFound(int n) { v = vector<int>(n); for (int i = 0; i < n; i++) v[i] = i; cnt = n; } int findParent(int i) { if (v[i] == i) return i; v[i] = findParent(v[i]); return v[i]; } void Union(int p, int c) { int pp = findParent(p); int cp = findParent(c); if (pp != cp) { v[cp] = pp; cnt--; } } };
Use map.
class UnionFound { public: unordered_map<int,int> parents; int cnt = 0; void AddItem(int i) { parents[i] = i; cnt++; } int GetParent(int i) { if (parents[i] == i) return i; return parents[i] = GetParent(parents[i]); } void Union(int p, int c) { int pp = GetParent(p); int cp = GetParent(c); if (pp != cp) { parents[cp] = pp; cnt--; } } };
【推荐】还在用 ECharts 开发大屏?试试这款永久免费的开源 BI 工具!
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步