Class Solution
-
- All Implemented Interfaces:
public final class Solution1916 - Count Ways to Build Rooms in an Ant Colony.
Hard
You are an ant tasked with adding
nnew rooms numbered0ton-1to your colony. You are given the expansion plan as a 0-indexed integer array of lengthn,prevRoom, whereprevRoom[i]indicates that you must build roomprevRoom[i]before building roomi, and these two rooms must be connected directly. Room0is already built, soprevRoom[0] = -1. The expansion plan is given such that once all the rooms are built, every room will be reachable from room0.You can only build one room at a time, and you can travel freely between rooms you have already built only if they are connected. You can choose to build any room as long as its previous room is already built.
Return the number of different orders you can build all the rooms in. Since the answer may be large, return it modulo <code>10<sup>9</sup> + 7</code>.
Example 1:
Input: prevRoom = -1,0,1
Output: 1
Explanation: There is only one way to build the additional rooms: 0 → 1 → 2
Example 2:
Input: prevRoom = -1,0,0,1,2
Output: 6
Explanation: The 6 ways are:
0 → 1 → 3 → 2 → 4
0 → 2 → 4 → 1 → 3
0 → 1 → 2 → 3 → 4
0 → 1 → 2 → 4 → 3
0 → 2 → 1 → 3 → 4
0 → 2 → 1 → 4 → 3
Constraints:
n == prevRoom.length<code>2 <= n <= 10<sup>5</sup></code>
prevRoom[0] == -10 <= prevRoom[i] < nfor all1 <= i < nEvery room is reachable from room
0once all the rooms are built.
-
-
Constructor Summary
Constructors Constructor Description Solution()
-
Method Summary
Modifier and Type Method Description final IntegerwaysToBuildRooms(IntArray prevRoom)-
-
Method Detail
-
waysToBuildRooms
final Integer waysToBuildRooms(IntArray prevRoom)
-
-
-
-