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

 Hint

 我的想法：

 我的代码：

if __name__ == '__main__':
    # 路由器B的路由表有M行
    # 路由器C的路由表有N行
    M, N = input().split(' ')

    count = 0

    B = []
    C = []
    # 将其存储到列表中
    for i in range(1, int(M) + int(N) + 1):
        line = input()
        l = line.split(' ')
        if int(M) <= count:
            C.append(l)
        else:
            B.append(l)
        count += 1

    # c向B发送消息
    # 先修改此 RIP 报文中的所有项目：把“下一跳”字段中的地址都改为 X，并把所有的“距离”字段的值加 1
    for c in C:
        cc = c
        c[2] = 'C'
        c[1] = int(c[1]) + 1

        flag = 1

        for b in B:
            if c[0] in b:
                # （即在路由表中有目的网络N），若下一跳R路由器地址为X，则把收到的项目替换原路由表中的项目
                flag = 0
                if b[2] == 'C':
                    B.remove(b)
                    B.append(cc)
                else:
                    if int(c[1]) < int(b[1]):
                        B.remove(b)
                        B.append(c)
                        break
                    else:
                        break
                break

        # 原来的路由表中没有目的网络N，则把该项目添加到路由表中
        if flag == 1:
            B.append(c)

    B.sort(key=lambda x: x[0])


    for b in B:
        print(f"{b[0]} {b[1]} {b[2]}")