Cod sursa(job #1679241)

Utilizator razvandRazvan Dumitru razvand Data 7 aprilie 2016 19:29:45
Problema Distante Scor 30
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.93 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
#include <limits.h>
#define MAX 50003

using namespace std;

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

typedef bool (*comp)(int,int);
bool cmp(pair<int, int> a, pair<int, int> b) {
    return a.first > b.first;
}

vector<pair<int, int>> vec[MAX];
//vector<pair<int, int>> dist;
priority_queue<pair<int, int>, vector<pair<int, int>>, less<pair<int, int>>> dist;
int dis[MAX];
int dis2[MAX];

void dijkstra(int s) {

    //dist.push_back(make_pair(0, s));
    dist.push(make_pair(0, s));
    int nod,cos;
    vector<pair<int, int>>::iterator it;

    while(!dist.empty()) {

        nod = dist.top().second;
        cos = dist.top().first;
        dist.pop();
        dis2[nod] = cos;

        for(it = vec[nod].begin(); it != vec[nod].end(); it++) {

            if(cos+it->second < dis2[it->first]) {

                dis2[it->first] = cos+it->second;
                dist.push(make_pair(dis2[it->first], it->first));
                //push_heap(dist.begin(), dist.end());

            }

        }

    }

}

void sol() {

    int n,m,s,a,b,c;
    in >> n >> m >> s;
    for(int i = 1; i <= n; i++)
        in >> dis[i];

    for(int i = 1; i <= n; i++) {
        while(!vec[i].empty())
            vec[i].pop_back();
        dis2[i] = INT_MAX;
    }

    for(int i = 1; i <= m; i++) {
        in >> a >> b >> c;
        vec[a].push_back(make_pair(b, c));
        vec[b].push_back(make_pair(a, c));
    }

    dijkstra(s);

    bool f = false;

    for(int i = 1; i <= n; i++) {

        if(dis2[i] == INT_MAX)
            dis2[i] = 0;

        if(dis[i] != dis2[i]) {

            f = true;

        }

    }

    if(f)
        out << "NU\n";
    else
        out << "DA\n";

}

int main() {

    int t;
    in >> t;

    for(int i = 0; i < t; i++)
        sol();

    return 0;
}