Problem p02871 (python)

Fixed submission:
import math

[v,e]=list(map(int,input().split()))
chizu=[]
for i in range(e):
	chizu.append(list(map(int,input().split())))
tmax=int(input())
info=[]
for i in range(tmax):
	kari=[]
	nnew=int(input())
	kari.append(nnew)
	for j in range(nnew):
		kari.append(list(map(int,input().split())))
	info.append(kari)
#入力受け取り完了

chizukai=[[0 for i in range(v+1)]for j in range(v+1)]
for i in range(e):
	chizukai[chizu[i][0]][chizu[i][1]]=chizu[i][2]
	chizukai[chizu[i][1]][chizu[i][0]]=chizu[i][2]
#地図完成

#ダイクストラ法により、あるノードからあるノードへの最短ルートを求める。
def dijkstra(s,g,chizukai):
	import math
	distance=[float("inf") for i in range(v+1)] #スタートから各ノードへの最短距離のリスト。最初は全部infで定義。
	distance[s]=0 #スタート地点への距離は0。
	previous_nodes=[-1 for i in range(v+1)] #ひとつ前に到達するノードのリスト。
	previous_nodes[0]=float("inf") #ここは定義できないのでとりあえず無限大
	unsearched_nodes=list(range(1,v+1)) #未探索ノードの一覧。確定したのは消していく。
	unsearched_nodes.remove(s) #スタートは確定しているので除く。
	min_index=s #distanceリスト作るために初期値はスタートに設定。
	mae_index=s #最初に確定するノードの前に到達するノードはs。
	
	while len(unsearched_nodes)!=0: #未探索ノードがある限り繰り返す。
		for i in range(v+1): #distanceリストを作っていく。
			if chizukai[min_index][i]!=0:
				if distance[i]>distance[min_index]+chizukai[min_index][i]: #distanceリスト更新
					distance[i]=distance[min_index]+chizukai[min_index][i]
					previous_nodes[i]=min_index #previous_nodesも更新
		mini=float("inf") #最小値を探す。初期値はinf。
		for i in unsearched_nodes: #未探索ノードでforループ。最短距離の点を確定させていく。
			if mini>distance[i]:
				mini=distance[i] #未確定の中の最小値を探す。
				min_index=i #最小値のインデックスを保存。
		unsearched_nodes.remove(min_index) #確定したので消す。
		mae_index=min_index
	
	routelist=[g]
	ima=g
	while True:
		routelist.append(previous_nodes[ima])
		ima=previous_nodes[ima]
		if ima==s:
			break
	routelist.reverse() #道順完成
	kyorilist=[]
	for i in range(len(routelist)-1):
		kyorilist.append(chizukai[routelist[i]][routelist[i+1]])
	
	return [routelist,kyorilist] 
	
	
shirushi=0 #stayならば1にする
maenode=1 #前にいたノード
tuginode=-1 #次に行くノード
kyori=0 #前にいたノードからの距離
tansakuzumi=0 #ルート探索済みかどうか
nidai=[] #目的地に行くだけで配達が終了する配達先
jimusho=[] #一度事務所へ戻らないといけない配達先
sindo=1 #次に行くノードはルートリストのどこか

for i in range(tmax):
	shirushi=0
	#とりあえず荷物渡せるなら渡す
	if kyori==0: #もしノード上にいるなら荷台リストからそのノードは消す。
		for j in range(nidai.count(maenode)):
			nidai.remove(maenode)

	#では注文の処理
	if maenode==1 and kyori==0: #事務所にいるならば
		for j in range(len(jimusho)): #まず事務所リストを積む
			nidai.append(jimusho[j])
		jimusho.clear() #事務所リストは空に
		for j in range(info[i][0]): #そのとき注文入ったものも積む
			nidai.append(info[i][j+1][1])
	else: #事務所にいなければ
		for j in range(info[i][0]): #入った注文は事務所リストに
			jimusho.append(info[i][j+1][1])
	
	#移動先を決める
	if tansakuzumi==0: #ルート探索する必要がある場合
		if len(nidai)!=0: #荷台に荷物があるならば
			[r,k]=dijkstra(maenode,nidai[0],chizukai)
			tuginode=r[sindo]
			tansakuzumi=1
		else: #荷台が空っぽならば
			if maenode!=1: #事務所じゃないなら事務所に戻る
				[r,k]=dijkstra(maenode,1,chizukai)
				tuginode=r[sindo]
				tansakuzumi=1
			else: #事務所戻ったけど注文来てないなら待機
				print(-1)
				shirushi=1

	#出力	
	if shirushi==0:
		print(tuginode)
	
	#次に向けて移動
	if shirushi==0: #stayじゃないなら
		kyori+=1
		if kyori==k[sindo-1]: #その移動によりノードに到着するなら
			if tuginode==r[-1]: #そこが目的地ならば
				kyori=0
				maenode=tuginode
				tuginode=0 #つぎのノードはまだわからない
				tansakuzumi=0
				sindo=1
			else: #そこが通過するノードならば	
				kyori=0
				maenode=tuginode
				tuginode=r[sindo+1]
				sindo+=1

Buggy submission (verdict: Runtime Error):
import math

[v,e]=list(map(int,input().split()))
chizu=[]
for i in range(e):
	chizu.append(list(map(int,input().split())))
tmax=int(input())
info=[]
for i in range(tmax):
	kari=[]
	nnew=int(input())
	kari.append(nnew)
	for j in range(nnew):
		kari.append(list(map(int,input().split())))
	info.append(kari)
#入力受け取り完了

chizukai=[[0 for i in range(v+1)]for j in range(v+1)]
for i in range(e):
	chizukai[chizu[i][0]][chizu[i][1]]=chizu[i][2]
	chizukai[chizu[i][1]][chizu[i][0]]=chizu[i][2]
#地図完成

#ダイクストラ法により、あるノードからあるノードへの最短ルートを求める。
def dijkstra(s,g,chizukai):
	import math
	distance=[float("inf") for i in range(v+1)] #スタートから各ノードへの最短距離のリスト。最初は全部infで定義。
	distance[s]=0 #スタート地点への距離は0。
	previous_nodes=[-1 for i in range(v+1)] #ひとつ前に到達するノードのリスト。
	previous_nodes[0]=float("inf") #ここは定義できないのでとりあえず無限大
	unsearched_nodes=list(range(1,v+1)) #未探索ノードの一覧。確定したのは消していく。
	unsearched_nodes.remove(s) #スタートは確定しているので除く。
	min_index=s #distanceリスト作るために初期値はスタートに設定。
	mae_index=s #最初に確定するノードの前に到達するノードはs。
	
	while len(unsearched_nodes)!=0: #未探索ノードがある限り繰り返す。
		for i in range(v+1): #distanceリストを作っていく。
			if chizukai[min_index][i]!=0:
				if distance[i]>distance[min_index]+chizukai[min_index][i]: #distanceリスト更新
					distance[i]=distance[min_index]+chizukai[min_index][i]
					previous_nodes[i]=min_index #previous_nodesも更新
		mini=float("inf") #最小値を探す。初期値はinf。
		for i in unsearched_nodes: #未探索ノードでforループ。最短距離の点を確定させていく。
			if mini>distance[i]:
				mini=distance[i] #未確定の中の最小値を探す。
				min_index=i #最小値のインデックスを保存。
		unsearched_nodes.remove(min_index) #確定したので消す。
		mae_index=min_index
	
	routelist=[g]
	ima=g
	while True:
		routelist.append(previous_nodes[ima])
		ima=previous_nodes[ima]
		if ima==s:
			break
	routelist.reverse() #道順完成
	kyorilist=[]
	for i in range(len(routelist)-1):
		kyorilist.append(chizukai[routelist[i]][routelist[i+1]])
	
	return [routelist,kyorilist] 
	
	
shirushi=0 #答え出力するか
maenode=1 #前にいたノード
tuginode=-1 #次に行くノード
kyori=0 #前にいたノードからの距離
tansakuzumi=0 #ルート探索済みかどうか
nidai=[] #目的地に行くだけで配達が終了する配達先
jimusho=[] #一度事務所へ戻らないといけない配達先
sindo=1 #次に行くノードはルートリストのどこか

for i in range(tmax):
	#とりあえず荷物渡せるなら渡す
	if kyori==0: #もしノード上にいるなら荷台リストからそのノードは消す。
		for j in range(nidai.count(maenode)):
			nidai.remove(maenode)

	#では注文の処理
	if maenode==1 and kyori==0: #事務所にいるならば
		for j in range(len(jimusho)): #まず事務所リストを積む
			nidai.append(jimusho[j])
		jimusho.clear() #事務所リストは空に
		for j in range(info[i][0]): #そのとき注文入ったものも積む
			nidai.append(info[i][j+1][1])
	else: #事務所にいなければ
		for j in range(info[i][0]): #入った注文は事務所リストに
			jimusho.append(info[i][j+1][1])
	
	#移動先を決める
	if tansakuzumi==0: #ルート探索する必要がある場合
		if len(nidai)!=0: #荷台に荷物があるならば
			[r,k]=dijkstra(maenode,nidai[0],chizukai)
			tuginode=r[sindo]
			tansakuzumi=1
		else: #荷台が空っぽならば
			if maenode!=1: #事務所じゃないなら事務所に戻る
				[r,k]=dijkstra(maenode,1,chizukai)
				tuginode=r[sindo]
				tansakuzumi=1
			else: #事務所戻ったけど注文来てないなら待機
				print(-1)
				shirushi=1

	#出力	
	if shirushi==0:
		print(tuginode)
	shirushi=0
	
	#次に向けて移動
	kyori+=1
	if kyori==k[sindo-1]: #その移動によりノードに到着するなら
		if tuginode==r[-1]: #そこが目的地ならば
			kyori=0
			maenode=tuginode
			tuginode=0 #つぎのノードはまだわからない
			tansakuzumi=0
			sindo=1
		else: #そこが通過するノードならば	
			kyori=0
			maenode=tuginode
			tuginode=r[sindo+1]
			sindo+=1

Buggy submission (verdict: Runtime Error):
import math

[v,e]=list(map(int,input().split()))
chizu=[]
for i in range(e):
	chizu.append(list(map(int,input().split())))
tmax=int(input())
info=[]
for i in range(tmax):
	kari=[]
	nnew=int(input())
	kari.append(nnew)
	for j in range(nnew):
		kari.append(list(map(int,input().split())))
	info.append(kari)
#入力受け取り完了

chizukai=[[0 for i in range(v+1)]for j in range(v+1)]
for i in range(e):
	chizukai[chizu[i][0]][chizu[i][1]]=chizu[i][2]
	chizukai[chizu[i][1]][chizu[i][0]]=chizu[i][2]
#地図完成

#ダイクストラ法により、あるノードからあるノードへの最短ルートを求める。
def dijkstra(s,g,chizukai):
	import math
	distance=[float("inf") for i in range(v+1)] #スタートから各ノードへの最短距離のリスト。最初は全部infで定義。
	distance[s]=0 #スタート地点への距離は0。
	previous_nodes=[-1 for i in range(v+1)] #ひとつ前に到達するノードのリスト。
	previous_nodes[0]=float("inf") #ここは定義できないのでとりあえず無限大
	unsearched_nodes=list(range(1,v+1)) #未探索ノードの一覧。確定したのは消していく。
	unsearched_nodes.remove(s) #スタートは確定しているので除く。
	min_index=s #distanceリスト作るために初期値はスタートに設定。
	mae_index=s #最初に確定するノードの前に到達するノードはs。
	
	while len(unsearched_nodes)!=0: #未探索ノードがある限り繰り返す。
		for i in range(v+1): #distanceリストを作っていく。
			if chizukai[min_index][i]!=0:
				if distance[i]>distance[min_index]+chizukai[min_index][i]: #distanceリスト更新
					distance[i]=distance[min_index]+chizukai[min_index][i]
					previous_nodes[i]=min_index #previous_nodesも更新
		mini=float("inf") #最小値を探す。初期値はinf。
		for i in unsearched_nodes: #未探索ノードでforループ。最短距離の点を確定させていく。
			if mini>distance[i]:
				mini=distance[i] #未確定の中の最小値を探す。
				min_index=i #最小値のインデックスを保存。
		unsearched_nodes.remove(min_index) #確定したので消す。
		mae_index=min_index
	
	routelist=[g]
	ima=g
	while True:
		routelist.append(previous_nodes[ima])
		ima=previous_nodes[ima]
		if ima==s:
			break
	routelist.reverse() #道順完成
	kyorilist=[]
	for i in range(len(routelist)-1):
		kyorilist.append(chizukai[routelist[i]][routelist[i+1]])
	
	return [routelist,kyorilist] 
	
	

maenode=1 #前にいたノード
tuginode=-1 #次に行くノード
kyori=0 #前にいたノードからの距離
tansakuzumi=0 #ルート探索済みかどうか
nidai=[] #目的地に行くだけで配達が終了する配達先
jimusho=[] #一度事務所へ戻らないといけない配達先
sindo=1 #次に行くノードはルートリストのどこか

for i in range(tmax):
	#とりあえず荷物渡せるなら渡す
	if kyori==0: #もしノード上にいるなら荷台リストからそのノードは消す。
		for j in range(len(nidai)):
			if nidai[j]==maenode:
				nidai.remove(maenode)

	#では注文の処理
	if maenode==1 and kyori==0: #事務所にいるならば
		for j in range(len(jimusho)): #まず事務所リストを積む
			nidai.append(jimusho[j])
		jimusho.clear() #事務所リストは空に
		for j in range(info[i][0]): #そのとき注文入ったものも積む
			nidai.append(info[i][j+1][1])
	else: #事務所にいなければ
		for j in range(info[i][0]): #入った注文は事務所リストに
			jimusho.append(info[i][j+1][1])
	
	#移動先を決める
	if tansakuzumi==0: #ルート探索する必要がある場合
		if len(nidai)!=0: #荷台に荷物があるならば
			[r,k]=dijkstra(maenode,nidai[0],chizukai)
			tuginode=r[sindo]
			tansakuzumi=1
		else: #荷台が空っぽならば
			[r,k]=dijkstra(maenode,1,chizukai)
			tuginode=r[sindo]
			tansakuzumi=1

	#出力		
	print(tuginode)
	
	#次に向けて移動
	kyori+=1
	if kyori==k[sindo-1]: #その移動によりノードに到着するなら
		if tuginode==r[-1]: #そこが目的地ならば
			kyori=0
			maenode=tuginode
			tuginode=0 #つぎのノードはまだわからない
			tansakuzumi=0
			sindo=1
		else: #そこが通過するノードならば	
			kyori=0
			maenode=tuginode
			tuginode=r[sindo+1]
			sindo+=1