Sadonkorjuu

from mimetypes import init
import sys, math
import asyncio
import warnings
#import datetime
#start_time = datetime.datetime.now()
f = open('200k.txt', 'r')
CityCount = int(f.readline())
CityStates = f.readline().split()
RoadDict = {}
[RoadDict.setdefault(x, []) for x in range(0, CityCount)]
Ports = []
warnings.filterwarnings("ignore")
def connection(node, target):
    for con in RoadDict[node]:
        if(int(con[0]) == target):
            return int(con[1])
    return 0
 
class Node():
    def __init__(self, nodeVal) -> None:
        self.value = nodeVal
        self.distance = math.inf
 
    def addDistance(self, distance):
        if(distance < self.distance):
            self.distance = distance
 
def background(f):
    def wrapped(*args, **kwargs):
        return asyncio.get_event_loop().run_in_executor(None, f, *args, **kwargs)
        pass
    return wrapped
    pass
 
@background
def ArrayConstructor(temp):
    global RoadDict
    global Ports
    global CityStates
    RoadDict[temp[0]-1].append([temp[1]-1,temp[2]])
    RoadDict[temp[1]-1].append([temp[0]-1, temp[2]])
#    RoadList.append(temp)
    if not int(CityStates[temp[0]-1]):
        Ports.append(temp[0]-1)
    if not int(CityStates[temp[1]-1]):
        Ports.append(temp[1]-1)
 
def Before():
    #Build a nested array of roads
    global TotalArray 
    TotalArray = []
    for n in range(0,CityCount-1):
        TotalArray += [list(map(int, f.readline().split()))]   
def After():
    global NodeList
    global RoadDict
    global CityCount
    global CityStates
    global Ports
    #Initialize nodes:
    NodeList = []
    for i in range(0, CityCount):
        NodeList.append(Node(i))
        if not int(CityStates[i]):
            NodeList[i].addDistance(0)
 
    PortNodes = []
    for i in range(0, len(Ports)):
        PortNodes.append(NodeList[Ports[i]])
    sum = 0
 
    for startingNode in PortNodes:
        node = startingNode
        startingNodevalue = startingNode.value
        travelledNodes = [False]*CityCount
        Queue = []
        Queue.append(startingNode)
        travelledNodes[startingNodevalue] = True
 
        while Queue:
            node = Queue[0]
            Queue.pop(0)
            nodeValue = node.value 
            for neighbor in RoadDict[nodeValue]:
                neighborNode = NodeList[neighbor[0]]
                if(CityStates[neighbor[0]] == "0"):
                    travelledNodes[neighbor[0]] = True
 
                elif(not travelledNodes[neighbor[0]]):
                    travelledNodes[neighbor[0]] = True
                    Queue.append(neighborNode)
                    neighborNode.addDistance(connection(nodeValue, neighborNode.value)+node.distance)
 
    for node in NodeList:
        sum += node.distance
    print(sum)
 
Before()
#print(datetime.datetime.now() - start_time)
loop = asyncio.get_event_loop()
looper = asyncio.gather(*[ArrayConstructor(i) for i in TotalArray])
results = loop.run_until_complete(looper)
#print(datetime.datetime.now() - start_time)
After()
#print(datetime.datetime.now() - start_time)

Test 1

Group: 1, 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 7, in <module>
    f = ope...

Test 2

Group: 1, 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 7, in <module>
    f = ope...

Test 3

Group: 1, 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 7, in <module>
    f = ope...

Test 4

Group: 1, 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 7, in <module>
    f = ope...

Test 5

Group: 1, 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 7, in <module>
    f = ope...

Test 6

Group: 1, 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 7, in <module>
    f = ope...

Test 7

Group: 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 7, in <module>
    f = ope...

Test 8

Group: 1, 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 7, in <module>
    f = ope...

Test 9

Group: 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 7, in <module>
    f = ope...

Test 10

Group: 1, 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 7, in <module>
    f = ope...

Test 11

Group: 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 7, in <module>
    f = ope...

Test 12

Group: 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 7, in <module>
    f = ope...

Test 13

Group: 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 7, in <module>
    f = ope...

Test 14

Group: 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 7, in <module>
    f = ope...

Test 15

Group: 1, 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 7, in <module>
    f = ope...

Test 16

Group: 1, 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 7, in <module>
    f = ope...

Test 17

Group: 1, 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 7, in <module>
    f = ope...

Test 18

Group: 1, 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 7, in <module>
    f = ope...

Test 19

Group: 1, 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 7, in <module>
    f = ope...

Test 20

Group: 1, 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 7, in <module>
    f = ope...

Test 21

Group: 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 7, in <module>
    f = ope...

Test 22

Group: 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 7, in <module>
    f = ope...

Test 23

Group: 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 7, in <module>
    f = ope...

Test 24

Group: 1, 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 7, in <module>
    f = ope...

Test 25

Group: 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 7, in <module>
    f = ope...

Test 26

Group: 1, 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 7, in <module>
    f = ope...

Test 27

Group: 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 7, in <module>
    f = ope...

Test 28

Group: 1, 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 7, in <module>
    f = ope...

Test 29

Group: 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 7, in <module>
    f = ope...

Test 30

Group: 1, 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 7, in <module>
    f = ope...

Test 31

Group: 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 7, in <module>
    f = ope...