服务器广播，一次遍历数组完成

服务器广播

题目：服务器连接方式包括直接相连，间接连接。 A 和 B 直接连接， B 和 c 直接连接，则 A 和 c 间接连接。直接连接和间接连接都可以发送广播。
给出一个 N * N 数组，代表 N 个服务器， matrix[i][j] == 1 ，则代表 i 和 j 直接连接；不等于 1 时，代表 i 和 j 不直接连接。 matrix[i][i]== 1 ，即自己和自己直接连接。 matrix[i][j]==matrix[j][i] 。计算初始需要给几台服务器广播，才可以使侮个服务器都收到广播。
输入描述： n * n 矩阵，

[[1,1,0],[1,1,0],[0,0,1]]
输出描述：整数

2
 

问题可以转化为最小最大子图问题，代码如下：

def minimal_broadcast(matrix):
    n = len(matrix)
    stack = []
    full_stack = [i for i in range(n)]
    iterator_stack = []
    res = []
    while full_stack:
        stack.append(full_stack[0])
        minimal_graph = []
        while stack:
            i = stack.pop(0)
            iterator_stack.append(i)
            if i in full_stack:
                full_stack.pop(full_stack.index(i))
            for j in range(n):
                if matrix[i][j] == 1:
                    if j not in stack and j not in iterator_stack:
                        stack.append(j)
                    if j not in minimal_graph:
                        minimal_graph.append(j)
                # print(minimal_graph)
        res.append(minimal_graph)
        # print("*****")
    return len(res)

if __name__ == '__main__':
    matrix = [[1,1,0], [1,1,1], [0,1,1]]
    n = minimal_broadcast(matrix)
    print(n)