-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy path22.py
More file actions
33 lines (27 loc) · 948 Bytes
/
22.py
File metadata and controls
33 lines (27 loc) · 948 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
class Solution:
def minNumberOfSemesters(self, n: int, deps: List[List[int]], k: int) -> int:
pre = [0] * n
for u, v in deps:
pre[v - 1] |= 1 << (u - 1)
dp = [-1] * (1 << n)
def dfs(mask):
if mask == (1 << n) - 1:
return 0
if dp[mask] != -1:
return dp[mask]
ready = 0
for i in range(n):
if (mask >> i) & 1 == 0 and (pre[i] & mask) == pre[i]:
ready |= 1 << i
if bin(ready).count("1") <= k:
dp[mask] = 1 + dfs(mask | ready)
return dp[mask]
best = float("inf")
sub = ready
while sub:
if bin(sub).count("1") == k:
best = min(best, 1 + dfs(mask | sub))
sub = (sub - 1) & ready
dp[mask] = best
return best
return dfs(0)