Package g2301_2400.s2368_reachable_nodes_with_restrictions

Class Solution

  • All Implemented Interfaces:
    
public final class Solution

    2368 - Reachable Nodes With Restrictions.

    Medium

    There is an undirected tree with n nodes labeled from 0 to n - 1 and n - 1 edges.

    You are given a 2D integer array edges of length n - 1 where <code>edgesi = a<sub>i</sub>, b<sub>i</sub></code> indicates that there is an edge between nodes <code>a<sub>i</sub></code> and <code>b<sub>i</sub></code> in the tree. You are also given an integer array restricted which represents restricted nodes.

    Return the maximum number of nodes you can reach from node 0 without visiting a restricted node.

    Note that node 0 will not be a restricted node.

    Example 1:

    Input: n = 7, edges = [0,1,1,2,3,1,4,0,0,5,5,6], restricted = 4,5

    Output: 4

    Explanation: The diagram above shows the tree. We have that 0,1,2,3 are the only nodes that can be reached from node 0 without visiting a restricted node.

    Example 2:

    Input: n = 7, edges = [0,1,0,2,0,5,0,4,3,2,6,5], restricted = 4,2,1

    Output: 3

    Explanation: The diagram above shows the tree. We have that 0,5,6 are the only nodes that can be reached from node 0 without visiting a restricted node.

    Constraints:

    • <code>2 <= n <= 10<sup>5</sup></code>

    • edges.length == n - 1

    • edges[i].length == 2

    • <code>0 <= a<sub>i</sub>, b<sub>i</sub>< n</code>

    • <code>a<sub>i</sub> != b<sub>i</sub></code>

    • edges represents a valid tree.

    • 1 &lt;= restricted.length &lt; n

    • 1 &lt;= restricted[i] &lt; n

    • All the values of restricted are unique.

    • Nested Class Summary

      Nested Classes 
      Modifier and Type Class Description

    • Field Summary

      Fields 
      Modifier and Type Field Description

    • Constructor Summary

      Constructors 
      Constructor Description
      Solution()

    • Enum Constant Summary

      Enum Constants 
      Enum Constant Description

    • Method Summary

      Modifier and Type Method Description
      final Integer reachableNodes(Integer n, Array<IntArray> edges, IntArray restricted)

      • Methods inherited from class java.lang.Object

        clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait