Cod sursa(job #1694651)

Utilizator ajeccAjechiloae Eugen ajecc Data 25 aprilie 2016 19:09:57
Problema Algoritmul lui Dijkstra Scor 10
Compilator cpp Status done
Runda Arhiva educationala Marime 2.33 kb
#include <bits/stdc++.h>
#define for0(i,n) for(int i=0; i<n; i++)
#define for1(i,n) for(int i=1; i<=n; i++)
#define pb push_back
#define mp make_pair
#define all(v) v.begin(), v.end()
#define V vector<int>
#define VP vector<pair<int, int> >
#define clr(A,x) memset(A, x, sizeof(A))
#define cpy(A,B) memcpy(A, B, sizeof(B))
#define g(s) getline(cin, s) ///ai grija la fin/cin ///
#define FASTIO ios_base::sync_with_stdio(0)
typedef long long ll;
typedef unsigned long long ull;
using namespace std;
/*template <typename T>
string to_string(const T& n){
    ostringstream os;
    os << n;
    return os.str();
}
*/

/*struct coord
{
    int x,y;

};
bool operator<(const coord &l, const coord &r)
{
    return (l.x<r.x || (l.x==r.x && l.y<r.y));
}*/

/*void invers_modular(int a, int b, int& d, int& x, int & y)
{
    if(!b)
    {
        d=a;
        x=1;
        y=0;
        return ;
    }
    int x0, y0;
    invers_modular(b, a%b, d, x0, y0);
    x=y0;
    y=x0-a/b*y0;
} // daca x<0 se aduna cu mod pana e mai mare, x fiind rezultatul*/

/*ull p(int baze, int exponent)
{
    if(exponent==0)
        return 1;

    if(exponent%2==0)
        return p(baze*baze, exponent/2);
    else return baze*p(baze, exponent-1);
}*/
ifstream fin("dijkstra.in"); /// modifica cu numele corespunzator
ofstream fout("dijkstra.out"); /// modifica cu numele corespunzator

int n, m;
int a, b, c;
map<pair<int,int>, int> lungime;
unordered_map<int, int> drum_nod;
vector<vector<int> > lista(50001);

void citire();
void dijkstra();
void afisare();

int main()
{
    citire();
    dijkstra();
    afisare();

    return 0;
}

void citire()
{
    fin>>n>>m;
    for1(i,m)
    {
        fin>>a>>b>>c;
        lungime[mp(a,b)]=c;
        lista[a].pb(b);
    }
    for1(i,n)
        drum_nod[i]=2100000000;
    drum_nod[1]=0;
}

void dijkstra()
{
    queue<int> coada;
    coada.push(1);
    while(!coada.empty())
    {
        int actual=coada.front();
        coada.pop();
        for(auto i: lista[actual])
            if(drum_nod[i]>drum_nod[actual]+lungime[mp(actual,i)])
            {
                drum_nod[i]=drum_nod[actual]+lungime[mp(actual,i)];
                coada.push(i);
            }
    }
}

void afisare()
{
    for(int i=2; i<=n; i++)
        fout<<drum_nod[i]<<' ';
}