蓝桥杯：网络分析之Python求解

【问题描述】
小明正在做一个网络实验。
他设置了 n 台电脑，称为节点，用于收发和存储数据。
初始时，所有节点都是独立的，不存在任何连接。
小明可以通过网线将两个节点连接起来，连接后两个节点就可以互相通信了。两个节点如果存在网线连接，称为相邻。
小明有时会测试当时的网络，他会在某个节点发送一条信息，信息会发送到每个相邻的节点，之后这些节点又会转发到自己相邻的节点，直到所有直接或间接相邻的节点都收到了信息。所有发送和接收的节点都会将信息存储下来。一条信息只存储一次。
给出小明连接和测试的过程，请计算出每个节点存储信息的大小。

【输入格式】
输入的第一行包含两个整数 n,m，分别表示节点数量和操作数量。节点从1 至 n 编号。
接下来 m 行，每行三个整数，表示一个操作。
如果操作为 1 a b，表示将节点 a 和节点 b 通过网线连接起来。当 a = b时，表示连接了一个自环，对网络没有实质影响。
如果操作为 2 p t，表示在节点 p 上发送一条大小为 t 的信息。

【输出格式】
输出一行，包含 n 个整数，相邻整数之间用一个空格分割，依次表示进行完上述操作后节点 1 至节点 n 上存储信息的大小。

【样例输入】
4 8
1 1 2
2 1 10
2 3 5
1 4 1
2 2 2
1 1 2
1 2 4
2 2 1

【样例输出】
13 13 5 3

我的解题思路：
这道题的关键是判断与节点p直接连通或间接连通的节点有哪些，我对此给出的方案是声明一个含有n+1个集合的列表 netL=[ {}, {}, {}, {}, …, {}, {} ]，下标为 p 的集合中存储着与节点 p 直接相连的所有 i, j, k，对 netL[ p ]、netL[ i ]、netL[ j ]、netL[ k ]求并集，即可求得与节点 p 直接或间接相连的所有节点。

完整的代码如下：

netL=[{}]  #其元素为集合，下标为1的集合中存与节点1连通的所有节点
infoL=[0]  #用于存储每个节点上所接收到的信息加和，infoL[1]表示节点1所收到的信息总和

'''根据输入信息，判断是进行连接操作还是发送操作'''
def judge_flag(flag,L):
    if flag==1:
        a=int(L[1])
        b=int(L[2])
        netL[a].add(b)
        netL[b].add(a)
    else:
        p=int(L[1])
        t=int(L[2])
        netSet=set()
        tempSet=netL[p]
        for item in tempSet:
            netSet=netSet | netL[item]  #求并集，以计算所有与节点p直接连通或间接连通的节点集合
        for item in netSet:
            infoL[item]+=t

'''按题目要求输出'''
def output():
    for i in range(1,len(infoL)):
        if i<len(infoL)-1:
            print(infoL[i],end=' ')
        else:
            print(infoL[i])        

def main():
    Str=input()
    L=Str.split(' ')
    n=int(L[0])    #注意：Python默认接收的数据为字符串
    m=int(L[1])
    nodeL=[]
    
    for i in range(1,n+1):
        netL.append({i})
        infoL.append(0)
        nodeL.append(i)
          
    for i in range(m):
        Str=input()
        L=Str.split(' ')
        flag=int(L[0])
        judge_flag(flag,L)

    output()

main()