Cod sursa(job #1110620)

Utilizator swim406Teudan Adina swim406 Data 18 februarie 2014 11:39:52
Problema Distante Scor 60
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.35 kb
#include <stdio.h>
#define nmax 50001
#include <vector>
#include <queue>
#define inf 1<<30

using namespace std;

vector < pair <int, int> > G[nmax];
queue <int> q;

int main() {
    freopen ("distante.in", "r", stdin);
    freopen ("distante.out", "w", stdout);
    int T, N, M, S, cmin[nmax], i, a, b, c, D[nmax], m;
    bool ok;
    scanf ("%d", &T);
    ++T;
    while (--T) {
        scanf ("%d %d %d", &N, &M, &S);
        for (i = 1; i <= N; ++i) {
            cmin[i] = inf;
            G[i].clear();
            scanf ("%d", &D[i]);
        }
        for (i = 1; i <= M; ++i) {
            scanf ("%d %d %d", &a, &b, &c);
            G[a].push_back(make_pair(b,c));
            G[b].push_back(make_pair(a,c));
        }
        cmin[S] = 0;
        q.push(S);
        while (!q.empty()) {
            a = q.front();
            q.pop();
            m = G[a].size();
            for (i = 0; i < m; ++i) {
                b = G[a][i].first;
                c = G[a][i].second;
                if (cmin[b] > cmin[a] + c) {
                    cmin[b] = cmin[a] + c;
                    q.push(b);
                }
            }
        }
        ok = 1;
        for (i = 1; i <= N; ++i)
            if (D[i] != cmin[i]) ok = 0, i = N + 1;
        if (ok) printf ("DA\n");
        else printf ("NU\n");
    }
    return 0;
}