LeetCode周赛284-第四题:dijkstra(golang+cpp)

题目链接:LeetCode6032


题意

        给定一个有向图,两个起点(s1,s2)和一个终点(dest),要求求出路径最小的带权子图,从两个起点都可以到达终点,且该子图的所有边权之和最小。

思路

【力扣周赛讲解】【第 284 场周赛】得到要求路径的最小带权子图_哔哩哔哩_bilibili【力扣周赛讲解】【第 284 场周赛】得到要求路径的最小带权子图https://www.bilibili.com/video/BV1BU4y1o7GS?from=search&seid=9213995241827836213&spm_id_from=333.337.0.0

  • zerotrac(零神yyds)讲得特别好。
  • 大体思路就是枚举一个中间节点 a ,求s1a的最短路,s2a的最短路,desta的最短路,使得路径之和最小。
  • dijkstra算法,先求出s1,s2到所有点的距离,还有反向图中dest到所有点的距离
    • 这里的dijkstra要用堆优化的dijkstra,朴素的dijkstra不行。
    • golang当中的优先队列(或者说堆)操作,相对cpp复杂一点,需要实现接口。

代码(Golang)

type edge struct{
    to,weight int
}
const maxx int=int(1e18)

func dij(g [][]edge,source int)[]int{
    dis:=make([]int,len(g))
    for i:=range dis{
        dis[i]=maxx 
    }
    dis[source]=0

    h:=hp{{source,0}}
    for len(h)>0{
        p:=heap.Pop(&h).(pair)
        v:=p.v
        if dis[v]j{
		return j
	}
	return i
}

代码(Cpp)

class Solution {
public:
    typedef long long ll;
    const ll maxx=1e18;
    vector dij(vector>>g,int source){
        vectordis(g.size(),maxx);
        dis[source]=0;
        priority_queue, vector>, greater>> pq;

        pq.push({0ll,source});
        while(pq.size()>0){
            auto now=pq.top();
            pq.pop();
            ll w=now.first;int to=now.second;
            if (dis[to]>& edges, int src1, int src2, int dest) {
        vector>>g(n),rg(n);
        for(int i=0;idis1,dis2,dis3;
        dis1=dij(g,src1);
        dis2=dij(g,src2);
        dis3=dij(rg,dest);
        ll ans=maxx;
        for(int i=0;i

你可能感兴趣的:(LeetCode,leetcode,golang,算法)