Cod sursa(job #1315884)

Utilizator hanganflorinHangan Florin hanganflorin Data 13 ianuarie 2015 11:26:51
Problema Distante Scor 50
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.53 kb
#include <fstream>
#include <vector>
#include <queue>
#include <cstring>
#define INF 0x3f3f3f3f
using namespace std;

ifstream is("distante.in");
ofstream os("distante.out");

int T, n, m, s;
int d[50001], a[50001];
vector<pair<int, int> > G[50001];
priority_queue<pair<int, int>, vector<pair<int, int> >, greater<pair<int, int> > > Q;

void Read();
void Dijkstra(int x);
bool Ok();

int main()
{
    is >> T;
    for ( int t = 0; t < T; ++t )
    {
        Read();
        Dijkstra(s);
        if ( Ok() )
            os << "DA\n";
        else
            os << "NU\n";
    }
    is.close();
    os.close();
    return 0;
}
void Read()
{
    is >> n >> m >> s;
    for ( int i = 1; i <= n; ++i )
    {
        is >> a[i];
        G[i].clear();
    }
    for ( int i = 0, x, y, c; i < m; ++i )
    {
        is >> x >> y >> c;
        G[x].push_back({y, c});
        G[y].push_back({x, c});
    }
}
void Dijkstra(int x)
{
    int c;
    memset(d, 63, sizeof(d) );
    d[x] = 0;
    Q.push({0, x});
    while ( !Q.empty() )
    {
        c = Q.top().first;
        x = Q.top().second;
        Q.pop();
        for (const auto& v : G[x] )
            if ( d[v.first] > c + v.second )
            {
                d[v.first] = c + v.second;
                Q.push({d[v.first], v.first});
            }
    }
}
bool Ok()
{
    for ( int i = 1; i <= n; ++i )
    {
        if ( d[i] == INF )
            d[i] = 0;
        if ( d[i] != a[i] )
            return false;
    }
    return true;
}