Problem p03181 (python)

Fixed submission:
# coding: utf-8
# Your code here!
import sys
input = sys.stdin.readline
sys.setrecursionlimit(10**7)

n,MOD = [int(i) for i in input().split()]

to = [[] for _ in [0]*n]
for _ in range(n-1):
    x,y = [int(i)-1 for i in input().split()]
    to[x].append(y)
    to[y].append(x)


dp = [[0]*len(to_i) for to_i in to]
parent_index = [-9999]*n 
ans = [0]*n

def dfs(v, p):
    res = 1
    for i, vc in enumerate(to[v]):
        if vc == p:
            parent_index[v] = i
            continue
        dp[v][i] = dfs(vc, v)
        res *= dp[v][i]+1
        res %= MOD
    return res

def dfs2(v, res_p, p):
    deg = len(to[v])
    if p >= 0:
        dp[v][parent_index[v]] = res_p

    accl = [1]*(deg+1)
    accr = [1]*(deg+1)

    for i in range(1, deg+1):
        accl[i] = accl[i-1] * (dp[v][i-1] + 1) % MOD
        accr[-i-1] = accr[-i] * (dp[v][-i] + 1) % MOD

    ans[v] = accr[0]
    for i,vc in enumerate(to[v]):
        if vc == p: continue
        dfs2(vc, accl[i]*accr[i+1]%MOD, v)

dfs(0,-1)
#print(dp)
dfs2(0,0,-1)
#print(dp)
for i in ans:
    print(i)

Buggy submission (verdict: Runtime Error):
# coding: utf-8
# Your code here!
import sys
sys.setrecursionlimit(10**7)

n,MOD = [int(i) for i in input().split()]

to = [[] for _ in [0]*n]
for _ in range(n-1):
    x,y = [int(i)-1 for i in input().split()]
    to[x].append(y)
    to[y].append(x)


dp = [[0]*len(to_i) for to_i in to]
parent_index = [-1]*n 
ans = [0]*n

def dfs(v, p):
    res = 1
    for i, vc in enumerate(to[v]):
        if vc == p:
            parent_index[vc] = i
            continue
        """子供から情報をもらって処理する"""
        dp[v][i] = dfs(vc, v)
        res *= dp[v][i]+1
        res %= MOD
        """処理ここまで"""
    return res

def bfs(v, res_p, p):
    deg = len(to[v])
    if p >= 0:
        dp[v][parent_index[v]] = res_p

    accl = [1]*(deg+1)
    for i, dpvi in enumerate(dp[v]):
        accl[i+1] = accl[i]*(dpvi+1)%MOD

    accr = [1]*(deg+1)
    for i, dpvir in enumerate(reversed(dp[v])):
        accr[-i-2] = accr[-i-1]*(dpvir+1)%MOD

    ans[v] = accr[0]
    for i,vc in enumerate(to[v]):
        if vc == p: continue
        bfs(vc, accl[i]*accr[i+1]%MOD, v)

dfs(0,-1)
#print(dp)
bfs(0,0,-1)
#print(dp)
for i in ans:
    print(i)

Buggy submission (verdict: Runtime Error):
# coding: utf-8
# Your code here!
import sys
sys.setrecursionlimit(10**7)

n,MOD = [int(i) for i in input().split()]

to = [[] for i in [0]*n]
for _ in range(n-1):
    x,y = [int(i)-1 for i in input().split()]
    to[x].append(y)
    to[y].append(x)


dp = [[0]*len(to_i) for to_i in to]
parent_index = [-1]*n 
ans = [0]*n

def dfs(v, p):
    res = 1
    for i, vc in enumerate(to[v]):
        if vc == p:
            parent_index[vc] = i
            continue
        """子供から情報をもらって処理する"""
        dp[v][i] = dfs(vc, v)
        res *= 1+dp[v][i]
        res %= MOD
        """処理ここまで"""
    return res

def bfs(v, res_p, p):
    deg = len(to[v])
    if p >= 0:
        dp[v][parent_index[v]] = res_p

    accl = [1]*(deg+1)
    for i in range(deg):
        accl[i+1] = accl[i]*(dp[v][i]+1)%MOD

    accr = [1]*(deg+1)
    for i in range(deg-1,-1,-1):
        accr[i] = accr[i+1]*(dp[v][i]+1)%MOD

    ans[v] = accr[0]
    for i,vc in enumerate(to[v]):
        if vc == p: continue
        bfs(vc, accl[i]*accr[i+1]%MOD, v)

dfs(0,-1)
#print(dp)
bfs(0,None,-1)
#print(dp)
for i in ans:
    print(i)

Buggy submission (verdict: Runtime Error):
# coding: utf-8
# Your code here!
import sys
sys.setrecursionlimit(10**7)

n,MOD = [int(i) for i in input().split()]

to = [[] for i in [0]*n]
for _ in range(n-1):
    x,y = [int(i)-1 for i in input().split()]
    to[x].append(y)
    to[y].append(x)


dp = [[0]*len(to_i) for to_i in to]
parent_index = [-1]*n 
ans = [0]*n

def dfs(v, p):
    """葉の場合（必要なら）"""
#    if p >= 0 and len(to[v]) == 1: 
#        return 1

    res = 1
    for i, child in enumerate(to[v]):
        if child == p:
            parent_index[child] = i
            continue
        """子供から情報をもらって処理する"""
        dp[v][i] = dfs(child, v)
        res *= 1+dp[v][i]
        res %= MOD
        """処理ここまで"""
    return res

def bfs(v, res_p, p):
    deg = len(to[v])
    if p >= 0:
        dp[v][parent_index[v]] = res_p

    dpl = [1]*(deg+1)
    for i in range(deg):
        dpl[i+1] = dpl[i]*(dp[v][i]+1)%MOD

    dpr = [1]*(deg+1)
    for i in range(deg-1,-1,-1):
        dpr[i] = dpr[i+1]*(dp[v][i]+1)%MOD

    ans[v] = dpr[0]
    
    for child in to[v]:
        if child == p: continue
        bfs(child, dpl[i]*dpr[i+1]%MOD, v)

dfs(0,-1)
#print(dp)
bfs(0,None,-1)
#print(dp)
for i in ans:
    print(i)

Buggy submission (verdict: Runtime Error):
# coding: utf-8
# Your code here!
import sys
sys.setrecursionlimit(10**7)

n,MOD = [int(i) for i in input().split()]

to = [[] for i in [0]*n]
for _ in range(n-1):
    x,y = [int(i)-1 for i in input().split()]
    to[x].append(y)
    to[y].append(x)


dp = [[0]*len(to_i) for to_i in to]
parent_index = [-1]*n 
ans = [0]*n

def dfs(v, p):
    """葉の場合（必要なら）"""
#    if p >= 0 and len(to[v]) == 1: 
#        return 1

    res = 1
    for i, child in enumerate(to[v]):
        if child == p:
            parent_index[child] = i
            continue
        """子供から情報をもらって処理する"""
        dp[v][i] = dfs(child, v)
        res *= 1+dp[v][i]
        res %= MOD
        """処理ここまで"""
    return res

def bfs(v, res_p, p):
    deg = len(to[v])
    if p >= 0:
        dp[v][parent_index[v]] = res_p

    dpl = [1]*(deg+1)
    for i in range(deg):
        dpl[i+1] = dpl[i]*(dp[v][i]+1)%MOD

    dpr = [1]*(deg+1)
    for i in range(deg-1,-1,-1):
        dpr[i] = dpr[i+1]*(dp[v][i]+1)%MOD

    ans[v] = dpr[0]
    
    for child in to[v]:
        if child == p: continue
        bfs(child, dpl[i]*dpr[i+1]%MOD, v)

dfs(0,-1)
#print(dp)
bfs(0,None,-1)
#print(dp)
for i in ans:
    print(i)