Python 电网代码示例 - 数据结构与算法

以下是使用 Python 实现电网代码的示例：

class PowerGrid:
    def __init__(self, num_of_nodes):
        self.num_of_nodes = num_of_nodes
        self.grid = [[0 for x in range(num_of_nodes)] for y in range(num_of_nodes)]

    def add_edge(self, i, j):
        self.grid[i][j] = 1
        self.grid[j][i] = 1

    def remove_edge(self, i, j):
        self.grid[i][j] = 0
        self.grid[j][i] = 0

    def get_neighbors(self, node):
        neighbors = []
        for i in range(self.num_of_nodes):
            if self.grid[node][i] == 1:
                neighbors.append(i)
        return neighbors

该代码使用了邻接矩阵来表示电网结构，其中每个元素表示两个节点之间是否存在连接。示例代码中包含了添加边、删除边和获取节点邻居等基本操作，可作为学习电网数据结构与算法的入门参考。