Cod sursa(job #1059241)

Utilizator pulseOvidiu Giorgi pulse Data 16 decembrie 2013 14:54:15
Problema Algoritmul lui Dijkstra Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.36 kb
#include <fstream>
#include <vector>
#include <algorithm>
#include <queue>
#include <cstring>
using namespace std;

ifstream fin("dijkstra.in"); ofstream fout("dijkstra.out");

const int MAXN = 50005;
const int INF = 0x3f3f3f3f;

int N, M, GMAX;
vector< pair<int, int> > G[MAXN];
int Dmin[MAXN];

void ReadData() {
    fin >> N >> M >> GMAX;
    for (int i = 0; i < M; ++i) {
        int a, b, c;

        fin >> a >> b >> c;
        if (c < GMAX) c = 0;
        else if (c >= GMAX) c = 1;
        G[a].push_back(make_pair(b, c));
    }
}

void Solve() {
    bool InQueue[MAXN];
    queue<int> Q;

    memset(Dmin, INF, sizeof(Dmin));
    memset(InQueue, false, sizeof(InQueue));

    Dmin[1] = 0;
    Q.push(1);
    InQueue[1] = true;

    while (!Q.empty()) {
        int nod = Q.front();
        Q.pop();
        InQueue[nod] = false;

        for (vector< pair<int, int> >::iterator it = G[nod].begin(); it != G[nod].end(); ++it)
            if (Dmin[nod] + it->second < Dmin[it->first]) {
                Dmin[it->first] = Dmin[nod] + it->second;
                if (!InQueue[it->first]) {
                    Q.push(it->first);
                    InQueue[it->first] = true;
                }
            }
    }
}

void WriteData() {
	 fout << Dmin[N] << "\n";
}

int main() {
    ReadData();
    Solve();
    WriteData();
}