最短路模板spfa+dijstkra

代码是畅通工程续这儿、

这里用的是vector存图,换成链式前向星也是一样

spfa

spfa有两个数组,表示d[i]表示i点到s点的最短距离,inq[i]表示i点是否在队列里1表示在0表示不在

那么每次从队列中取出一个点进行松弛操作,松弛完之后看点是否在队列里,不在则加入队列

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const ll maxn = 105;
int d[maxn],inq[maxn],n,m;
vector<pair<int,int> > E[maxn];
void init()
{
    for(int i=0;i<maxn;i++)E[i].clear();
    for(int i=0;i<maxn;i++)d[i] = 1e9;
    for(int i=0;i<maxn;i++)inq[i] = 0;
}
int main()
{
    while(cin>>n>>m)
    {
        init();
        for(int i=0;i<m;i++)
        {
            int x,y,z;
            scanf("%d%d%d",&x,&y,&z);
            E[x].push_back(make_pair(y,z));
            E[y].push_back(make_pair(x,z));
        }
        int s,t;
        scanf("%d%d",&s,&t);
        queue<int> Q;
        Q.push(s),d[s] = 0,inq[s] = 1;

        while(!Q.empty())
        {
            int now = Q.front();
            Q.pop();
            inq[now] = 0;

            for(int i=0;i<E[now].size();i++)
            {
                int v = E[now][i].first;
                int x = E[now][i].second;
                if(d[v]>d[now]+x)
                {
                    d[v] = d[now]+x;
                    if(inq[v])continue;
                    inq[v] = 1;
                    Q.push(v);
                }
            }
        }
        if(d[t]==1e9)printf("-1\n");
        else printf("%d\n",d[t]);

    }
    return 0;
}

dijstkra堆优化

每次取出离s最近的那个点进行松弛

优先队列相当于一个自动排好序的队列(可以自己写优先级),这里不写队首就是最大的,

那么写-d[i]每次取出来的就是d[i]最小的那个,这里注意队列中pair的d[i]起到的就是一个标记第几小的作用

#include <bits/stdc++.h>
using namespace std;
vector<pair<int,int> > E[205];
int dis[205];
void init()
{
    for(int i=0;i<205;i++)dis[i] = 1e9;
    for(int i=0;i<205;i++)E[i].clear();
}

int main() {

    int n,m;
    while(~scanf("%d%d",&n,&m))
    {
        init();
        for(int i=0;i<m;i++)
        {
            int x,y,z;
            scanf("%d%d%d",&x,&y,&z);
            E[x].push_back(make_pair(y,z));
            E[y].push_back(make_pair(x,z));
        }
        int s,t;
        scanf("%d%d",&s,&t);

        priority_queue<pair<int,int> > Q;

        Q.push(make_pair(0,s));dis[s] = 0;

        while(!Q.empty())
        {
            int now = Q.top().second;
            Q.pop();

            for(int i=0;i<E[now].size();i++)
            {
                int v = E[now][i].first;
                int x = E[now][i].second;
                if(dis[v]>dis[now]+x)
                {
                    dis[v] = dis[now]+x;
                    Q.push(make_pair(-dis[v],v));
                }
            }
        }
        if(dis[t]!=1e9)printf("%d\n",dis[t]);
        else printf("-1\n");
    }
    return 0;
}

猜你喜欢

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