力扣每日一题受限条件下可到达节点的数目 DFS

最新推荐文章于 2024-06-12 10:00:23 发布

兑生

最新推荐文章于 2024-06-12 10:00:23 发布

阅读量480

点赞数 3

分类专栏： # 各平台OJ题解文章标签：深度优先 leetcode 算法

兑生：前世大厂架构师、上市公司CEO，被小人陷害，重生为25应届生，三连加关注，看我如何逆风翻盘。

本文链接：https://blog.csdn.net/lt6666678/article/details/136418051

版权

各平台OJ题解专栏收录该内容

73 篇文章 0 订阅

订阅专栏

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);
	}
}