dijikstra模板

/*适用于无负权且对稠密图有优势
假设一个图G=(V,E)有n个节点,图G的每个节点的出度是一个固定的常数:k。由于E=kV=O(V) ,所以我们把符合E=O(V) 条件的图称为稀疏图。
    同理 :
    如果一个图G=(V,E)有n个节点,假设图G的每个节点的出度是关于n的一个小数,并且0<f<=1,我们把符合E=fV2(平方)=V2(平方)条件的图称为稠密图。
    比如:一个图节点为16,节点的出度为4,那么f = 0.25。
    或以E是否远远大于V^2为标准*/
#include<bits/stdc++.h>
using namespace std;
const int nMax=1005,eMax=5005;
const long long inf=0x3f3f3f3f; //如果距离超出int范围,则0x3f3f3f3f3f3f3f3f.
struct pp
{
    int v, w, next;
}edge[2*eMax]; // 无向边
int head[nMax], dis[nMax], vis[nMax],n,m,cnt=0;
struct node
{
    int u, dis;
    bool operator < (const node &a) const
    {
        return a.dis < dis;
    }
};
void add(int t1,int t2,int t3)
{
    edge[++cnt].v=t2;
    edge[cnt].w=t3;
    edge[cnt].next=head[t1];
    head[t1]=cnt;
}
int dijkstra(int b,int ed) 
{
    priority_queue<node> que;
    int i;
     memset(dis,0x3f3f3f3f,sizeof(dis));
     memset(vis,0,sizeof(vis));
    dis[b] = 0;
    node  temp;
    temp.u = b; temp.dis = 0;
    que.push(temp);
    while (!que.empty()) 
    {
        int u = que.top().u;
        que.pop();
        if (vis[u]) continue;
        vis[u] = 1;
        for (i=head[u]; i!=0; i=edge[i].next) 
        {
            int v = edge[i].v;
            if (!vis[v] && dis[v] > dis[u] + edge[i].w) 
            {
                dis[v] = dis[u] + edge[i].w;
                temp.u = v; temp.dis = dis[v];
                que.push(temp);
            }
        }
    }
    return dis[ed];
}
int main()
{
    int t1,t2,t3;
    cin>>n>>m;
    while(m--)
    {
        scanf("%d%d%d",&t1,&t2,&t3);
        add(t1,t2,t3);
        add(t2,t1,t3);
    }
    cout<<dijkstra(1,n)<<endl;
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_37358510/article/details/81333428
今日推荐