路径问题
小蓝学习了最短路径之后特别高兴,他定义了一个特别的图,希望找到图 中的最短路径。
小蓝的图由 2021 个结点组成,依次编号 1 至 2021。
对于两个不同的结点 a, b,如果 a 和 b 的差的绝对值大于 21,则两个结点 之间没有边相连;如果 a 和 b 的差的绝对值小于等于 21,则两个点之间有一条 长度为 a 和 b 的最小公倍数的无向边相连。
例如:结点 1 和结点 23 之间没有边相连;结点 3 和结点 24 之间有一条无 向边,长度为 24;结点 15 和结点 25 之间有一条无向边,长度为 75。
请计算,结点 1 和结点 2021 之间的最短路径长度是多少。
import os
import sys
# 请在此输入您的代码
import math
def gbs(a,b):
return a*b//math.gcd(a,b)
a=[0]*2022
for i in range(1,2021):
for j in range(i+1,i+22):
if j>2021:
break
if a[j]==0:
a[j]=a[i]+gbs(i,j)
else:
a[j]=min(a[j],a[i]+gbs(i,j))
print(a[2021])
思路:
如何快速求最小公倍数(利用最小公倍数和最大公约数的关系)
设M,N为A,B的最小公倍数和最大公约数,则有A*B == M*N
进而得出最小公倍数M == A*B//N
最大公约数N可以使用辗转相除法(欧几里得算法)来快速求解。具体公式为:
设A除以B的余数为X(A B X都为正整数)
gcd(A,B)=gcd(B,X)
递归使用gcd()直到括号内第二位为零(余数为零),此时括号的第一位就是A , B的最大公约数,以求32,12的最小公约数“4”来举个例子:
gcd(32,12)=gcd(12,8)=gcd(8,4)=gcd(4,0)
至此,现在已经解决了最小公倍数,即题目中无向图相连节点之间的权值的求法。
利用列表构造出图结构,列表的索引表示1~2021号节点(不使用零号索引),对应每个索引储存的元素为“从一号节点到当前节点的最小开销”,进而类似一个“动态规划”的想法得到“从1号节点到2021号节点的最小开销”。
要注意的是,依题意,每个节点能连接21个其它节点。