Skip to content

Commit f1fe6d9

Browse files
authored
Update README.md
1 parent d9d4ef1 commit f1fe6d9

File tree

1 file changed

+25
-25
lines changed
  • solution/1900-1999/1916.Count Ways to Build Rooms in an Ant Colony

1 file changed

+25
-25
lines changed

solution/1900-1999/1916.Count Ways to Build Rooms in an Ant Colony/README.md

+25-25
Original file line numberDiff line numberDiff line change
@@ -81,31 +81,31 @@ tags:
8181

8282
```python
8383
class Solution:
84-
def waysToBuildRooms(self, prevRoom: List[int]) -> int:
85-
modulo = 10 ** 9 + 7
86-
ingoing = defaultdict(set)
87-
outgoing = defaultdict(set)
88-
89-
for i in range(1, len(prevRoom)):
90-
ingoing[i].add(prevRoom[i])
91-
outgoing[prevRoom[i]].add(i)
92-
ans = [1]
93-
94-
def recurse(i):
95-
if len(outgoing[i]) == 0:
96-
return 1 # just self
97-
98-
nodes_in_tree = 0
99-
for v in outgoing[i]:
100-
cn = recurse(v)
101-
if nodes_in_tree != 0:
102-
ans[0] *= comb(nodes_in_tree + cn, cn)
103-
ans[0] %= modulo
104-
nodes_in_tree += cn
105-
return nodes_in_tree + 1
106-
107-
recurse(0)
108-
return ans[0] % modulo
84+
def waysToBuildRooms(self, prevRoom: List[int]) -> int:
85+
modulo = 10**9 + 7
86+
ingoing = defaultdict(set)
87+
outgoing = defaultdict(set)
88+
89+
for i in range(1, len(prevRoom)):
90+
ingoing[i].add(prevRoom[i])
91+
outgoing[prevRoom[i]].add(i)
92+
ans = [1]
93+
94+
def recurse(i):
95+
if len(outgoing[i]) == 0:
96+
return 1
97+
98+
nodes_in_tree = 0
99+
for v in outgoing[i]:
100+
cn = recurse(v)
101+
if nodes_in_tree != 0:
102+
ans[0] *= comb(nodes_in_tree + cn, cn)
103+
ans[0] %= modulo
104+
nodes_in_tree += cn
105+
return nodes_in_tree + 1
106+
107+
recurse(0)
108+
return ans[0] % modulo
109109
```
110110

111111
#### Java

0 commit comments

Comments
 (0)