Cod sursa(job #2361303)

Utilizator ionanghelinaIonut Anghelina ionanghelina Data 2 martie 2019 14:25:52
Problema Algoritmul lui Dijkstra Scor 10
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.64 kb

#include<bits/stdc++.h>
using namespace std;

const int maxN=(5e4)+5;
int dp[maxN],pre[maxN];

struct tip
{
    int nod;
    bool operator<(const tip&other) const
    {
        if(dp[nod]==dp[other.nod]) return nod<other.nod;
        return dp[nod]<dp[other.nod];
    }
};

set<tip> q;



int n,m,x,y,z;
vector<pair<int,int> > edges,v[maxN];

const int inf=(1e9);
int sol;
bool seen[maxN],inSet[maxN];


int main()
{
    freopen("dijkstra.in","r",stdin);
    freopen("dijkstra.out","w",stdout);


    scanf("%d%d",&n,&m);

    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&x,&y,&z);

        v[x].push_back(make_pair(y,z));
        v[y].push_back(make_pair(x,z));
    }

    dp[1]=0;

    for(int i=2;i<=n;i++)
        dp[i]=inf;


    q.insert({1});
    inSet[1]=1;

    int nodes=0;
    while(!q.empty())
    {
        int nod=(*q.begin()).nod;
        int cost=dp[nod];

        q.erase(q.begin());

     //   if(cost>dp[nod]) continue;
        seen[nod]=1;
        sol+=cost;
        if(nod!=1)
            edges.push_back(make_pair(pre[nod],nod));

        for(auto it:v[nod])
        {
            int z=cost+it.second;
            if(seen[it.first]) continue;
            if(dp[it.first]>z)
            {
                if(inSet[it.first]) q.erase(q.find({it.first}));
                    else inSet[it.first]=1;
                pre[it.first]=nod;
                dp[it.first]=z;
                q.insert({it.first});
       //         q.push({it.first,dp[it.first]});

            }
        }
    }

    for(int i=2;i<=n;i++)
        printf("%d ",dp[i]);


    return 0;
}