Question

假设有数组和无向边的节点作为向量，如下所示：

int nodes[n] = {1, 2, 3, ... ,n };
vector<pair<int, int>> edges;

edges.push_back(std::make_pair(0, 2));
edges.push_back(std::make_pair(2, 4));

其中数组的每个元素都是值，n是数组的数量。按照上面的代码，有两个边。一个是0到2.另一个是2到4.这些数字表示数组的索引。在这种情况下，最大子树的大小为3，0-2-4，最小子树的大小明显为1。

我解决了这个问题，如下所示：

排序edges向量
在edges
重复2直到探索所有可能的情况

但我不确定这是否有效。如何在问题域中获取所有子树？有什么通用和有效的方法吗？

Answer 1

我使用基于边缘信息的BFS（广度优先搜索）解决了这个问题。为了避免制作循环图并将节点保存为树，我使用set。我也会在搜索之前申请sort。减少时间复杂度很有用。

void BFS(set<int> nodes, vector<pair<int,int>>& edges, vector<set<int>>& result) {
    result.push_back(nodes);
    int lastNode = *max_element(nodes.begin(), nodes.end());
    auto findIt = find_if(edges.begin(), edges.end(), [](const pair<int, int>& element){ return element.first == lastNode});
    if(findIt != edges.end()) {
        nodes.insert((*findIt).second);
        BFS(nodes, edges, result);
    }
}

sort(edges.begin(), edges.end());

vector<set<int>> result;
for(auto it : edges) {
    set<int> nodes;
    nodes.insert((*it).first);
    nodes.insert((*it).second);
    BFS(nodes, edges, result);
}

从节点数组和边向量中获取所有子树的最有效方法是什么？

1 个答案: