Problem p02936 (python)

Fixed submission:
import sys
sys.setrecursionlimit(10**6)

def input():
  return sys.stdin.readline()[:-1]

N, Q=map(int, input().split())

link=[[] for i in range(N+1)]
for i in range(N-1):
  a, b=map(int, input().split())
  link[a].append(b)
  link[b].append(a)

px=[list(map(int, input().split())) for i in range(Q)]

counter=[0]*(N+1)
for i in range(Q):
  counter[px[i][0]]+=px[i][1]
  
def f(num, prenum=-1, p=0):
  counter[num]+=p
  for postnum in link[num]:
    if postnum==prenum:
      continue
    f(postnum, num, counter[num])
    
f(1)

print(*counter[1:])

Buggy submission (verdict: Runtime Error):
N, Q=map(int, input().split())
ab=[list(map(int, input().split())) for i in range(N-1)]
px=[list(map(int, input().split())) for i in range(Q)]

link=[[] for i in range(N+1)]
for i in range(N-1):
  link[ab[i][0]].append(ab[i][1])
  link[ab[i][1]].append(ab[i][0])

counter=[0 for i in range(N+1)]
for i in range(Q):
  counter[px[i][0]]+=px[i][1]
  
def f(num, prenum=-1, p=0):
  counter[num]+=p
  for postnum in link[num]:
    if postnum==prenum:
      continue
    f(postnum, num, counter[num])
    
f(1)

print(*counter[1:])

Buggy submission (verdict: Runtime Error):
N, Q=map(int, input().split())

edge=[list(map(int, input().split())) for i in range(N-1)]
adds=[list(map(int, input().split())) for i in range(Q)]

point=[0]*N

for i in range(Q):
  point[adds[i][1]-1]+=adds[i][0]
  
for i in range(N-1):
  point[edge[i][1]-1]+=point[edge[i][0]-1]
  
print(point)

Buggy submission (verdict: Runtime Error):
import numpy as np

N, Q=map(int, input().split())
matrix=np.zeros((N, N))
point=np.zeros(N)

for i in range(N-1):
  a, b=map(int, input().split())
  matrix[int(a-1), int(b-1)]=1
  
for i in range(Q):
  p, x=map(int, input().split())
  point[int(p-1)]+=x
  
for i in range(N-1):
  point+=matrix[int(i), :]*point[int(i)]
  
print(' '.join(map(str, map(int, point.tolist()))))

Buggy submission (verdict: Runtime Error):
import numpy as np

N, Q=map(int, input().split())
matrix=np.zeros((N, N))
point=np.zeros(N)

for i in range(N-1):
  a, b=map(int, input().split())
  matrix[a-1, b-1]=1
  
for i in range(Q):
  p, x=map(int, input().split())
  point[p-1]+=x
  
for i in range(N-1):
  point+=matrix[i, :]*point[i]
  
print(' '.join(map(str, map(int, point.tolist()))))

Buggy submission (verdict: Runtime Error):
import numpy as np

temp=list(map(int, input().split()))
N=temp[0]
Q=temp[1]

matrix=np.zeros((N, N))
for i in range(N):
  matrix[i, i]=1
  
for i in range(N-1):
  temp=list(map(int, input().split()))
  matrix[temp[0]-1, temp[1]-1]=1
  
counter=np.zeros(N)
for i in range(Q):
  temp=list(map(int, input().split()))
  counter[temp[0]-1]+=temp[1]
  
for i in range(N):
  counter[i]=np.dot(counter, matrix[:, i])
  
print(' '.join(map(str, map(int, counter.tolist()))))

Buggy submission (verdict: Runtime Error):
import numpy as np

temp=list(map(int, input().split()))
N=temp[0]
Q=temp[1]

matrix=np.zeros((N, N), dtype=int)
for i in range(N):
  matrix[i, i]=1

for i in range(N-1):
  temp=list(map(int, input().split()))
  matrix[int(temp[0]-0.5), int(temp[1]-0.5)]=1
  
index=int(N-0.5)
while index>0:
  matrix[index, :]+=np.sum(matrix*matrix[index, :].reshape(-1, 1), axis=0)
  index=int(index-0.5)
  
matrix[0, :]+=np.sum(matrix*matrix[0, :].reshape(-1, 1), axis=0)

matrix[matrix>0]=1

counter=np.zeros(N, dtype=int)
for i in range(Q):
  temp=list(map(int, input().split()))
  counter+=matrix[int(temp[0]-0.5), :]*temp[1]
  
print(' '.join(map(str, map(int, counter.tolist()))))

Buggy submission (verdict: Runtime Error):
import numpy as np

temp=list(map(int, input().split()))
N=temp[0]
Q=temp[1]

matrix=np.zeros((N, N), dtype=int)
for i in range(N):
  matrix[i, i]=1

for i in range(N-1):
  temp=list(map(int, input().split()))
  matrix[int(temp[0]-0.5), int(temp[1]-0.5)]=1
  
index=int(N-0.5)
while index>=0:
  matrix[index, :]+=np.sum(matrix*matrix[index, :].reshape(-1, 1), axis=0)
  index=int(index-0.5)
  
matrix[matrix>0]=1

counter=np.zeros(N, dtype=int)
for i in range(Q):
  temp=list(map(int, input().split()))
  counter+=matrix[int(temp[0]-0.5), :]*temp[1]
  
print(' '.join(map(str, map(int, counter.tolist()))))

Buggy submission (verdict: Runtime Error):
import numpy as np

temp=list(map(int, input().split()))
N=temp[0]
Q=temp[1]

matrix=np.zeros((N, N), dtype=int)
for i in range(N):
  matrix[i, i]=1

for i in range(N-1):
  temp=list(map(int, input().split()))
  matrix[int(temp[0]-0.5), int(temp[1]-0.5)]=1
  
index=int(N-0.5)
while index>=0:
  matrix[index, :]+=np.sum(matrix*matrix[index, :].reshape(-1, 1), axis=0)
  index-=1
  
matrix[matrix>0]=1

counter=np.zeros(N)
for i in range(Q):
  temp=list(map(int, input().split()))
  counter+=matrix[int(temp[0]-0.5), :]*temp[1]
  
print(' '.join(map(str, map(int, counter.tolist()))))

Buggy submission (verdict: Runtime Error):
import numpy as np

temp=list(map(int, input().split()))
N=temp[0]
Q=temp[1]

matrix=np.zeros((N, N))
for i in range(N):
  matrix[i, i]=1

for i in range(N-1):
  temp=list(map(int, input().split()))
  matrix[int(temp[0]-1), int(temp[1]-1)]=1
  
index=int(N-1)
while index>=0:
  matrix[index, :]+=np.sum(matrix*matrix[index, :].reshape(-1, 1), axis=0)
  index-=1
  
matrix[matrix>0]=1

counter=np.zeros(N)
for i in range(Q):
  temp=list(map(int, input().split()))
  counter+=matrix[int(temp[0]-1), :]*temp[1]
  
print(' '.join(map(str, map(int, counter.tolist()))))

Buggy submission (verdict: Runtime Error):
import numpy as np

temp=list(map(int, input().split()))
N=temp[0]
Q=temp[1]

matrix=np.zeros((N, N))
for i in range(N):
  matrix[i, i]=1

for i in range(N-1):
  temp=list(map(int, input().split()))
  matrix[temp[0]-1, temp[1]-1]=1
  
index=N-1
while index>=0:
  matrix[index, :]+=np.sum(matrix*matrix[index, :].reshape(-1, 1), axis=0)
  index-=1
  
matrix[matrix>0]=1

counter=np.zeros(N)
for i in range(Q):
  temp=list(map(int, input().split()))
  counter+=matrix[temp[0]-1, :]*temp[1]
  
print(' '.join(map(str, map(int, counter.tolist()))))