/
写一个经典 Dijkstra + pre数组 结合的算法模板。
解决 “第一条件最优下选择第二条件最优” 的问题
/
#include<bits/stdc++.h>
// 系统资源
vector<vector<int>> G;
vector<vector<int>> pre;
vector<int> pointValue;
// DFS的判断条件
vector<int> tempPath;
vector<int> bestPath;
int bestValue ;
int tempValue ;
// 出发点 和 终点
int from , go ;
// 选择更优【条件自定】
void DFS(int nowPoint){
tempPath.push_back(nowPoint);
tempValue+=pointValue[nowPoint];
if(nowPoint == endPoint){
// 当到了最后一个点的时候。 需要看看 if tempValue > bestValue 则需要更新 bestPath 其余的不变。。
if(bestPath < tempValue){
bestPath = tempPath;
bestValue = tempValue;
}
}else{
vector<int> v = pre[nowPoint];
for(int i = 0 ; i < v.size() ; i++){
DFS(v[i]);
}
}
// 最终都需要原路返回。。
tempPath.pop_back();
tempValue-=pointValue[nowPoint];
}
// 找到单源路径数组
vector<int> dijst(int s){
int len = G.size();
// 每个点,是否已经成为过最短距离点
vector<bool> vis(len,true);
vector<long> mindis(len,INT_MAX);
// 起点0距离
mindis[s] = 0;
// 进行N次循环,可以探索N次,找到N个最短距离点
for(int i = 0 ; i < len ; i++){
int u = -1 , cur_mindis = INF_MAX;
if(cur_mindis > mindis[i]){
u = i ;
cur_mindis = mindis[i];
}
// 这里找到本轮可以到达的本次的最短距离点【同时这个点,未成为过最短距离中续点!!】
if(u==-1)return -1;
// U点成为最短距离中续点
vis[u] = true;
// 开始更新最短距离
for(int j = 0 ; j < len ; j++){
if( mindis[u] + G[u][j] < mindis[j] ){
mindis[j] = mindis[u] + G[u][j];
// 找到最短距离,可以更新pre数组
pre[j].clear();
pre[j].push_back(u);
}else if(mindis[u] + G[u][j] == mindis[j]){
pre[j].push_back(u);
}
}
}
return mindis;
}
int main(){
// 输入数据,构造 from , go , G , pointValue
// dijst(from)
// DFS(go)
// 输出 bestValue AND bestPath
}