LC 2368. 受限条件下可到达节点的数目

2368. 受限条件下可到达节点的数目

��度 : 中等

题目大意：

现有一棵由 n 个节点组成的无向树，节点编号从 0 到 n - 1 ，共有 n - 1 条边。

给你一个二维整数数组 edges ，长度为 n - 1 ，其中 edges[i] = [ai, bi] 表示树中节点 ai 和 bi 之间存在一条边。另给你一个整数数组 restricted 表示 受限 节点。

在不访问受限节点的前提下，返回你可以从节点 0 到达的 最多 节点数目*。*

注意，节点 0 不 会标记为受限节点。

提示：

• 2 public: int reachableNodes(int n, vector int st[n + 1]; memset(st, 0, sizeof st); for (int x : restricted) st[x] = true; vector int a = edges[i][0], b = edges[i][1]; if (st[a] || st[b]) continue; g[a].push_back(b); g[b].push_back(a); } int res = 1; function for (int x : g[u]) { if (x == fa) continue; res ++; dfs(x, u); } }; dfs(0, -1); return res; }};public: int reachableNodes(int n, vector unordered_map if (p[x] != x) p[x] = find(p[x]); return p[x]; }; for (int i = 0; i