Cod sursa(job #2521669)

Utilizator SoranaAureliaCatrina Sorana SoranaAurelia Data 11 ianuarie 2020 12:22:03
Problema Distante Scor 60
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.68 kb
#include <fstream>
#include <queue>
#include <vector>

#define NMAX 50005
using namespace std;

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

struct nod{
    int y;
    int cost;
};

int t, n, m, s;
int dmindeverif[NMAX], dmin[NMAX];
int viz[NMAX];
int decateori[NMAX];

vector<nod>graph[NMAX];
queue<int>Q;
int parcurgere(){
    Q.push(s);
    viz[s]=1;
    dmin[s]=0;
    while(!Q.empty()){
        int x=Q.front();
        for(auto &v:graph[x]){
            if(dmin[v.y] > dmin[x]+v.cost){
                dmin[v.y]=dmin[x]+v.cost;
                if(viz[v.y]==0)
                    Q.push(v.y);
                viz[v.y]=1;
                decateori[v.y]++;
                if(decateori[v.y]>n){
                    return 0;
                }
            }
        }
        viz[x]=0;
        Q.pop();
    }

}

void init(){
    for(int i=1; i<=n; i++)
    {
        dmin[i] = 0x3f3f3f3f;
        viz[i] = 0;
        graph[i].clear();
    }

}
void teste(){
    f>>n>>m>>s;
    init();
    for(int i=1; i<=n; i++)
        f>>dmindeverif[i];
    int x, y, c;
    for(int i=1; i<=m; i++){
        f>>x>>y>>c;
        graph[x].push_back({y, c});
        graph[y].push_back({x, c});
    }
    int ok1 = parcurgere();
    if(ok1 == 0)
        g<<"NU"<<'\n';
    else{
        dmin[s] = 0;
        int ok=1;
        for(int i=1; i<=n && ok==1; i++){
            if(dmin[i] != dmindeverif[i])
                ok=0;
        }
        if(ok == 0)
            g<<"NU"<<'\n';
        else g<<"DA"<<'\n';
    }
}
int main()
{
    ios::sync_with_stdio(false);
    f>>t;
    for(int i=1; i<=t; i++)
        teste();
    return 0;
}