Cod sursa(job #2713052)

Utilizator eugen5092eugen barbulescu eugen5092 Data 27 februarie 2021 10:27:01
Problema Distante Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.27 kb
#include <bits/stdc++.h>
#define inf 2000000000
using namespace std;

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

struct date{
    int x,c;
};

int t,n,m,s;
int d[50005];
int dp[50005];
vector<date>v[50005];

queue<int>q;

void rst(){
    for(int i=1;i<=n;i++){
        dp[i]=inf;
    }
    dp[s]=0;
}

void Djakstra(int r){
    q.push(r);
    int w,w1;
    while(!q.empty()){
        w=q.front();
        q.pop();
        for(auto i:v[w]){
            if(dp[i.x]>dp[w]+i.c){
                dp[i.x]=dp[w]+i.c;
                q.push(i.x);
            }
        }
    }
}

void verif(){
    for(int i=1;i<=n;i++){
        if(d[i]!=dp[i]){
            cou<<"NU \n";
            return;
        }
    }
    cou<<"DA \n";
}


void citire(){
    ci>>t;
    while(t--){
        ci>>n>>m>>s;
        for(int i=1;i<=n;i++){
            ci>>d[i];
        }
        rst();
        int a,b,c;
        for(int i=1;i<=m;i++){
            ci>>a>>b>>c;
            date t;
            t.x=b;
            t.c=c;
            v[a].push_back(t);
            t.x=a;
            v[b].push_back(t);
        }
        Djakstra(s);
        verif();
        //cout<<"\n\n";
    }
}

int main()
{
    citire();
    return 0;
}