题意:有\(n\)个点,\(m\)条双向边,两个方向的权值都是相等的,可以从\(A\)中的某个点出发走到\(B\)中的某个点,求所有路径中的最短距离,如果A和B中没有点联通,则输出\(-1\).
题解:感觉是个阅读理解啊,题目看懂了就是个裸的单源最短路,我们首先将牛牛的所有星球初始化作为起点,然后建边跑个dijkstra,最后再枚举牛妹的星球维护一个最小值即可.
-
代码:
class Solution {
public:
/**
*
* @param niuniu int整型vector 牛牛占领的p个星球的编号
* @param niumei int整型vector 牛妹占领的q个星球的编号
* @param path int整型vector<vector<>> m条隧道,每条隧道有三个数分别是ui,vi,wi。ui,vi分别是隧道的两边星球的编号,wi是它们之间的距离
* @param nn int整型 星球个数n
* @return int整型
*/
struct misaka{
int out;
int val;
}e;
int dis[1000000];
bool st[1000000];
vector<misaka> v[1000000];
priority_queue<pair<int,int>,vector<pair<int,int>>,greater<pair<int,int>>> h; void dijkstra(){
while(!h.empty()){
auto tmp=h.top();
h.pop(); int num=tmp.second;
int dist=tmp.first; if(st[num]) continue;
st[num]=true; for(auto w:v[num]){
int node=w.out;
if(dis[node]>w.val+dist){
dis[node]=w.val+dist;
h.push({dis[node],node});
}
}
}
} int Length(vector<int>& niuniu, vector<int>& niumei, vector<vector<int> >& path, int nn) {
// write code here
memset(dis,0x3f3f3f3f,sizeof(dis));
memset(st,false,sizeof(st));
for(auto w:niuniu){
dis[w]=0;
h.push({0,w});
} for(auto w:path){
int a=w[0];
int b=w[1];
int val=w[2];
e.out=b,e.val=val;
v[a].push_back(e);
e.out=a;
v[b].push_back(e);
} dijkstra();
int ans=0x3f3f3f3f; for(auto w:niumei){
ans=min(ans,dis[w]);
} if(ans==0x3f3f3f3f) ans=-1;
return ans; }
};