我有一个函数，我可以从稀疏邻接矩阵中的任何初始节点集中找到所有节点 k 步数。通常这个初始集合是单个分支的 to-from 节点。我想要一个“用于”到达这些节点的分支列表。
我用来查找连接节点的函数如下：
% nb = number of buses in the system
% branch_list = [from to] = the from/to nodes of each branch
adj_mat = sparse(from,to,1,nb,nb);
node_vec0 = sparse([from to],1,1,nb,1);
A = adj_mat + speye(nb); % Add 1's on the diagonal of the adjacency matrix
node_vec = A * node_vec0; % Vector containing all nodes connected to node_vec0
我可以重复最后一行 k 次，并从初始节点找到所有节点 k 步。
我想要做的是找到用于到达这些节点的每个分支的行（在分支列表中）。
有没有一种有效的方法来做到这一点？
谢谢！