【模板】单源最短路径(标准版)

洛谷P4779

尝试了下洛谷里大佬说的几个Ac的方法

感觉都差不多

用时: 1487ms / 内存: 13272KB------------------------------------迪杰斯特拉堆优化+标记数组
用时: 1525ms / 内存: 12656KB------------------------------------魔改spfa(迪杰斯特拉+spfa的队列数组)
用时: 1501ms / 内存: 12960KB------------------------------------迪杰斯特拉堆优化+判断

第1,2个都需要额外开一个n的数组,3不需要

O2优化之后就都300ms左右了(话说手写堆的是真的强)

用时: 365ms / 内存: 13012KB

用时: 329ms / 内存: 12704KB

用时: 335ms / 内存: 13376KB

板子,堆优化+标记数组优化

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
vector<pair<ll,ll> > E[200005];
ll dis[100005];
bool vis[100005];
void init(ll n,ll m)
{
    for(ll i=0;i<=n;i++)dis[i] = 1e18;
    for(ll i=0;i<=m;i++)E[i].clear();
}
void dijstkra(ll s){
    priority_queue<pair<ll,ll> > Q;
    Q.push(make_pair(0,s));dis[s] = 0;
    while(!Q.empty())
    {
        ll now = Q.top().second;
        Q.pop();
        if(vis[now])continue;
        vis[now] = 1;
        for(ll i=0;i<E[now].size();i++)
        {
            ll v = E[now][i].first;
            ll x = E[now][i].second;
            if(dis[v]>dis[now]+x)
            {
                dis[v] = dis[now]+x;
                Q.push(make_pair(-dis[v],v));
            }
        }
    }
}
int main() {

    ll n,m,s;
    while(~scanf("%lld%lld%lld",&n,&m,&s))
    {
        init(n,m);
        for(ll i=0;i<m;i++)
        {
            ll x,y,z;
            scanf("%lld%lld%lld",&x,&y,&z);
            E[x].push_back(make_pair(y,z));
            //E[y].push_back(make_pair(x,z));
        }
        dijstkra(s);
        for(int i=1;i<n;i++)
            printf("%lld ",dis[i]!=1e18?dis[i]:2147483647);
        printf("%lld\n",dis[n]!=1e18?dis[n]:2147483647);
    }
    return 0;
}

spfa魔改

void dijstkra(ll s){
    priority_queue<pair<ll,ll> > Q;
    Q.push(make_pair(0,s));dis[s] = 0;
    while(!Q.empty())
    {
        ll now = Q.top().second;
        Q.pop();
        in[now] = 0;
        for(ll i=0;i<E[now].size();i++)
        {
            ll v = E[now][i].first;
            ll x = E[now][i].second;
            if(dis[v]>dis[now]+x)
            {
                dis[v] = dis[now]+x;
                if(!in[v])
                    Q.push(make_pair(-dis[v],v)),in[v] = 1;
            }
        }
    }
}

迪杰斯特拉堆优化+判断

void dijstkra(ll s){
    priority_queue<pair<ll,ll> > Q;
    Q.push(make_pair(0,s));dis[s] = 0;
    while(!Q.empty())
    {
        ll now = Q.top().second;
        ll t = -Q.top().first;
        Q.pop();
        if(t>dis[now])continue;
        for(ll i=0;i<E[now].size();i++)
        {
            ll v = E[now][i].first;
            ll x = E[now][i].second;
            if(dis[v]>dis[now]+x)
            {
                dis[v] = dis[now]+x;
                Q.push(make_pair(-dis[v],v));
            }
        }
    }
}

猜你喜欢

转载自blog.csdn.net/Du_Mingm/article/details/82961942
今日推荐