Problem: 2368. 受限條件下可到達節點的數目

思路

👨?🏫 靈神

復雜度

時間復雜度: $O(n)$

空間復雜度: $O(n)$

Code

class Solution {int ans = 0;boolean[] set;List<Integer>[] es;public int reachableNodes(int n, int[][] edges, int[] restricted){es = new ArrayList[n];Arrays.setAll(es, x->new ArrayList<>());set = new boolean[n];for (int[] e : edges){int a = e[0];int b = e[1];es[a].add(b);es[b].add(a);}for (int x : restricted)set[x] = true;dfs(0);return ans;}private void dfs(int x){if (set[x])return;ans++;set[x] = true;for (int y : es[x])dfs(y);}
}