Cod sursa(job #1518682)

Utilizator PetrutiuPaulPetrutiu Paul Gabriel PetrutiuPaul Data 6 noiembrie 2015 08:42:03
Problema Distante Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.28 kb
#include <iostream>
#include <fstream>
#include <set>
#include <vector>

using namespace std;

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

vector<int> v[1000005],c[1000007];
set<pair< int, int> >s;
int d[1000005],n,m,inf=10000000000;
int ok[1000005];

void dijkstra(int nod)
{
    int x,val;
    for(int i=1;i<=n;i++)d[i]=inf;

    d[nod]=0;

    s.insert(make_pair(0,nod));

    while(s.size()>0)
    {
        val=(*s.begin()).first;
        x=(*s.begin()).second;
        s.erase(*s.begin());

        for(int i=0;i<v[x].size();i++)
        {
            if(d[ v[x][i]] > c[x][i]+val)
            {
                d[v[x][i]]=c[x][i]+val;
                s.insert(make_pair(d[v[x][i]],v[x][i]));
            }
        }
    }
}

int main()
{
    int x,y,z,t,nod;

    fin>>t;

    for(int j=1;j<=t;j++)
    {
        fin>>n>>m>>nod;

        for(int i=1;i<=n;i++)
          fin>>ok[i];

        for(int i=1;i<=m;i++)
        {
            fin>>x>>y>>z;
            v[x].push_back(y);
            c[x].push_back(z);
        }

        dijkstra(nod);

        int k=0;

        for(int i=1;i<=n;i++)
            if(ok[i]!=d[i])
                k++;
        if(k!=0)fout<<"NU\n";
        else fout<<"DA\n";
    }
}