算法思路
这题要注意几点:
1、所有边的权值相同,且均为正值,那么往后走,路径必然递增,而且是层序递增。所以这时不需要用优先队列,也不需要将每个点路径初始化为无穷,初始化比0小的值就可以了。
2、红绿灯的影响对每层是一样的,所以不用担心它会影响路径的递增
3、最短路径必然只访问所有节点一次,而第二短路径最多访问节点2次,因为如果存在访问某节点3次,它的路径必然比访问该节点2次的路径长。所以更新时只要限定为访问2次并且不是同一层级的更新,就可以了。
class Solution {
public:
int secondMinimum(int n, vector<vector<int>>& edges, int time, int change) {
vector<int> g[n+1];
for(auto & e:edges){
g[e[0]].emplace_back(e[1]);
g[e[1]].emplace_back(e[0]);
}
vector<int> visited(n+1,0);
vector<int> dist(n+1,-1);
queue<pair<int,int>> q;
q.emplace(1,0);
while(!q.empty()){
auto [i,t]=q.front();q.pop();
int tt;
int round=t/change; //求改变次数
if(round%2==0) tt=t+time;
else tt=(round+1)*change+time;
for(auto & nxt:g[i]){
if(visited[nxt]<2 && dist[nxt]<tt){//路径必须比tt短才可更新,否则同一层级也会将nxt路径更新
dist[nxt]=tt;
visited[nxt]++;
q.emplace(nxt,tt);
if(n==nxt && visited[nxt]==2) return tt;
}
}
}
return -1;
}
};