forked from adnanaziz/EPIJudge
-
Notifications
You must be signed in to change notification settings - Fork 0
/
is_circuit_wirable.py
51 lines (38 loc) · 1.4 KB
/
is_circuit_wirable.py
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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
import collections
import functools
from typing import List
from test_framework import generic_test
from test_framework.test_utils import enable_executor_hook
class GraphVertex:
def __init__(self) -> None:
self.d = -1
self.edges: List[GraphVertex] = []
def is_any_placement_feasible(graph: List[GraphVertex]) -> bool:
def bfs(s):
s.d = 0
q = collections.deque([s])
while q:
for t in q[0].edges:
if t.d == -1: # Unvisited vertex.
t.d = q[0].d + 1
q.append(t)
elif t.d == q[0].d:
return False
del q[0]
return True
return all(bfs(v) for v in graph if v.d == -1)
@enable_executor_hook
def is_any_placement_feasible_wrapper(executor, k, edges):
if k <= 0:
raise RuntimeError('Invalid k value')
graph = [GraphVertex() for _ in range(k)]
for (fr, to) in edges:
if fr < 0 or fr >= k or to < 0 or to >= k:
raise RuntimeError('Invalid vertex index')
graph[fr].edges.append(graph[to])
return executor.run(functools.partial(is_any_placement_feasible, graph))
if __name__ == '__main__':
exit(
generic_test.generic_test_main('is_circuit_wirable.py',
'is_circuit_wirable.tsv',
is_any_placement_feasible_wrapper))