OJ-Problem G: 距离向量算法（D-V）

Problem Description

RIP路由协议是一种分布式的基于距离向量的路由选择协议。
距离向量（D-V）算法：
收到相邻路由器（设其地址为 X）的一个 RIP 报文： 
① 先修改此 RIP 报文中的所有项目：把“下一跳”字段中的地址都改为 X，并把所有的“距离”字段的值加 1。 
② 对修改后的RIP报文中的每一个项目，重复以下步骤：
若原来的路由表中没有目的网络N，则把该项目添加到路由表中；
否则（即在路由表中有目的网络N），若下一跳R路由器地址为X，则把收到的项目替换原路由表中的项目；
否则（即下一跳路由器不是X），若收到的项目中的距离d小于路由表中的距离，则进行更新。否则，什么也不做。 
③ 若 3 分钟还没有收到相邻Router的更新信息，则把此相邻Router标记为不可达的路由器，即把距离置为16（距离为16表示不可达）。
④ 返回。
现在假设有一台路由器B，收到邻居路由器C发送过来的RIP路由更新报文。编写程序：使用D-V算法更新路由器B的路由表。

Input Description

输入的第一行为两个正整数M、N（M,N<100），分别表示路由器B的路由表有M行，邻居路由器C发送过来的路由表有N行。
接下来的M+N行，每行包括3个字段：目的网络、距离、下一跳路由器。
前M行为路由器B的路由表，后N行为邻居路由器C发送过来的路由表。

Output Description

输出更新后路由器B的路由表，格式详见样例输出。

Sample Input

2 3
N1 3 C
N3 4 E
N1 3 D
N2 4 F
N3 1 -

Sample Output

N1 4 C
N2 5 C
N3 2 C

源代码

 

def getDict(dic, length):
    for i in range(length):
        temp = list(input().split(" "))
        dic[temp[0]] = temp[1:len(temp)]


def compare(dic1: dict, dic2: dict):
    for key in dic1.keys():
        if key in dic2:#目的网络在C的路由表中也存在
            if dic1[key][1] == dic2[key][1]:#下一跳相同
                dic1[key][0] = dic2[key][0]#更新B表
            else:#下一跳不相同
                if int(dic1[key][0]) > int(dic2[key][0]):#比较跳数
                    dic1[key][0] = dic2[key][0]#取跳数小的
                    dic1[key][1] = dic2[key][1]#更新下一跳地址
    for key in dic2.keys():#把存在于C中但不存在于B中的表项添加到B中
        if key in dic1:
            continue
        else:
            dic1[key]=dic2[key]
    for key in sorted(dic1.keys()):#按键升序输出最终B的路由表信息
        print(f"{key} {dic1[key][0]} {dic1[key][1]}")


def change(dic):
    for key in dic.keys():
        dic[key][0] = int(dic[key][0]) + 1#把所有的“距离”字段的值加 1。
        dic[key][1] = "C"#把“下一跳”字段中的地址都改为 X


m, n = map(int, input().split(" "))
dic1 = {}
dic2 = {}
#getDict函数读入信息，并把目的网络作为键，列表[跳数,下一跳地址]作为值，构成字典
getDict(dic1, m)
getDict(dic2, n)
#change函数：把“下一跳”字段中的地址都改为 X，并把所有的“距离”字段的值加 1。
change(dic2)
#compare函数：对表项进行对比，确定B的表项
compare(dic1, dic2)

 运行结果