美文网首页
【算法题】2368. 受限条件下可到达节点的数目

【算法题】2368. 受限条件下可到达节点的数目

作者: 程序员小2 | 来源:发表于2023-04-26 07:51 被阅读0次

题目:

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

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

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

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

示例 1:

image.png

输入:n = 7, edges = [[0,1],[1,2],[3,1],[4,0],[0,5],[5,6]], restricted = [4,5]
输出:4
解释:上图所示正是这棵树。
在不访问受限节点的前提下,只有节点 [0,1,2,3] 可以从节点 0 到达。

示例 2:


image.png

输入:n = 7, edges = [[0,1],[0,2],[0,5],[0,4],[3,2],[6,5]], restricted = [4,2,1]
输出:3
解释:上图所示正是这棵树。
在不访问受限节点的前提下,只有节点 [0,5,6] 可以从节点 0 到达。

提示:

2 <= n <= 10^5
edges.length == n - 1
edges[i].length == 2
0 <= ai, bi < n
ai != bi
edges 表示一棵有效的树
1 <= restricted.length < n
1 <= restricted[i] < n
restricted 中的所有值 互不相同

java代码:

class Solution {
    public int reachableNodes(int n, int[][] edges, int[] restricted) {
        List<Integer>[] adj = new List[n];
        for (int i = 0; i < n; ++i){
            adj[i] = new ArrayList<>();
        }
        // 邻接表建图
        for (int i = 0; i < n - 1; ++i){
            adj[edges[i][0]].add(edges[i][1]);
            adj[edges[i][1]].add(edges[i][0]);
        }
        boolean[] vis = new boolean[n];
        // 处理restricted数组
        for (int num : restricted) vis[num] = true;
        Deque<Integer> q = new LinkedList<>();
        q.addLast(0);
        vis[0] = true;
        int ans = 1;
        while (!q.isEmpty()){
            int cur = q.pollFirst();
            for (int next : adj[cur]){
                if (!vis[next]){
                    q.addLast(next);
                    vis[next] = true;
                    ++ans;
                }
            }
        }
        return ans;
    }
}

相关文章

  • 图论(七)图的广度优先遍历BFS

    前言 广度优先搜索是对图中的边进行系统性的探索来发现可以从源节点出发到达的所有节点。该算法能够计算从源节点到每个可...

  • 2020-01-09 本体的VBFT共识算法

    概括 共识节点 共识候选节点 共识网络构建 随机节点完成共识 算法概述 VBFT算法可以认为是传统BFT算法在可验...

  • KNN算法进行字母识别、决策树进行页面块分类

    第一题:采用Knn算法分析letter Recognition Datasets数据集,要求测试集数目至少为50个...

  • 赫夫曼树

    1. 基本介绍: 路径:从一个节点到达其后辈节点的通路,称为路径。通路中分支的数目称为路径长度。上面这棵二叉树,黄...

  • 【算法题】19.好数对的数目

    题目 给你一个整数数组 nums 。 如果一组数字 满足 nums[i] == nums[j] 且 i <...

  • 贝叶斯网络推理

    精确推理算法的鼻祖是联合树。1、计算边缘分布在已知证据节点的条件下,求某一个节点的条件概率。首先要先建立联合树推理...

  • leetcode-买卖股票的最佳时机

    本次分享一道经典的算法题,准确的说是一道题的不同条件下的不同求法。这道题一共有六种情况,每种情况都是不同的解法,在...

  • 算法赏析

    算法题不同思路赏析:例题2,求根节点左右子树的高度差,然后再把根节点的左右子树按根节点的逻辑处理一遍。例题3,这块...

  • 狄克斯特拉算法

    定义 加权图运算图路径最低代价算法。主要包含四个步骤1.找出最便宜的节点,即在最短时间内到达的节点。2.更新该节点...

  • Codeforces 1363C - Game On Leave

    日常一道算法题。 翻译 叶子游戏 Ayush 和 Ashish 在一个有 n 的节点的无根树玩游戏,节点为 1 到...

网友评论

      本文标题:【算法题】2368. 受限条件下可到达节点的数目

      本文链接:https://www.haomeiwen.com/subject/fxqvjdtx.html