Problem p02850 (python)

Fixed submission:
#幅優先探索
from collections import deque
d=deque()
e=deque()
import sys
sys.setrecursionlimit(1000000000)
n=int(input())
start_node=0
querie=[]
edge={i:[]for i in range(1,n+1)}
for j in range(n-1):
    a,b=map(int,input().split())
    querie.append((a,b))
    edge[a].append(b)
    edge[b].append(a)
    if j==0:
        start_node=a  

d.append((start_node,0))

#はじめにdequeに入れておいて幅優先探索を行うものを決める。そして、塗り分けの結果はans_dictに保存する
#print(edge) 
ans_dict={i:{}for i in range(1,n+1)}
K=0
for some in edge.values():
    K=max(K,len(some))
    
#幅優先探索に使用する すでに訪れたか確認するためのリスト

visited=[False for i in range(n+1)]
visited[start_node]=True

#使う頂点の数を記録した
#nodeをリスト型にしたリストを返す
#start_nodeを記録しておく
def determine_edge_color(node,color):
    new_color=0
    for some in edge[node]:
        if visited[some]==False:
            visited[some]=True
            new_color+=1
            if new_color==color:
                new_color+=1
                ans_dict[node][some]=new_color
                ans_dict[some][node]=new_color
                e.append((some,new_color))
            else:
                ans_dict[node][some]=new_color
                ans_dict[some][node]=new_color
                e.append((some,new_color))
        else:
            pass
            
while d:
    for some in d:
        if visited[some[0]]==True:
            determine_edge_color(some[0],some[1])
        else:
            determine_edge_color(some[1],some[0])
    d=e
    e=deque()
    
#ans_dictの完成
print(K)
#答えの出力
for item in querie:
    a,b=item[0],item[1]
    print(ans_dict[a][b])

Buggy submission (verdict: Runtime Error):
n=int(input())
node={i:[] for i in range(1,n+1)}
visited=[False for i in range(n+1)]
q=[]
for _ in range(n-1):
    a,b=map(int,input().split())
    node[a].append(b)
    node[b].append(a)
    q.append((a,b))

mother=q[0][0]
visited[mother]=True

d=[]
e=[]


paint={i:{} for i in range(1,n+1)}

mother_paint=[0 for i in range(n+1)]

#使う要素 node visited paint mother_paint の四つのリスト
c=1
for child in node[mother]:
    visited[child]=True
    paint[mother][child]=c
    paint[child][mother]=c
    mother_paint[child]=c
    c+=1
    d.append(child)
    
while d:
    for parent in d:
        p=1 
        for child in node[parent]:
            if visited[child]==False:
                visited[child]=True
                if p!=mother_paint[parent]:
                    paint[parent][child]=p
                    paint[child][parent]=p
                    mother_paint[child]=p
                    p+=1
                    e.append(child)
                else:
                    p+=1
                    paint[parent][child]=p
                    paint[child][parent]=p
                    mother_paint[child]=p
                    p+=1
                    e.append(child)
    if e:
        d=e.copy()
        e=[]
    else:
        d=[]
        e=[]
K=0
for i in range(1,n+1):
    K=max(K,len(node[i]))
print(K)
for i in range(n-1):
    s,t=q[i][0],q[i][1]
    print(paint[s][t])

Buggy submission (verdict: Runtime Error):
n=int(input())
edge={i:[] for i in range(n+1)}
colored={j:{} for j in range(n+1)}
before_colored=[0 for i in range(n+1)]
querie=[]
start=[]
for i in range(n-1):
    a,b=map(int,input().split())
    edge[a].append(b)
    edge[b].append(a)
    querie.append((a,b))
    if i==0:
        start=[a,b]
    
    
#どの点とどの点がつながっているかを隣接リスト型にして保管する
from collections import deque
d=deque()
e=deque()
#訪れたリスト
visited=[False for i in range(n+1)]

visited[start[0]]=True
#頂点を一つ登録
color=1
for node in edge[start[0]]:
    visited[node]=True
    d.append(node)
    colored[start[0]][node]=color
    colored[node][start[0]]=color
    before_colored[node]=color
    color+=1
while d:
    for some in d:
        if before_colored[some]!=1:
        	color=1
        else:
            color=2
        for node in edge[some]:
            if visited[node]==False:
                e.append(node)
                visited[node]=True        
                if before_colored[some]!=color:
                    colored[some][node]=color
                    colored[node][some]=color
                    before_colored[node]=color
                    color+=1
                elif before_colored[some]==color:
                    colored[some][node]=color+1
                    colored[node][some]=color+1
                    before_colored[node]=color+1
                    color+=2
        if e:
            d=e
            e=deque()
        else:
            d=deque()
K=0
for i in range(n):
    K=max(K,len(edge[i+1]))
print(K)
for  i in range(n-1):
    a,b=querie[i][0],querie[i][1]
    print(colored[a][b])

Buggy submission (verdict: Runtime Error):
#幅優先探索
from collections import deque
d=deque()
e=deque()
import sys
sys.setrecursionlimit(100000000)
n=int(input())
start_node=0
querie=[]
edge={i:[]for i in range(1,n+1)}
for j in range(n-1):
    a,b=map(int,input().split())
    querie.append((a,b))
    edge[a].append(b)
    edge[b].append(a)
    if j==0:
        start_node=a  
for some in edge[start_node]:
    d.append((start_node,0))

#はじめにdequeに入れておいて幅優先探索を行うものを決める。そして、塗り分けの結果はans_dictに保存する
#print(edge) 
ans_dict={i:{}for i in range(1,n+1)}
K=0
for some in edge.values():
    K=max(K,len(some))
    
#幅優先探索に使用する すでに訪れたか確認するためのリスト

visited=[False for i in range(n+1)]
visited[start_node]=True

#使う頂点の数を記録した
#nodeをリスト型にしたリストを返す
#start_nodeを記録しておく
def determine_edge_color(node,color):
    new_color=0
    for some in edge[node]:
        if visited[some]==False:
            visited[some]=True
            new_color+=1
            if new_color==color:
                new_color+=1
                ans_dict[node][some]=new_color
                e.append((some,new_color))
            else:
                ans_dict[node][some]=new_color
                e.append((some,new_color))
while d:
    for some in d:
        determine_edge_color(some[0],some[1])
    d=e
    e=deque()
#ans_dictの完成
print(K)
#答えの出力
for item in querie:
    a,b=item[0],item[1]
    print(ans_dict[a][b])