Cod sursa(job #1110716)

Utilizator swim406Teudan Adina swim406 Data 18 februarie 2014 12:28:44
Problema Distante Scor 60
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.78 kb
#include <stdio.h>
#define nmax 50001
#include <vector>
#include <queue>

using namespace std;

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

int main() {
    freopen ("distante.in", "r", stdin);
    freopen ("distante.out", "w", stdout);
    int T, N, M, S, i, a, b, c, D[nmax], m, j;
    bool ok, oki;
    scanf ("%d", &T);
    ++T;
    while (--T) {
        scanf ("%d %d %d", &N, &M, &S);
        for (i = 1; i <= N; ++i) {
            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));
        }
        ok = 1;
        if (D[S]) ok = 0;
        if (ok) {
            for (i = 1; i <= N; ++i) {
                if (i != S) {
                    m = G[i].size();
                    for (j = 0; j < m; ++j) {
                        a = G[i][j].first;
                        b = G[i][j].second;
                        if (D[a] + b < D[i])
                            ok = 0, j = m, i = N + 1;
                    }
                }
            }
        }
        if (ok) {
            for (i = 1; i <= N; ++i) {
                if (i != S) {
                    oki = 0;
                    m = G[i].size();
                    for (j = 0; j < m; ++j) {
                        a = G[i][j].first;
                        b = G[i][j].second;
                        if (D[a] + b == D[i])
                            oki = 1, j = m;
                    }
                }
                if (i == S) oki = 1;
                ok = ok && oki;
                if (!ok) i = N + 1;
            }
        }
        if (ok) printf ("DA\n");
        else printf ("NU\n");
    }
    return 0;
}