题意
给定一个有向图,两个起点(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 ,求s1到a的最短路,s2到a的最短路,dest到a的最短路,使得路径之和最小。
- 用dijkstra算法,先求出s1,s2到所有点的距离,还有反向图中dest到所有点的距离。
- 这里的dijkstra要用堆优化的dijkstra,朴素的dijkstra不行。
- golang当中的优先队列(或者说堆)操作,相对cpp复杂一点,需要实现接口。
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



