[PAT]算法模版- Dijkstra+pre数组

/
写一个经典 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
    
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值