问题是:
有一条铁路从一个城镇到不同的城镇。由于预算问题,铁路被认为是一种方式。如果有A市,B市和C市之间有铁路连接,则A不一定有直达B和C的铁路。从A到C的路线可以是A到B,然后B到C . 上面的场景用图表来描述。节点是城镇,边是它们之间的距离。
给出的输入将是一个图表,也是通往城镇的路线。输出必须是城镇之间的总铁路距离,如果不存在路线,则必须说“不存在路线”。
Input: AB5, BC2, CD3, BE4
Input: A-B-E
Input: A-C-E
Output: 9
Output: No Route Exists
我的代码是:
print "Welcome to the total path calculation program"
n=1
inp=1
graph=dict()
while(n==1):
print "Enter the connection:"
x=raw_input()
new={x[0]:{x[1]:x[2]}}
graph.update(new)
print "Do you want to enter another connection?"
y=raw_input()
if y=='y':
n=1
else:
n=0
while(inp):
print "Now enter the connection to find the total cost:"
x=raw_input()
try:
t=int(graph[x[0]][x[2]])+int(graph[x[2]][x[4]])
print "The total cost is %d" %(t)
except KeyError:
print "No route exists"
print "Do you want to find cost for more connections?"
x=raw_input()
if x=='y':
inp=1
else:
inp=0