Cod sursa(job #2422971)

Utilizator pasoi_stefanPasoi Stefan pasoi_stefan Data 20 mai 2019 15:10:58
Problema Distante Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.25 kb
#include<fstream>
#include<vector>
#include<string>
#include<queue>
using namespace std;
ifstream cin("distante.in");
ofstream cout("distante.out");

typedef unsigned short us;

int t;
int n,m,s;
int cnt;
vector<int> d;
vector<bool> viz;
vector<vector<pair<us,us> > > G;
queue<us> Q;

string ans(){

    if(d[s]!=0) return "NU";
    Q.push(s);
    viz[s]=1;
    cnt=n;
    while(!Q.empty()){

        us x=Q.front();
        Q.pop(); --cnt;

        for(int i=0;i<G[x].size();i++){

            us y=G[x][i].first;
            us z=G[x][i].second;

            if(!viz[y] && d[x]+z==d[y]){

                Q.push(y);
                viz[y]=1;

            }

        }

    }

    if(cnt>0) return "NU";
    return "DA";

}

int main(){

    cin>>t;
    while(t--){

        cin>>n>>m>>s;

        viz.resize(n+1,0);

        d.resize(n+1);
        for(us i=1;i<=n;i++)
            cin>>d[i];

        G.resize(n+1);
        for(int i=1;i<=m;i++){

            us x,y,z;
            cin>>x>>y>>z;
            G[x].push_back(make_pair(y,z));
            G[y].push_back(make_pair(x,z));

        }

        cout<<ans()<<'\n';

        d.clear();
        viz.clear();
        G.clear();

    }

}