堆优化拓扑排序

拓扑排序用于解决有向无环图的一些相关问题。

    priority_queue<pair<int, int>,vector<pair<int,int>>,greater<pair<int, int>>> deg_idx;
    for (int i = 1; i <= n; i++)
    {
        deg_idx.push({ out_degree[i],i });
    }
    while (deg_idx.size())
    {
        auto [deg, idx] = deg_idx.top();
        deg_idx.pop();
        if (vis[idx]) continue;
        vis[idx] = true;
        for (auto child : ch[idx])
        {
            out_degree[child]--;
            deg_idx.push({ out_degree[child] ,child });
            new_val[child] = max(new_val[child], new_val[idx]);
        }
    }
    int res = 0;
    for (int i = 1; i <= n; i++)
        res =max(res, new_val[i] - val[i]  );
    cout << res << endl;
View Code

可以使用堆优化拓扑排序的构建。

首先先把全部点的度和idx放入优先队列

循环拿到堆顶选择最小度的点,并更新其子节点的数值,更新子节点的度数并重新插入子节点。

例题:

https://www.hackerearth.com/practice/algorithms/graphs/topological-sort/practice-problems/algorithm/wise-business-89b04f1a/

posted @ 2022-07-13 16:39  80k  阅读(87)  评论(0)    收藏  举报