此题看题意还是挺难的,但我们可以想象一下最终获得的子图是个什么形状:一定是图中有任意一个节点,两个出发点先到这个点汇合,再去终点。
所以,我们通过dijkstra算法求三个点的单源最短路,其中终点的最短路需要存反的邻接表。然后枚举每个中转点即可。
注意:dijkstra有几个优化点,一个是用两层vector实现邻接表,用map会慢。还有就是用-1初始dis表会好一些,如果用大数还需要考虑溢出,而且初始化较慢。
typedef long long ll;
typedef pair<ll, int> P;
class Solution {
public:
vector<ll> dijkstra(int n, int src, vector<vector<P>>& adj)
{
vector<ll> dis(n, -1);
dis[src] = 0;
priority_queue<P> q;
q.push(P(0, src));
while