最短路堆优化

#include <bits/stdc++.h>
#define debug(x) cout<<#x<<" = "<<x<<endl
using namespace std;
typedef long long ll;
const int mod=1e9+7;
#define INF 0X7FFFFFFF
#define MAXN 1000005
struct node
{
    bool know; //当前顶点距离起点是否确定最短
    int id;     //当前顶点编号
    int dist;   //当前顶点到起点的最短距离
    int path;   //当前顶点距离起点的最短路径的前一个顶点(即该顶点的前一个点)
    friend bool operator < ( node first , node second )
    {
        return first.dist > second.dist;//从小到大排列
    }
}spot[MAXN];
struct Edge
{
    int to; // 末点编号
    int dis;    // 顶点到末点的距离
};
vector<Edge>Map[MAXN];
int val[MAXN];// 当前位置的权值
priority_queue<node>Q;
void Dijkstra()
{
    int s=1;    // 起点
    spot[s].dist=0; //初始化起点
    spot[s].know=true;  
    for(unsigned int i=0;i<Map[s].size();i++){
        if( spot[s].dist + Map[s][i].dis < spot[ Map[s][i].to ].dist ){
            spot[ Map[s][i].to ].dist = spot[s].dist + Map[s][i].dis ;
            spot[ Map[s][i].to ].path = s;
            Q.push(spot[ Map[s][i].to ]);
        }   //寻找与起点相连的顶点记录其距离
    }
    node now;
    while( !Q.empty() ){
        now = Q.top();
        Q.pop();
        if(now.know) continue;// 如果该顶点距离起点已经最近则跳过
        now.know=true;      //否则对该点寻找最短到达起点的距离
        for(unsigned int i=0;i<Map[ now.id ].size();i++){
            if( !spot[ now.id ].know ){
                    if( spot[ now.id ].dist + Map[ now.id ][i].dis < spot[ Map[ now.id ][i].to ].dist ){
                    spot[ Map[ now.id ][i].to ].dist = spot[ now.id ].dist + Map[ now.id ][i].dis ;
                    spot[ Map[ now.id ][i].to ].path =  now.id ;
                    Q.push(spot[ Map[s][i].to ]);
                }//寻找最短路径
            }
        }
    }
    return ;
}
int dp[MAXN];
void solve()
{
    int n,m,t;
    cin>>n>>m>>t;
    for(int i=2;i<=n;i++){
        cin>>val[i]; //顶点的权值
    }
    int from,to; //初点,末点
    for(int i=1;i<=m;i++){
        cin>>from>>to;
        Map[from].push_back( { to , 2 } );//无向图 
        Map[to].push_back( { from , 2 } );// 两点之间距离
    }
    for(int i=1;i<=n;i++){
        spot[i].know=false;
        spot[i].dist=INF;
        spot[i].path=0;
        spot[i].id=i;
    }// Dijkstra算法表 数据初始化
    Dijkstra();
    for(int i=1;i<=t;i++){
        for(int j=2;j<=n;j++){
            if( i >= spot[j].dist ){
                dp[i] = max( dp[i], dp[ i - spot[j].dist ]+val[j] );
            }
        }
    } // 寻找在当前步数中能获得的最大权值
    for(int i=1;i<=t;i++){
        cout<<dp[i]<<" ";
    }// 输出
    cout<<endl;
    return ;
}
signed main()
{
    ios::sync_with_stdio(false);
    cin.tie(0),cout.tie(0);
//    clock_t start,last;
//    start = clock();
    int _=1;
//    cin>>_;
    while(_--){
        solve();
    }
//    last = clock();
//    cout<<(double)( last - start )/CLOCKS_PER_SEC;
    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值